From 2b1f0890db62890495834895812b27757300af9a Mon Sep 17 00:00:00 2001
From: Vladimir Barsukov <barsukov.vv@mail.ru>
Date: Fri, 17 Nov 2023 11:22:20 +0200
Subject: [PATCH] zgo

---
 zq/zq.go       | 4 +++-
 zquit/zquit.go | 4 ++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/zq/zq.go b/zq/zq.go
index 0bd47e4..477bf90 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 08a3d20..412b02e 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
+}
-- 
GitLab