diff --git a/zq/zq.go b/zq/zq.go index 0bd47e4d8bff9cc0dc8a5dfd26145502e295f529..477bf903fd08c90287c6a2fc14bcf61bcf0caee8 100644 --- a/zq/zq.go +++ b/zq/zq.go @@ -62,12 +62,14 @@ func (q *ZQ) RemoveWorker() { q.AddWorker(-1) } -func (q *ZQ) AddJob(a any) { +func (q *ZQ) AddJob(a any) *ZQ { q.JobsWg.Inc() go func() { q.jobs <- a }() + + return q } func (q *ZQ) Wait() { diff --git a/zquit/zquit.go b/zquit/zquit.go index 08a3d2092508ce54769b9c925c71fc658437f621..412b02ea43e0a7a176196c309fdfd1395ce68285 100644 --- a/zquit/zquit.go +++ b/zquit/zquit.go @@ -72,3 +72,7 @@ func (q *ZQuit) ZQuitMiddleware(c *gin.Context) { c.Next() } + +func (q *ZQuit) Shutdown() { + q.isQuit = true +}