fix(SSE): only initialize http.Server when not set (#229)

cryo-zd · web-flow · commit f0a648b91d85 · 2025-05-01T11:04:19.000+03:00
diff --git a/server/sse.go b/server/sse.go
@@ -190,12 +190,21 @@ func NewTestServer(server *MCPServer, opts ...SSEOption) *httptest.Server {
 // It sets up HTTP handlers for SSE and message endpoints.
 func (s *SSEServer) Start(addr string) error {
 	s.mu.Lock()
-	s.srv = &http.Server{
-		Addr:    addr,
-		Handler: s,
-	}
-	s.mu.Unlock()
+	defer s.mu.Unlock()
 
+	if s.srv == nil {
+		s.srv = &http.Server{
+			Addr:		addr,
+			Handler:	s,
+		}
+	} else {
+		if s.srv.Addr == "" {
+			s.srv.Addr = addr
+		} else if s.srv.Addr != addr {
+			return fmt.Errorf("conflicting listen address: WithHTTPServer(%q) vs Start(%q)", s.srv.Addr, addr)
+		}
+	}
+	
 	return s.srv.ListenAndServe()
 }
 
