Backport Connections method to GetChannel

Signed-off-by: Lorenzo <lorenzo.donini90@gmail.com>

Author: lorenzodonini

ws/server.go

@@ -117,6 +117,10 @@ type Server interface {
 	// Addr gives the address on which the server is listening, useful if, for
 	// example, the port is system-defined (set to 0).
 	Addr() *net.TCPAddr
+	// GetChannel retrieves an active Channel connection by its unique identifier.
+	// If a connection with the given ID exists, it returns the corresponding webSocket instance.
+	// If no connection is found with the specified ID, it returns nil and a false flag.
+	GetChannel(websocketId string) (Channel, bool)
 }
 
 // Default implementation of a Websocket server.
@@ -304,6 +308,13 @@ func (s *server) StopConnection(id string, closeError websocket.CloseError) erro
 	return w.Close(closeError)
 }
 
+func (s *server) GetChannel(websocketId string) (Channel, bool) {
+	s.connMutex.RLock()
+	defer s.connMutex.RUnlock()
+	c, ok := s.connections[websocketId]
+	return c, ok
+}
+
 func (s *server) stopConnections() {
 	s.connMutex.RLock()
 	defer s.connMutex.RUnlock()

ws/websocket_test.go

@@ -486,6 +486,7 @@ func (s *WebSocketSuite) TestServerStopConnection() {
 		Code: websocket.CloseGoingAway,
 		Text: "CloseClientConnection",
 	}
+	wsID := "testws"
 	s.server = newWebsocketServer(s.T(), nil)
 	s.server.SetNewClientHandler(func(ws Channel) {
 		triggerC <- struct{}{}
@@ -508,14 +509,25 @@ func (s *WebSocketSuite) TestServerStopConnection() {
 	// Start server
 	go s.server.Start(serverPort, serverPath)
 	time.Sleep(100 * time.Millisecond)
+	var c Channel
+	var ok bool
+	c, ok = s.server.GetChannel(wsID)
+	s.False(ok)
+	s.Nil(c)
 	// Connect client
 	host := fmt.Sprintf("localhost:%v", serverPort)
 	u := url.URL{Scheme: "ws", Host: host, Path: testPath}
 	err := s.client.Start(u.String())
 	s.NoError(err)
 	// Wait for client to connect
-	_, ok := <-triggerC
+	_, ok = <-triggerC
+	s.True(ok)
+	// Verify channel
+	c, ok = s.server.GetChannel(wsID)
 	s.True(ok)
+	s.NotNil(c)
+	s.Equal(wsID, c.ID())
+	s.True(c.IsConnected())
 	// Close connection and wait for client to be closed
 	err = s.server.StopConnection(path.Base(testPath), closeError)
 	s.NoError(err)