| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium 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 by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 package featureBreaker | 5 package featureBreaker |
| 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" |
| (...skipping 21 matching lines...) Expand all Loading... |
| 32 | 32 |
| 33 // TODO(riannucci): Allow the user to specify a multierror which will propagate | 33 // TODO(riannucci): Allow the user to specify a multierror which will propagate |
| 34 // to the callback correctly. | 34 // to the callback correctly. |
| 35 | 35 |
| 36 func (r *dsState) DeleteMulti(keys []ds.Key, cb ds.DeleteMultiCB) error { | 36 func (r *dsState) DeleteMulti(keys []ds.Key, cb ds.DeleteMultiCB) error { |
| 37 return r.run(func() error { | 37 return r.run(func() error { |
| 38 return r.RawInterface.DeleteMulti(keys, cb) | 38 return r.RawInterface.DeleteMulti(keys, cb) |
| 39 }) | 39 }) |
| 40 } | 40 } |
| 41 | 41 |
| 42 func (r *dsState) GetMulti(keys []ds.Key, cb ds.GetMultiCB) error { | 42 func (r *dsState) GetMulti(keys []ds.Key, meta ds.MultiMetaGetter, cb ds.GetMult
iCB) error { |
| 43 return r.run(func() error { | 43 return r.run(func() error { |
| 44 » » return r.RawInterface.GetMulti(keys, cb) | 44 » » return r.RawInterface.GetMulti(keys, meta, cb) |
| 45 }) | 45 }) |
| 46 } | 46 } |
| 47 | 47 |
| 48 func (r *dsState) PutMulti(keys []ds.Key, vals []ds.PropertyMap, cb ds.PutMultiC
B) error { | 48 func (r *dsState) PutMulti(keys []ds.Key, vals []ds.PropertyMap, cb ds.PutMultiC
B) error { |
| 49 return r.run(func() (err error) { | 49 return r.run(func() (err error) { |
| 50 return r.RawInterface.PutMulti(keys, vals, cb) | 50 return r.RawInterface.PutMulti(keys, vals, cb) |
| 51 }) | 51 }) |
| 52 } | 52 } |
| 53 | 53 |
| 54 // FilterRDS installs a counter datastore filter in the context. | 54 // FilterRDS installs a counter datastore filter in the context. |
| 55 func FilterRDS(c context.Context, defaultError error) (context.Context, FeatureB
reaker) { | 55 func FilterRDS(c context.Context, defaultError error) (context.Context, FeatureB
reaker) { |
| 56 state := newState(defaultError) | 56 state := newState(defaultError) |
| 57 return ds.AddRawFilters(c, func(ic context.Context, RawDatastore ds.RawI
nterface) ds.RawInterface { | 57 return ds.AddRawFilters(c, func(ic context.Context, RawDatastore ds.RawI
nterface) ds.RawInterface { |
| 58 return &dsState{state, RawDatastore} | 58 return &dsState{state, RawDatastore} |
| 59 }), state | 59 }), state |
| 60 } | 60 } |
| OLD | NEW |