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 |