Sync mutex when cleanup

kfsoftware · Feb 15, 2023 · 4297d03 · 4297d03
1 parent 43fbf77
commit 4297d03
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cmd/server/server.go b/cmd/server/server.go
@@ -44,6 +44,8 @@ type SessionRegistry struct {
 }
 
 func (s *Session) cleanup() {
+	s.Lock()
+	defer s.Unlock()
 	defer func() {
 		s.cleanupDone = true
 	}()
@@ -341,6 +343,7 @@ func NewServerCmd() *cobra.Command {
 }
 
 type Session struct {
+	sync.RWMutex
 	cleanupDone bool
 	SNI         string `json:"sni"`
 	//RemoteAddr string `json:"remoteAddr"`