fix oauth SSO panic on close

gravitl · Apr 30, 2024 · 3db596e · 3db596e
1 parent c5a5f9b
commit 3db596e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/functions/register_sso.go b/functions/register_sso.go
@@ -88,11 +88,11 @@ func handeServerSSORegisterConn(reqMsg *models.RegisterMsg, apiURI string, conn
 		return err
 	}
 	done := make(chan struct{})
-	defer close(done)
 	interrupt := make(chan os.Signal, 1)
 	signal.Notify(interrupt, os.Interrupt)
 
 	go func() {
+		defer close(done)
 		for {
 			msgType, msg, err := conn.ReadMessage()
 			if err != nil {
@@ -123,6 +123,7 @@ func handeServerSSORegisterConn(reqMsg *models.RegisterMsg, apiURI string, conn
 				}
 				handleRegisterResponse(&response)
 			}
+
 		}
 	}()