| OLD | NEW | 
|---|
| 1 // Copyright 2015 The LUCI Authors. All rights reserved. | 1 // Copyright 2015 The LUCI Authors. All rights reserved. | 
| 2 // Use of this source code is governed under the Apache License, Version 2.0 | 2 // Use of this source code is governed under the Apache License, Version 2.0 | 
| 3 // that can be found in the LICENSE file. | 3 // that can be 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 { | 45 func (t *tqCounter) GetTestable() tq.Testable { | 
| 46 »       return t.tq.Testable() | 46 »       return t.tq.GetTestable() | 
| 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         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 { | 
| 53                 return &tqCounter{state, tq} | 53                 return &tqCounter{state, tq} | 
| 54         }), state | 54         }), state | 
| 55 } | 55 } | 
| OLD | NEW | 
|---|