| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The LUCI Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed under the Apache License, Version 2.0 |
| 3 // found in the LICENSE file. | 3 // that can be found in the LICENSE file. |
| 4 | 4 |
| 5 package txnBuf | 5 package txnBuf |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 "golang.org/x/net/context" | 8 "golang.org/x/net/context" |
| 9 | 9 |
| 10 ds "github.com/luci/gae/service/datastore" | 10 ds "github.com/luci/gae/service/datastore" |
| 11 ) | 11 ) |
| 12 | 12 |
| 13 type dsBuf struct { | 13 type dsBuf struct { |
| 14 ds.RawInterface | 14 ds.RawInterface |
| 15 } | 15 } |
| 16 | 16 |
| 17 var _ ds.RawInterface = (*dsBuf)(nil) | 17 var _ ds.RawInterface = (*dsBuf)(nil) |
| 18 | 18 |
| 19 func (d *dsBuf) RunInTransaction(f func(context.Context) error, opts *ds.Transac
tionOptions) error { | 19 func (d *dsBuf) RunInTransaction(f func(context.Context) error, opts *ds.Transac
tionOptions) error { |
| 20 return doRunInTransaction(d.RawInterface, f, opts) | 20 return doRunInTransaction(d.RawInterface, f, opts) |
| 21 } | 21 } |
| 22 | 22 |
| 23 func doRunInTransaction(base ds.RawInterface, f func(context.Context) error, opt
s *ds.TransactionOptions) error { | 23 func doRunInTransaction(base ds.RawInterface, f func(context.Context) error, opt
s *ds.TransactionOptions) error { |
| 24 return base.RunInTransaction(func(ctx context.Context) error { | 24 return base.RunInTransaction(func(ctx context.Context) error { |
| 25 return withTxnBuf(ctx, f, opts) | 25 return withTxnBuf(ctx, f, opts) |
| 26 }, opts) | 26 }, opts) |
| 27 } | 27 } |
| OLD | NEW |