| Index: filter/count/tq.go
|
| diff --git a/filter/count/tq.go b/filter/count/tq.go
|
| index 054dae7eee59d74fd02c86ee3b3605b89201dfcf..65f0955b054e86be5a6027f843ba783f7579a4a4 100644
|
| --- a/filter/count/tq.go
|
| +++ b/filter/count/tq.go
|
| @@ -12,70 +12,40 @@ import (
|
|
|
| // TQCounter is the counter object for the TaskQueue service.
|
| type TQCounter struct {
|
| - Add Entry
|
| - Delete Entry
|
| AddMulti Entry
|
| DeleteMulti Entry
|
| - Lease Entry
|
| - LeaseByTag Entry
|
| - ModifyLease Entry
|
| Purge Entry
|
| - QueueStats Entry
|
| + Stats Entry
|
| }
|
|
|
| type tqCounter struct {
|
| c *TQCounter
|
|
|
| - tq tq.Interface
|
| + tq tq.RawInterface
|
| }
|
|
|
| -var _ tq.Interface = (*tqCounter)(nil)
|
| +var _ tq.RawInterface = (*tqCounter)(nil)
|
|
|
| -func (t *tqCounter) Add(task *tq.Task, queueName string) (*tq.Task, error) {
|
| - ret, err := t.tq.Add(task, queueName)
|
| - return ret, t.c.Add.up(err)
|
| +func (t *tqCounter) AddMulti(tasks []*tq.Task, queueName string, cb tq.RawTaskCB) error {
|
| + return t.c.AddMulti.up(t.tq.AddMulti(tasks, queueName, cb))
|
| }
|
|
|
| -func (t *tqCounter) Delete(task *tq.Task, queueName string) error {
|
| - return t.c.Delete.up(t.tq.Delete(task, queueName))
|
| -}
|
| -
|
| -func (t *tqCounter) AddMulti(tasks []*tq.Task, queueName string) ([]*tq.Task, error) {
|
| - ret, err := t.tq.AddMulti(tasks, queueName)
|
| - return ret, t.c.AddMulti.up(err)
|
| -}
|
| -
|
| -func (t *tqCounter) DeleteMulti(tasks []*tq.Task, queueName string) error {
|
| - return t.c.DeleteMulti.up(t.tq.DeleteMulti(tasks, queueName))
|
| -}
|
| -
|
| -func (t *tqCounter) Lease(maxTasks int, queueName string, leaseTime int) ([]*tq.Task, error) {
|
| - ret, err := t.tq.Lease(maxTasks, queueName, leaseTime)
|
| - return ret, t.c.Lease.up(err)
|
| -}
|
| -
|
| -func (t *tqCounter) LeaseByTag(maxTasks int, queueName string, leaseTime int, tag string) ([]*tq.Task, error) {
|
| - ret, err := t.tq.LeaseByTag(maxTasks, queueName, leaseTime, tag)
|
| - return ret, t.c.LeaseByTag.up(err)
|
| -}
|
| -
|
| -func (t *tqCounter) ModifyLease(task *tq.Task, queueName string, leaseTime int) error {
|
| - return t.c.ModifyLease.up(t.tq.ModifyLease(task, queueName, leaseTime))
|
| +func (t *tqCounter) DeleteMulti(tasks []*tq.Task, queueName string, cb tq.RawCB) error {
|
| + return t.c.DeleteMulti.up(t.tq.DeleteMulti(tasks, queueName, cb))
|
| }
|
|
|
| func (t *tqCounter) Purge(queueName string) error {
|
| return t.c.Purge.up(t.tq.Purge(queueName))
|
| }
|
|
|
| -func (t *tqCounter) QueueStats(queueNames []string) ([]tq.Statistics, error) {
|
| - ret, err := t.tq.QueueStats(queueNames)
|
| - return ret, t.c.QueueStats.up(err)
|
| +func (t *tqCounter) Stats(queueNames []string, cb tq.RawStatsCB) error {
|
| + return t.c.Stats.up(t.tq.Stats(queueNames, cb))
|
| }
|
|
|
| // FilterTQ installs a counter TaskQueue filter in the context.
|
| func FilterTQ(c context.Context) (context.Context, *TQCounter) {
|
| state := &TQCounter{}
|
| - return tq.AddFilters(c, func(ic context.Context, tq tq.Interface) tq.Interface {
|
| + return tq.AddRawFilters(c, func(ic context.Context, tq tq.RawInterface) tq.RawInterface {
|
| return &tqCounter{state, tq}
|
| }), state
|
| }
|
|
|