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 tq "github.com/luci/gae/service/taskqueue" | 10 tq "github.com/luci/gae/service/taskqueue" |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
42 return t.c.Stats.up(t.tq.Stats(queueNames, cb)) | 42 return t.c.Stats.up(t.tq.Stats(queueNames, cb)) |
43 } | 43 } |
44 | 44 |
45 func (t *tqCounter) Testable() tq.Testable { | 45 func (t *tqCounter) Testable() tq.Testable { |
46 return t.tq.Testable() | 46 return t.tq.Testable() |
47 } | 47 } |
48 | 48 |
49 // FilterTQ installs a counter TaskQueue filter in the context. | 49 // FilterTQ installs a counter TaskQueue filter in the context. |
50 func FilterTQ(c context.Context) (context.Context, *TQCounter) { | 50 func FilterTQ(c context.Context) (context.Context, *TQCounter) { |
51 state := &TQCounter{} | 51 state := &TQCounter{} |
| 52 state.AddMulti.init() |
| 53 state.DeleteMulti.init() |
| 54 state.Purge.init() |
| 55 state.Stats.init() |
| 56 |
52 return tq.AddRawFilters(c, func(ic context.Context, tq tq.RawInterface)
tq.RawInterface { | 57 return tq.AddRawFilters(c, func(ic context.Context, tq tq.RawInterface)
tq.RawInterface { |
53 return &tqCounter{state, tq} | 58 return &tqCounter{state, tq} |
54 }), state | 59 }), state |
55 } | 60 } |
OLD | NEW |