chore: support custom parameter in queue (#608)

* chore: support custom parameter in queue Signed-off-by: Bo-Yi Wu <appleboy.tw@gmail.com>
2021-07-24 01:29:47 +08:00
parent 66923789ff
commit ce6e87639a
5 changed files with 66 additions and 10 deletions
--- a/router/server_test.go
+++ b/router/server_test.go
@@ -37,7 +37,10 @@ func TestMain(m *testing.M) {
 	}

 	w = simple.NewWorker()
-	q = queue.NewQueue(w, 4)
+	q, _ = queue.NewQueue(
+		queue.WithWorker(w),
+		queue.WithWorkerCount(4),
+	)
 	q.Start()
 	defer func() {
 		q.Shutdown()