Index: filter/featureBreaker/rds.go |
diff --git a/filter/featureBreaker/rds.go b/filter/featureBreaker/rds.go |
index d95643910827a6cdc9398534c57872c4e321c033..e1ef7f20fcfb6a9b52a70b3705452d1ca3264c62 100644 |
--- a/filter/featureBreaker/rds.go |
+++ b/filter/featureBreaker/rds.go |
@@ -7,16 +7,16 @@ package featureBreaker |
import ( |
"golang.org/x/net/context" |
- rds "github.com/luci/gae/service/rawdatastore" |
+ ds "github.com/luci/gae/service/datastore" |
) |
-type rdsState struct { |
+type dsState struct { |
*state |
- rds.Interface |
+ ds.Interface |
} |
-func (r *rdsState) DecodeKey(encoded string) (ret rds.Key, err error) { |
+func (r *dsState) DecodeKey(encoded string) (ret ds.Key, err error) { |
err = r.run(func() (err error) { |
ret, err = r.Interface.DecodeKey(encoded) |
return |
@@ -24,7 +24,7 @@ func (r *rdsState) DecodeKey(encoded string) (ret rds.Key, err error) { |
return |
} |
-func (r *rdsState) RunInTransaction(f func(c context.Context) error, opts *rds.TransactionOptions) error { |
+func (r *dsState) RunInTransaction(f func(c context.Context) error, opts *ds.TransactionOptions) error { |
return r.run(func() error { |
return r.Interface.RunInTransaction(f, opts) |
}) |
@@ -33,28 +33,28 @@ func (r *rdsState) RunInTransaction(f func(c context.Context) error, opts *rds.T |
// TODO(riannucci): Allow the user to specify a multierror which will propagate |
// to the callback correctly. |
-func (r *rdsState) DeleteMulti(keys []rds.Key, cb rds.DeleteMultiCB) error { |
+func (r *dsState) DeleteMulti(keys []ds.Key, cb ds.DeleteMultiCB) error { |
return r.run(func() error { |
return r.Interface.DeleteMulti(keys, cb) |
}) |
} |
-func (r *rdsState) GetMulti(keys []rds.Key, cb rds.GetMultiCB) error { |
+func (r *dsState) GetMulti(keys []ds.Key, cb ds.GetMultiCB) error { |
return r.run(func() error { |
return r.Interface.GetMulti(keys, cb) |
}) |
} |
-func (r *rdsState) PutMulti(keys []rds.Key, vals []rds.PropertyLoadSaver, cb rds.PutMultiCB) error { |
+func (r *dsState) PutMulti(keys []ds.Key, vals []ds.PropertyLoadSaver, cb ds.PutMultiCB) error { |
return r.run(func() (err error) { |
return r.Interface.PutMulti(keys, vals, cb) |
}) |
} |
-// FilterRDS installs a counter RawDatastore filter in the context. |
+// FilterRDS installs a counter datastore filter in the context. |
func FilterRDS(c context.Context, defaultError error) (context.Context, FeatureBreaker) { |
state := newState(defaultError) |
- return rds.AddFilters(c, func(ic context.Context, RawDatastore rds.Interface) rds.Interface { |
- return &rdsState{state, RawDatastore} |
+ return ds.AddFilters(c, func(ic context.Context, datastore ds.Interface) ds.Interface { |
+ return &dsState{state, datastore} |
}), state |
} |