chore(worker): Add disable consumer method

appleboy · appleboy · commit fb8b6a578fcb · 2022-04-01T16:04:44.000+08:00
Signed-off-by: Bo-Yi Wu &lt;appleboy.tw@gmail.com&gt;
diff --git a/_example/server-client/go.mod b/_example/server-client/go.mod
@@ -4,7 +4,7 @@ go 1.16
 
 require (
 	github.com/golang-queue/nsq v0.0.0-00010101000000-000000000000
-	github.com/golang-queue/queue v0.0.12-0.20220122071422-2c41650f045a
+	github.com/golang-queue/queue v0.0.13-0.20220330060848-d1a0d31ce747
 )
 
 replace github.com/golang-queue/nsq => ../../
diff --git a/_example/server-client/go.sum b/_example/server-client/go.sum
@@ -7,6 +7,8 @@ github.com/golang-queue/queue v0.0.8-0.20210905095503-cc99dff8fdc3 h1:ka4/BRgVnd
 github.com/golang-queue/queue v0.0.8-0.20210905095503-cc99dff8fdc3/go.mod h1:JS5tYJacahCjafcplU5idNLX2vkYioqh6wEDX5o9Nms=
 github.com/golang-queue/queue v0.0.12-0.20220122071422-2c41650f045a h1:olgbdu53u5A6Babr1w+CUO19UdZASYPPqqIBfdFWrWE=
 github.com/golang-queue/queue v0.0.12-0.20220122071422-2c41650f045a/go.mod h1:ku8iyjYffqYY6Duts+xl+QYfN3/KDK4MEvXMZUkHyio=
+github.com/golang-queue/queue v0.0.13-0.20220330060848-d1a0d31ce747 h1:uNTbCoWORAcna89KcKgP22WFGv5fsij05e70DCnLrUU=
+github.com/golang-queue/queue v0.0.13-0.20220330060848-d1a0d31ce747/go.mod h1:ku8iyjYffqYY6Duts+xl+QYfN3/KDK4MEvXMZUkHyio=
 github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4=
 github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
 github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
diff --git a/_example/server-client/server/main.go b/_example/server-client/server/main.go
@@ -30,6 +30,7 @@ func main() {
 		nsq.WithAddr("127.0.0.1:4150"),
 		nsq.WithTopic("example"),
 		nsq.WithChannel("foobar"),
+		nsq.WithDisableConsumer(),
 	)
 
 	// define the queue
diff --git a/nsq_test.go b/nsq_test.go
@@ -276,7 +276,7 @@ func TestHandleTimeout(t *testing.T) {
 			time.Sleep(200 * time.Millisecond)
 			return nil
 		}),
-		withDisableConsumer(),
+		WithDisableConsumer(),
 	)
 
 	err := w.handle(job)
@@ -293,7 +293,7 @@ func TestHandleTimeout(t *testing.T) {
 			time.Sleep(200 * time.Millisecond)
 			return nil
 		}),
-		withDisableConsumer(),
+		WithDisableConsumer(),
 	)
 
 	done := make(chan error)
@@ -317,7 +317,7 @@ func TestJobComplete(t *testing.T) {
 		WithRunFunc(func(ctx context.Context, m queue.QueuedMessage) error {
 			return errors.New("job completed")
 		}),
-		withDisableConsumer(),
+		WithDisableConsumer(),
 	)
 
 	err := w.handle(job)
diff --git a/options.go b/options.go
@@ -62,7 +62,8 @@ func WithLogger(l queue.Logger) Option {
 	}
 }
 
-func withDisableConsumer() Option {
+// WithDisableConsumer disable consumer
+func WithDisableConsumer() Option {
 	return func(w *options) {
 		w.disableConsumer = true
 	}

Original file line number	Diff line number	Diff line change
`@@ -4,7 +4,7 @@ go 1.16`
`4`	`4`
`5`	`5`	`require (`
`6`	`6`	`github.com/golang-queue/nsq v0.0.0-00010101000000-000000000000`
`7`		`- github.com/golang-queue/queue v0.0.12-0.20220122071422-2c41650f045a`
	`7`	`+ github.com/golang-queue/queue v0.0.13-0.20220330060848-d1a0d31ce747`
`8`	`8`	`)`
`9`	`9`
`10`	`10`	`replace github.com/golang-queue/nsq => ../../`
Original file line number	Diff line number	Diff line change
`@@ -30,6 +30,7 @@ func main() {`
`30`	`30`	`nsq.WithAddr("127.0.0.1:4150"),`
`31`	`31`	`nsq.WithTopic("example"),`
`32`	`32`	`nsq.WithChannel("foobar"),`
	`33`	`+ nsq.WithDisableConsumer(),`
`33`	`34`	`)`
`34`	`35`
`35`	`36`	`// define the queue`
Original file line number	Diff line number	Diff line change
`@@ -62,7 +62,8 @@ func WithLogger(l queue.Logger) Option {`
`62`	`62`	`}`
`63`	`63`	`}`
`64`	`64`
`65`		`-func withDisableConsumer() Option {`
	`65`	`+// WithDisableConsumer disable consumer`
	`66`	`+func WithDisableConsumer() Option {`
`66`	`67`	`return func(w *options) {`
`67`	`68`	`w.disableConsumer = true`
`68`	`69`	`}`