Merge pull request #2449 from stakwork/try-fix-ws

ws id fallbacks
stakwork · Jan 15, 2025 · 623e610 · 623e610
2 parents 9f1db07 + 0319b90
commit 623e610
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/websocket/websocket.go b/websocket/websocket.go
@@ -7,6 +7,7 @@ import (
 
 	"github.com/gorilla/websocket"
 	"github.com/stakwork/sphinx-tribes/config"
+	"github.com/stakwork/sphinx-tribes/utils"
 )
 
 var WebsocketPool = NewPool()
@@ -35,8 +36,14 @@ func Upgrade(w http.ResponseWriter, r *http.Request) (*websocket.Conn, error) {
 }
 
 func ServeWs(pool *Pool, w http.ResponseWriter, r *http.Request) { // get url query params
+
+	// get url query params
 	queryParams := r.URL.Query()
 	uniqueId := queryParams.Get("uniqueId")
+	if uniqueId == "" || uniqueId == "null" || uniqueId == "undefined" {
+		// or random
+		uniqueId = utils.GetRandomToken(40)
+	}
 
 	conn, err := Upgrade(w, r)
 	if err != nil {