Interface update, per-method Contexts.

filter/featureBreaker/rds.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 featureBreaker
 
 import (
         "golang.org/x/net/context"
 
         ds "github.com/luci/gae/service/datastore"
@@ skipping 55 matching lines @@
                 return r.rds.GetMulti(keys, meta, cb)
         })
 }
 
 func (r *dsState) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.NewKeyCB) error {
         return r.run(func() (err error) {
                 return r.rds.PutMulti(keys, vals, cb)
         })
 }
 
-func (r *dsState) Testable() ds.Testable {
-        return r.rds.Testable()
+func (r *dsState) WithTransaction(t ds.Transaction) context.Context {
+        return r.rds.WithTransaction(t)
+}
+
+func (r *dsState) CurrentTransaction() ds.Transaction {
+        return r.rds.CurrentTransaction()
+}
+
+func (r *dsState) GetTestable() ds.Testable {
+        return r.rds.GetTestable()
 }
 
 // FilterRDS installs a featureBreaker datastore filter in the context.
 func FilterRDS(c context.Context, defaultError error) (context.Context, FeatureBreaker) {
         state := newState(defaultError)
         return ds.AddRawFilters(c, func(ic context.Context, RawDatastore ds.RawInterface) ds.RawInterface {
                 return &dsState{state, RawDatastore}
         }), state
 }