close all function

topfreegames · May 10, 2019 · 812e4fe · 812e4fe
1 parent b92161d
commit 812e4fe
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/app.go b/app.go
@@ -439,6 +439,7 @@ func Start() {
 
 	logger.Log.Warn("server is stopping...")
 
+	session.CloseAll()
 	shutdownModules()
 	shutdownComponents()
 }

diff --git a/session/session.go b/session/session.go
@@ -188,6 +188,18 @@ func OnSessionClose(f func(s *Session)) {
 	SessionCloseCallbacks = append(SessionCloseCallbacks, f)
 }
 
+// CloseAll calls Close on all sessions
+func CloseAll() {
+	// TODO: check if frontend?
+	logger.Log.Debug("closing all sessions, %d sessions", SessionCount)
+	sessionsByID.Range(func(_, value interface{}) bool {
+		s := value.(*Session)
+		s.Close()
+		return true
+	})
+	logger.Log.Debug("finished closing sessions")
+}
+
 func (s *Session) updateEncodedData() error {
 	var b []byte
 	b, err := json.Marshal(s.data)