| 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 count | 5 package count |
| 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 // MCCounter is the counter object for the Memcache service. | 13 // MCCounter is the counter object for the Memcache service. |
| 14 type MCCounter struct { | 14 type MCCounter struct { |
| 15 NewItem Entry | 15 NewItem Entry |
| 16 Add Entry | |
| 17 Set Entry | |
| 18 Get Entry | |
| 19 Delete Entry | |
| 20 CompareAndSwap Entry | |
| 21 AddMulti Entry | 16 AddMulti Entry |
| 22 SetMulti Entry | 17 SetMulti Entry |
| 23 GetMulti Entry | 18 GetMulti Entry |
| 24 DeleteMulti Entry | 19 DeleteMulti Entry |
| 25 CompareAndSwapMulti Entry | 20 CompareAndSwapMulti Entry |
| 26 Increment Entry | 21 Increment Entry |
| 27 IncrementExisting Entry | |
| 28 Flush Entry | 22 Flush Entry |
| 29 Stats Entry | 23 Stats Entry |
| 30 } | 24 } |
| 31 | 25 |
| 32 type mcCounter struct { | 26 type mcCounter struct { |
| 33 c *MCCounter | 27 c *MCCounter |
| 34 | 28 |
| 35 » mc mc.Interface | 29 » mc mc.RawInterface |
| 36 } | 30 } |
| 37 | 31 |
| 38 var _ mc.Interface = (*mcCounter)(nil) | 32 var _ mc.RawInterface = (*mcCounter)(nil) |
| 39 | 33 |
| 40 func (m *mcCounter) NewItem(key string) mc.Item { | 34 func (m *mcCounter) NewItem(key string) mc.Item { |
| 41 m.c.NewItem.up() | 35 m.c.NewItem.up() |
| 42 return m.mc.NewItem(key) | 36 return m.mc.NewItem(key) |
| 43 } | 37 } |
| 44 | 38 |
| 45 func (m *mcCounter) Get(key string) (mc.Item, error) { | 39 func (m *mcCounter) GetMulti(keys []string, cb mc.RawItemCB) error { |
| 46 » ret, err := m.mc.Get(key) | 40 » return m.c.GetMulti.up(m.mc.GetMulti(keys, cb)) |
| 47 » return ret, m.c.Get.up(err) | |
| 48 } | 41 } |
| 49 | 42 |
| 50 func (m *mcCounter) GetMulti(keys []string) (map[string]mc.Item, error) { | 43 func (m *mcCounter) AddMulti(items []mc.Item, cb mc.RawCB) error { |
| 51 » ret, err := m.mc.GetMulti(keys) | 44 » return m.c.AddMulti.up(m.mc.AddMulti(items, cb)) |
| 52 » return ret, m.c.GetMulti.up(err) | |
| 53 } | 45 } |
| 54 | 46 |
| 55 func (m *mcCounter) Add(item mc.Item) error { return m.c.Add.up(m.mc.Add(item))
} | 47 func (m *mcCounter) SetMulti(items []mc.Item, cb mc.RawCB) error { |
| 56 func (m *mcCounter) Set(item mc.Item) error { return m.c.Set.up(m.mc.Set(item))
} | 48 » return m.c.SetMulti.up(m.mc.SetMulti(items, cb)) |
| 57 func (m *mcCounter) Delete(key string) error { return m.c.Delete.up(m.mc.Delete(
key)) } | |
| 58 func (m *mcCounter) CompareAndSwap(item mc.Item) error { | |
| 59 » return m.c.CompareAndSwap.up(m.mc.CompareAndSwap(item)) | |
| 60 } | 49 } |
| 61 func (m *mcCounter) AddMulti(items []mc.Item) error { return m.c.AddMulti.up(m.m
c.AddMulti(items)) } | 50 |
| 62 func (m *mcCounter) SetMulti(items []mc.Item) error { return m.c.SetMulti.up(m.m
c.SetMulti(items)) } | 51 func (m *mcCounter) DeleteMulti(keys []string, cb mc.RawCB) error { |
| 63 func (m *mcCounter) DeleteMulti(keys []string) error { | 52 » return m.c.DeleteMulti.up(m.mc.DeleteMulti(keys, cb)) |
| 64 » return m.c.DeleteMulti.up(m.mc.DeleteMulti(keys)) | |
| 65 } | 53 } |
| 54 |
| 55 func (m *mcCounter) CompareAndSwapMulti(items []mc.Item, cb mc.RawCB) error { |
| 56 return m.c.CompareAndSwapMulti.up(m.mc.CompareAndSwapMulti(items, cb)) |
| 57 } |
| 58 |
| 66 func (m *mcCounter) Flush() error { return m.c.Flush.up(m.mc.Flush()) } | 59 func (m *mcCounter) Flush() error { return m.c.Flush.up(m.mc.Flush()) } |
| 67 | 60 |
| 68 func (m *mcCounter) CompareAndSwapMulti(items []mc.Item) error { | 61 func (m *mcCounter) Increment(key string, delta int64, initialValue *uint64) (ne
wValue uint64, err error) { |
| 69 » return m.c.CompareAndSwapMulti.up(m.mc.CompareAndSwapMulti(items)) | |
| 70 } | |
| 71 | |
| 72 func (m *mcCounter) Increment(key string, delta int64, initialValue uint64) (new
Value uint64, err error) { | |
| 73 ret, err := m.mc.Increment(key, delta, initialValue) | 62 ret, err := m.mc.Increment(key, delta, initialValue) |
| 74 return ret, m.c.Increment.up(err) | 63 return ret, m.c.Increment.up(err) |
| 75 } | 64 } |
| 76 | 65 |
| 77 func (m *mcCounter) IncrementExisting(key string, delta int64) (newValue uint64,
err error) { | |
| 78 ret, err := m.mc.IncrementExisting(key, delta) | |
| 79 return ret, m.c.IncrementExisting.up(err) | |
| 80 } | |
| 81 | |
| 82 func (m *mcCounter) Stats() (*mc.Statistics, error) { | 66 func (m *mcCounter) Stats() (*mc.Statistics, error) { |
| 83 ret, err := m.mc.Stats() | 67 ret, err := m.mc.Stats() |
| 84 return ret, m.c.Stats.up(err) | 68 return ret, m.c.Stats.up(err) |
| 85 } | 69 } |
| 86 | 70 |
| 87 // FilterMC installs a counter Memcache filter in the context. | 71 // FilterMC installs a counter Memcache filter in the context. |
| 88 func FilterMC(c context.Context) (context.Context, *MCCounter) { | 72 func FilterMC(c context.Context) (context.Context, *MCCounter) { |
| 89 state := &MCCounter{} | 73 state := &MCCounter{} |
| 90 » return mc.AddFilters(c, func(ic context.Context, mc mc.Interface) mc.Int
erface { | 74 » return mc.AddRawFilters(c, func(ic context.Context, mc mc.RawInterface)
mc.RawInterface { |
| 91 return &mcCounter{state, mc} | 75 return &mcCounter{state, mc} |
| 92 }), state | 76 }), state |
| 93 } | 77 } |
| OLD | NEW |