chore(queue): Shutdown service once. (#610)

Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
This commit is contained in:
Bo-Yi Wu
2021-07-24 10:17:42 +08:00
committed by GitHub
parent d9947ea44d
commit 6ebbbe5026
3 changed files with 28 additions and 2 deletions

View File

@@ -76,3 +76,23 @@ func TestCustomFuncAndWait(t *testing.T) {
q.Wait()
// you will see the execute time > 1000ms
}
func TestShutDonwPanic(t *testing.T) {
w := NewWorker(
WithRunFunc(func(msg queue.QueuedMessage) error {
logx.LogAccess.Infof("get message: %s", msg.Bytes())
time.Sleep(100 * time.Millisecond)
return nil
}),
)
q, err := queue.NewQueue(
queue.WithWorker(w),
queue.WithWorkerCount(2),
)
assert.NoError(t, err)
q.Start()
q.Shutdown()
// check shutdown once
q.Shutdown()
q.Wait()
}