Interface update, per-method Contexts.

filter/txnBuf/ds_txn.go

 // Copyright 2015 The LUCI Authors. All rights reserved.
 // Use of this source code is governed under the Apache License, Version 2.0
 // that can be found in the LICENSE file.
 
 package txnBuf
 
 import (
         ds "github.com/luci/gae/service/datastore"
         "github.com/luci/luci-go/common/errors"
         "golang.org/x/net/context"
@@ skipping 98 matching lines @@
 }
 
 func (d *dsTxnBuf) RunInTransaction(cb func(context.Context) error, opts *ds.TransactionOptions) error {
         if !d.haveLock {
                 d.state.Lock()
                 defer d.state.Unlock()
         }
         return withTxnBuf(d.ic, cb, opts)
 }
 
-func (d *dsTxnBuf) Testable() ds.Testable {
-        return d.state.parentDS.Testable()
+func (d *dsTxnBuf) CurrentTransaction() ds.Transaction { return d.state.parentDS.CurrentTransaction() }
+
+func (d *dsTxnBuf) WithoutTransaction() context.Context {
+        c := d.state.parentDS.WithoutTransaction()
+        c = context.WithValue(c, dsTxnBufParent, nil)
+        c = context.WithValue(c, dsTxnBufHaveLock, nil)
+        return c
 }
+
+func (d *dsTxnBuf) GetTestable() ds.Testable {
+        return d.state.parentDS.GetTestable()
+}