Methods for stopping default consumer and producer

crackcomm · Mar 20, 2017 · 6ee4382 · 6ee4382
1 parent f2c4cb1
commit 6ee4382
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/consumer/default.go b/consumer/default.go
@@ -26,3 +26,8 @@ func Register(topic, channel string, maxInFlight int, fnc Handler) error {
 func Start(debug bool) {
 	DefaultConsumer.Start(debug)
 }
+
+// Stop - Look for Consumer.Stop() (on default consumer)
+func Stop() {
+	DefaultConsumer.Stop()
+}
diff --git a/producer/default.go b/producer/default.go
@@ -50,3 +50,8 @@ func Connect(addr string) error {
 func ConnectConfig(addr string, config *nsq.Config) error {
 	return DefaultProducer.ConnectConfig(addr, config)
 }
+
+// Stop initiates a graceful stop of the Producer (permanent)
+func Stop() {
+	DefaultProducer.Stop()
+}