Replace secret workqueue

This commit is contained in:
Manuel de Brito Fontes 2017-04-09 13:52:10 -03:00
parent 7c635a8c83
commit f28142ae8e
4 changed files with 86 additions and 136 deletions

View file

@ -82,15 +82,17 @@ func (t *Queue) worker() {
close(t.workerDone)
return
}
defer t.queue.Done(key)
glog.V(3).Infof("syncing %v", key)
if err := t.sync(key); err != nil {
glog.Warningf("requeuing %v, err %v", key, err)
t.queue.AddRateLimited(key)
} else {
err := t.sync(key)
if err == nil {
t.queue.Forget(key)
return
}
t.queue.Done(key)
glog.Warningf("requeuing %v, err %v", key, err)
t.queue.AddRateLimited(key)
}
}

View file

@ -65,7 +65,7 @@ func TestEnqueueSuccess(t *testing.T) {
q := NewCustomTaskQueue(mockSynFn, mockKeyFn)
stopCh := make(chan struct{})
// run queue
go q.Run(10*time.Second, stopCh)
go q.Run(5*time.Second, stopCh)
// mock object whichi will be enqueue
mo := mockEnqueueObj{
k: "testKey",
@ -88,7 +88,7 @@ func TestEnqueueFailed(t *testing.T) {
q := NewCustomTaskQueue(mockSynFn, mockKeyFn)
stopCh := make(chan struct{})
// run queue
go q.Run(10*time.Second, stopCh)
go q.Run(5*time.Second, stopCh)
// mock object whichi will be enqueue
mo := mockEnqueueObj{
k: "testKey",
@ -114,7 +114,7 @@ func TestEnqueueKeyError(t *testing.T) {
q := NewCustomTaskQueue(mockSynFn, mockErrorKeyFn)
stopCh := make(chan struct{})
// run queue
go q.Run(10*time.Second, stopCh)
go q.Run(5*time.Second, stopCh)
// mock object whichi will be enqueue
mo := mockEnqueueObj{
k: "testKey",