Interface update, per-method Contexts.

filter/count/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 count
 
 import (
         "golang.org/x/net/context"
 
         ds "github.com/luci/gae/service/datastore"
@@ skipping 46 matching lines @@
 }
 
 func (r *dsCounter) GetMulti(keys []*ds.Key, meta ds.MultiMetaGetter, cb ds.GetMultiCB) error {
         return r.c.GetMulti.upFilterStop(r.ds.GetMulti(keys, meta, cb))
 }
 
 func (r *dsCounter) PutMulti(keys []*ds.Key, vals []ds.PropertyMap, cb ds.NewKeyCB) error {
         return r.c.PutMulti.upFilterStop(r.ds.PutMulti(keys, vals, cb))
 }
 
-func (r *dsCounter) Testable() ds.Testable {
-        return r.ds.Testable()
+func (r *dsCounter) CurrentTransaction() ds.Transaction {
+        return r.ds.CurrentTransaction()
+}
+func (r *dsCounter) WithoutTransaction() context.Context {
+        return r.ds.WithoutTransaction()
+}
+
+func (r *dsCounter) GetTestable() ds.Testable {
+        return r.ds.GetTestable()
 }
 
 // FilterRDS installs a counter datastore filter in the context.
 func FilterRDS(c context.Context) (context.Context, *DSCounter) {
         state := &DSCounter{}
         return ds.AddRawFilters(c, func(ic context.Context, ds ds.RawInterface) ds.RawInterface {
                 return &dsCounter{state, ds}
         }), state
 }
 
 // upFilterStop wraps up, handling the special case datastore.Stop error.
 // datastore.Stop will pass through this function, but, unlike other error
 // codes, will be counted as a success.
 func (e *Entry) upFilterStop(err error) error {
         upErr := err
         if upErr == ds.Stop {
                 upErr = nil
         }
         e.up(upErr)
         return err
 }