added system shutdown stream error.

concentrateyou · May 29, 2018 · 04f440f · 04f440f
1 parent 0beedee
commit 04f440f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/errors/error.go b/errors/error.go
@@ -45,6 +45,9 @@ var (
 	// ErrResourceConstraint represents 'resource-constraint' stream error.
 	ErrResourceConstraint = newStreamError("resource-constraint")
 
+	// ErrSystemShutdown represents 'system-shutdown' stream error.
+	ErrSystemShutdown = newStreamError("system-shutdown")
+
 	// ErrInternalServerError represents 'internal-server-error' stream error.
 	ErrInternalServerError = newStreamError("internal-server-error")
 )

diff --git a/router/router.go b/router/router.go
@@ -11,6 +11,7 @@ import (
 	"sync"
 	"sync/atomic"
 
+	"github.com/ortuman/jackal/errors"
 	"github.com/ortuman/jackal/log"
 	"github.com/ortuman/jackal/storage"
 	"github.com/ortuman/jackal/stream"
@@ -102,7 +103,7 @@ func Shutdown() {
 		for _, stm := range inst.streams {
 			wg.Add(1)
 			go func(s stream.C2S) {
-				s.Disconnect(nil)
+				s.Disconnect(streamerror.ErrSystemShutdown)
 				wg.Done()
 			}(stm)
 		}