Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(3)

Unified Diff: filter/featureBreaker/tq.go

Issue 2512093002: Add support for Pull Queues to prod implementation. (Closed)
Patch Set: Created 4 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: filter/featureBreaker/tq.go
diff --git a/filter/featureBreaker/tq.go b/filter/featureBreaker/tq.go
index c5c84482c67e91159d8ae844530c8c16c2004df1..9fdc79b013f35e5093992cc070a82a33ffc66007 100644
--- a/filter/featureBreaker/tq.go
+++ b/filter/featureBreaker/tq.go
@@ -26,6 +26,32 @@ func (t *tqState) DeleteMulti(tasks []*tq.Task, queueName string, cb tq.RawCB) e
return t.run(func() error { return t.tq.DeleteMulti(tasks, queueName, cb) })
}
+func (t *tqState) Lease(maxTasks int, queueName string, leaseTime int) (tasks []*tq.Task, err error) {
+ err = t.run(func() (err error) {
+ tasks, err = t.tq.Lease(maxTasks, queueName, leaseTime)
+ return
+ })
+ if err != nil {
+ tasks = nil
+ }
+ return
+}
+
+func (t *tqState) LeaseByTag(maxTasks int, queueName string, leaseTime int, tag string) (tasks []*tq.Task, err error) {
+ err = t.run(func() (err error) {
+ tasks, err = t.tq.LeaseByTag(maxTasks, queueName, leaseTime, tag)
+ return
+ })
+ if err != nil {
+ tasks = nil
+ }
+ return
+}
+
+func (t *tqState) ModifyLease(task *tq.Task, queueName string, leaseTime int) error {
+ return t.run(func() error { return t.tq.ModifyLease(task, queueName, leaseTime) })
+}
+
func (t *tqState) Purge(queueName string) error {
return t.run(func() error { return t.tq.Purge(queueName) })
}
« no previous file with comments | « filter/count/tq.go ('k') | impl/dummy/dummy.go » ('j') | service/taskqueue/interface.go » ('J')

Powered by Google App Engine
This is Rietveld 408576698