| 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 24 matching lines...) Expand all Loading... |
| 35 } | 35 } |
| 36 | 36 |
| 37 func (t *tqCounter) Purge(queueName string) error { | 37 func (t *tqCounter) Purge(queueName string) error { |
| 38 return t.c.Purge.up(t.tq.Purge(queueName)) | 38 return t.c.Purge.up(t.tq.Purge(queueName)) |
| 39 } | 39 } |
| 40 | 40 |
| 41 func (t *tqCounter) Stats(queueNames []string, cb tq.RawStatsCB) error { | 41 func (t *tqCounter) Stats(queueNames []string, cb tq.RawStatsCB) error { |
| 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 { |
| 46 return t.tq.Testable() |
| 47 } |
| 48 |
| 45 // FilterTQ installs a counter TaskQueue filter in the context. | 49 // FilterTQ installs a counter TaskQueue filter in the context. |
| 46 func FilterTQ(c context.Context) (context.Context, *TQCounter) { | 50 func FilterTQ(c context.Context) (context.Context, *TQCounter) { |
| 47 state := &TQCounter{} | 51 state := &TQCounter{} |
| 48 return tq.AddRawFilters(c, func(ic context.Context, tq tq.RawInterface)
tq.RawInterface { | 52 return tq.AddRawFilters(c, func(ic context.Context, tq tq.RawInterface)
tq.RawInterface { |
| 49 return &tqCounter{state, tq} | 53 return &tqCounter{state, tq} |
| 50 }), state | 54 }), state |
| 51 } | 55 } |
| OLD | NEW |