chore(x): expose a WrapConn to wrap a Gorilla conn

Jigsaw-Code · Jan 30, 2025 · 72643a6 · 72643a6
1 parent 5f6d095
commit 72643a6
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/x/websocket/endpoint.go b/x/websocket/endpoint.go
@@ -106,6 +106,12 @@ func newEndpoint[ConnType net.Conn](urlStr string, sd transport.StreamDialer, ws
 	}, nil
 }
 
+// WrapConn wraps a [websocket.Conn] from the Gorilla library into a
+// [transport.StreamConn].
+func WrapConn(wsConn *websocket.Conn) transport.StreamConn {
+	return &gorillaConn{wsConn: wsConn}
+}
+
 type gorillaConn struct {
 	wsConn        *websocket.Conn
 	writeErr      error