| Index: service/taskqueue/taskqueue.go
|
| diff --git a/service/taskqueue/taskqueue.go b/service/taskqueue/taskqueue.go
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..97ab2ffe60d506ccac863e557b157855fb930533
|
| --- /dev/null
|
| +++ b/service/taskqueue/taskqueue.go
|
| @@ -0,0 +1,82 @@
|
| +// Copyright 2015 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +package taskqueue
|
| +
|
| +import (
|
| + "github.com/luci/luci-go/common/errors"
|
| + "golang.org/x/net/context"
|
| +)
|
| +
|
| +type taskqueueImpl struct{ RawInterface }
|
| +
|
| +func (t *taskqueueImpl) NewTask(path string) *Task {
|
| + return &Task{Path: path}
|
| +}
|
| +
|
| +func (t *taskqueueImpl) Add(task *Task, queueName string) error {
|
| + return errors.SingleError(t.AddMulti([]*Task{task}, queueName))
|
| +}
|
| +
|
| +func (t *taskqueueImpl) Delete(task *Task, queueName string) error {
|
| + return errors.SingleError(t.DeleteMulti([]*Task{task}, queueName))
|
| +}
|
| +
|
| +func (t *taskqueueImpl) AddMulti(tasks []*Task, queueName string) error {
|
| + lme := errors.LazyMultiError{Size: len(tasks)}
|
| + i := 0
|
| + err := t.RawInterface.AddMulti(tasks, queueName, func(t *Task, err error) {
|
| + if !lme.Assign(i, err) {
|
| + *tasks[i] = *t
|
| + }
|
| + i++
|
| + })
|
| + if err == nil {
|
| + err = lme.Get()
|
| + }
|
| + return err
|
| +}
|
| +
|
| +func (t *taskqueueImpl) DeleteMulti(tasks []*Task, queueName string) error {
|
| + lme := errors.LazyMultiError{Size: len(tasks)}
|
| + i := 0
|
| + err := t.RawInterface.DeleteMulti(tasks, queueName, func(err error) {
|
| + lme.Assign(i, err)
|
| + i++
|
| + })
|
| + if err == nil {
|
| + err = lme.Get()
|
| + }
|
| + return err
|
| +}
|
| +
|
| +func (t *taskqueueImpl) Purge(queueName string) error {
|
| + return t.RawInterface.Purge(queueName)
|
| +}
|
| +
|
| +func (t *taskqueueImpl) Stats(queueNames ...string) ([]Statistics, error) {
|
| + ret := make([]Statistics, len(queueNames))
|
| + lme := errors.LazyMultiError{Size: len(queueNames)}
|
| + i := 0
|
| + err := t.RawInterface.Stats(queueNames, func(s *Statistics, err error) {
|
| + if !lme.Assign(i, err) {
|
| + ret[i] = *s
|
| + }
|
| + i++
|
| + })
|
| + if err == nil {
|
| + err = lme.Get()
|
| + }
|
| + return ret, err
|
| +}
|
| +
|
| +func (t *taskqueueImpl) Raw() RawInterface {
|
| + return t.RawInterface
|
| +}
|
| +
|
| +var _ Interface = (*taskqueueImpl)(nil)
|
| +
|
| +func Get(c context.Context) Interface {
|
| + return &taskqueueImpl{GetRaw(c)}
|
| +}
|
|
|