69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
package handlers
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/M1ngdaXie/realtime-collab/internal/hub"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/google/uuid"
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
// loadTestUpgrader allows any origin for load testing
|
|
var loadTestUpgrader = websocket.Upgrader{
|
|
ReadBufferSize: 1024,
|
|
WriteBufferSize: 1024,
|
|
CheckOrigin: func(r *http.Request) bool {
|
|
return true // Allow all origins for load testing
|
|
},
|
|
}
|
|
|
|
// HandleWebSocketLoadTest is a backdoor WebSocket handler that skips all authentication.
|
|
// WARNING: Only use this for local load testing! Do not expose in production.
|
|
//
|
|
// Usage: ws://localhost:8080/ws/loadtest/:roomId
|
|
// Optional query param: ?user=customUserName
|
|
func (wsh *WebSocketHandler) HandleWebSocketLoadTest(c *gin.Context) {
|
|
roomID := c.Param("roomId")
|
|
if roomID == "" {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "roomId is required"})
|
|
return
|
|
}
|
|
|
|
// Generate or use provided user identifier
|
|
userName := c.Query("user")
|
|
if userName == "" {
|
|
userName = "LoadTestUser-" + uuid.New().String()[:8]
|
|
}
|
|
|
|
// Upgrade connection - NO AUTH CHECK
|
|
conn, err := loadTestUpgrader.Upgrade(c.Writer, c.Request, nil)
|
|
if err != nil {
|
|
log.Printf("Failed to upgrade connection: %v", err)
|
|
return
|
|
}
|
|
|
|
// Create client with full edit permissions (no userID, anonymous)
|
|
clientID := uuid.New().String()
|
|
client := hub.NewClient(
|
|
clientID,
|
|
nil, // userID - nil for anonymous
|
|
userName, // userName
|
|
nil, // userAvatar
|
|
"edit", // permission - full access for load testing
|
|
conn,
|
|
wsh.hub,
|
|
roomID,
|
|
)
|
|
|
|
// Register client
|
|
wsh.hub.Register <- client
|
|
|
|
// Start goroutines
|
|
go client.WritePump()
|
|
go client.ReadPump()
|
|
|
|
log.Printf("[LOADTEST] Client connected: %s (user: %s) to room: %s", clientID, userName, roomID)
|
|
}
|