Replace secret workqueue
This commit is contained in:
parent
7c635a8c83
commit
f28142ae8e
4 changed files with 86 additions and 136 deletions
|
|
@ -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)
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
|||
|
|
@ -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",
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue