| OLD | NEW |
|---|
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 package count | |
| 6 | |
| 7 import ( | |
| 8 "golang.org/x/net/context" | |
| 9 | |
| 10 tq "github.com/luci/gae/service/taskqueue" | |
| 11 ) | |
| 12 | |
| 13 // TQCounter is the counter object for the TaskQueue service. | |
| 14 type TQCounter struct { | |
| 15 Add Entry | |
| 16 Delete Entry | |
| 17 AddMulti Entry | |
| 18 DeleteMulti Entry | |
| 19 Lease Entry | |
| 20 LeaseByTag Entry | |
| 21 ModifyLease Entry | |
| 22 Purge Entry | |
| 23 QueueStats Entry | |
| 24 } | |
| 25 | |
| 26 type tqCounter struct { | |
| 27 c *TQCounter | |
| 28 | |
| 29 tq tq.Interface | |
| 30 } | |
| 31 | |
| 32 var _ tq.Interface = (*tqCounter)(nil) | |
| 33 | |
| 34 func (t *tqCounter) Add(task *tq.Task, queueName string) (*tq.Task, error) { | |
| 35 ret, err := t.tq.Add(task, queueName) | |
| 36 return ret, t.c.Add.up(err) | |
| 37 } | |
| 38 | |
| 39 func (t *tqCounter) Delete(task *tq.Task, queueName string) error { | |
| 40 return t.c.Delete.up(t.tq.Delete(task, queueName)) | |
| 41 } | |
| 42 | |
| 43 func (t *tqCounter) AddMulti(tasks []*tq.Task, queueName string) ([]*tq.Task, er
ror) { | |
| 44 ret, err := t.tq.AddMulti(tasks, queueName) | |
| 45 return ret, t.c.AddMulti.up(err) | |
| 46 } | |
| 47 | |
| 48 func (t *tqCounter) DeleteMulti(tasks []*tq.Task, queueName string) error { | |
| 49 return t.c.DeleteMulti.up(t.tq.DeleteMulti(tasks, queueName)) | |
| 50 } | |
| 51 | |
| 52 func (t *tqCounter) Lease(maxTasks int, queueName string, leaseTime int) ([]*tq.
Task, error) { | |
| 53 ret, err := t.tq.Lease(maxTasks, queueName, leaseTime) | |
| 54 return ret, t.c.Lease.up(err) | |
| 55 } | |
| 56 | |
| 57 func (t *tqCounter) LeaseByTag(maxTasks int, queueName string, leaseTime int, ta
g string) ([]*tq.Task, error) { | |
| 58 ret, err := t.tq.LeaseByTag(maxTasks, queueName, leaseTime, tag) | |
| 59 return ret, t.c.LeaseByTag.up(err) | |
| 60 } | |
| 61 | |
| 62 func (t *tqCounter) ModifyLease(task *tq.Task, queueName string, leaseTime int)
error { | |
| 63 return t.c.ModifyLease.up(t.tq.ModifyLease(task, queueName, leaseTime)) | |
| 64 } | |
| 65 | |
| 66 func (t *tqCounter) Purge(queueName string) error { | |
| 67 return t.c.Purge.up(t.tq.Purge(queueName)) | |
| 68 } | |
| 69 | |
| 70 func (t *tqCounter) QueueStats(queueNames []string) ([]tq.Statistics, error) { | |
| 71 ret, err := t.tq.QueueStats(queueNames) | |
| 72 return ret, t.c.QueueStats.up(err) | |
| 73 } | |
| 74 | |
| 75 // FilterTQ installs a counter TaskQueue filter in the context. | |
| 76 func FilterTQ(c context.Context) (context.Context, *TQCounter) { | |
| 77 state := &TQCounter{} | |
| 78 return tq.AddFilters(c, func(ic context.Context, tq tq.Interface) tq.Int
erface { | |
| 79 return &tqCounter{state, tq} | |
| 80 }), state | |
| 81 } | |
| OLD | NEW |
|---|
Chromium Code Reviews
