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 mc "github.com/luci/gae/service/memcache" | 10 mc "github.com/luci/gae/service/memcache" |
11 ) | 11 ) |
12 | 12 |
13 type mcState struct { | 13 type mcState struct { |
14 *state | 14 *state |
15 | 15 |
16 » mc.Interface | 16 » mc.RawInterface |
17 } | 17 } |
18 | 18 |
19 func (m *mcState) Get(key string) (ret mc.Item, err error) { | 19 func (m *mcState) GetMulti(keys []string, cb mc.RawItemCB) error { |
20 » err = m.run(func() (err error) { | 20 » return m.run(func() error { return m.RawInterface.GetMulti(keys, cb) }) |
21 » » ret, err = m.Interface.Get(key) | |
22 » » return | |
23 » }) | |
24 » return | |
25 } | 21 } |
26 | 22 |
27 func (m *mcState) GetMulti(keys []string) (ret map[string]mc.Item, err error) { | 23 func (m *mcState) AddMulti(items []mc.Item, cb mc.RawCB) error { |
28 » err = m.run(func() (err error) { | 24 » return m.run(func() error { return m.RawInterface.AddMulti(items, cb) }) |
29 » » ret, err = m.Interface.GetMulti(keys) | |
30 » » return | |
31 » }) | |
32 » return | |
33 } | 25 } |
34 | 26 |
35 func (m *mcState) Add(item mc.Item) error { | 27 func (m *mcState) SetMulti(items []mc.Item, cb mc.RawCB) error { |
36 » return m.run(func() error { return m.Interface.Add(item) }) | 28 » return m.run(func() error { return m.RawInterface.SetMulti(items, cb) }) |
37 } | 29 } |
38 | 30 |
39 func (m *mcState) Set(item mc.Item) error { | 31 func (m *mcState) DeleteMulti(keys []string, cb mc.RawCB) error { |
40 » return m.run(func() error { return m.Interface.Set(item) }) | 32 » return m.run(func() error { return m.RawInterface.DeleteMulti(keys, cb)
}) |
41 } | 33 } |
42 | 34 |
43 func (m *mcState) Delete(key string) error { | 35 func (m *mcState) CompareAndSwapMulti(items []mc.Item, cb mc.RawCB) error { |
44 » return m.run(func() error { return m.Interface.Delete(key) }) | 36 » return m.run(func() error { return m.RawInterface.CompareAndSwapMulti(it
ems, cb) }) |
45 } | |
46 | |
47 func (m *mcState) CompareAndSwap(item mc.Item) error { | |
48 » return m.run(func() error { return m.Interface.CompareAndSwap(item) }) | |
49 } | |
50 | |
51 func (m *mcState) AddMulti(items []mc.Item) error { | |
52 » return m.run(func() error { return m.Interface.AddMulti(items) }) | |
53 } | |
54 | |
55 func (m *mcState) SetMulti(items []mc.Item) error { | |
56 » return m.run(func() error { return m.Interface.SetMulti(items) }) | |
57 } | |
58 | |
59 func (m *mcState) DeleteMulti(keys []string) error { | |
60 » return m.run(func() error { return m.Interface.DeleteMulti(keys) }) | |
61 } | 37 } |
62 | 38 |
63 func (m *mcState) Flush() error { | 39 func (m *mcState) Flush() error { |
64 » return m.run(func() error { return m.Interface.Flush() }) | 40 » return m.run(m.RawInterface.Flush) |
65 } | |
66 | |
67 func (m *mcState) CompareAndSwapMulti(items []mc.Item) error { | |
68 » return m.run(func() error { return m.Interface.CompareAndSwapMulti(items
) }) | |
69 } | |
70 | |
71 func (m *mcState) Increment(key string, delta int64, initialValue uint64) (newVa
lue uint64, err error) { | |
72 » err = m.run(func() (err error) { | |
73 » » newValue, err = m.Interface.Increment(key, delta, initialValue) | |
74 » » return | |
75 » }) | |
76 » return | |
77 } | |
78 | |
79 func (m *mcState) IncrementExisting(key string, delta int64) (newValue uint64, e
rr error) { | |
80 » err = m.run(func() (err error) { | |
81 » » newValue, err = m.Interface.IncrementExisting(key, delta) | |
82 » » return | |
83 » }) | |
84 » return | |
85 } | 41 } |
86 | 42 |
87 func (m *mcState) Stats() (ret *mc.Statistics, err error) { | 43 func (m *mcState) Stats() (ret *mc.Statistics, err error) { |
88 err = m.run(func() (err error) { | 44 err = m.run(func() (err error) { |
89 » » ret, err = m.Interface.Stats() | 45 » » ret, err = m.RawInterface.Stats() |
90 return | 46 return |
91 }) | 47 }) |
92 return | 48 return |
93 } | 49 } |
94 | 50 |
95 // FilterMC installs a counter mc filter in the context. | 51 // FilterMC installs a counter mc filter in the context. |
96 func FilterMC(c context.Context, defaultError error) (context.Context, FeatureBr
eaker) { | 52 func FilterMC(c context.Context, defaultError error) (context.Context, FeatureBr
eaker) { |
97 state := newState(defaultError) | 53 state := newState(defaultError) |
98 » return mc.AddFilters(c, func(ic context.Context, rds mc.Interface) mc.In
terface { | 54 » return mc.AddRawFilters(c, func(ic context.Context, rds mc.RawInterface)
mc.RawInterface { |
99 return &mcState{state, rds} | 55 return &mcState{state, rds} |
100 }), state | 56 }), state |
101 } | 57 } |
OLD | NEW |