2 Commits

3 changed files with 6 additions and 6 deletions

View File

@@ -40,7 +40,7 @@ app.Use(async (context, next) =>
{ {
if (context.WebSockets.IsWebSocketRequest && context.Request.Path.StartsWithSegments("/ws")) if (context.WebSockets.IsWebSocketRequest && context.Request.Path.StartsWithSegments("/ws"))
{ {
LogToFile($"WebSocket request received for path: {context.Request.Path}"); // LogToFile($"WebSocket request received for path: {context.Request.Path}");
try try
{ {
@@ -65,7 +65,7 @@ app.Use(async (context, next) =>
} }
catch (Exception ex) catch (Exception ex)
{ {
LogToFile($"WebSocket proxy error: {ex.Message}"); //LogToFile($"WebSocket proxy error: {ex.Message}");
context.Response.StatusCode = (int)HttpStatusCode.BadGateway; context.Response.StatusCode = (int)HttpStatusCode.BadGateway;
await context.Response.WriteAsync($"WebSocket proxy error: {ex.Message}"); await context.Response.WriteAsync($"WebSocket proxy error: {ex.Message}");
} }

View File

@@ -11,7 +11,7 @@ var (
broadcaster = make(chan logging.Message) broadcaster = make(chan logging.Message)
) )
func RegisterSocketRoutes(r *gin.Engine) { func RegisterSocketRoutes(r *gin.Engine, base_url string) {
// Initialize all channels // Initialize all channels
InitializeChannels() InitializeChannels()
@@ -23,11 +23,11 @@ func RegisterSocketRoutes(r *gin.Engine) {
go StartBroadcasting(broadcaster, channels) go StartBroadcasting(broadcaster, channels)
// WebSocket route // WebSocket route
r.GET("/ws", func(c *gin.Context) { r.GET(base_url+"/ws", func(c *gin.Context) {
SocketHandler(c, channels) SocketHandler(c, channels)
}) })
r.GET("/ws/clients", AdminAuthMiddleware(), handleGetClients) r.GET(base_url+"/ws/clients", AdminAuthMiddleware(), handleGetClients)
} }
func handleGetClients(c *gin.Context) { func handleGetClients(c *gin.Context) {

View File

@@ -114,7 +114,7 @@ func main() {
}) })
//logging.RegisterLoggerRoutes(r, basePath) //logging.RegisterLoggerRoutes(r, basePath)
websocket.RegisterSocketRoutes(r) websocket.RegisterSocketRoutes(r, basePath)
config.RegisterConfigRoutes(r, basePath) config.RegisterConfigRoutes(r, basePath)
r.Any(basePath+"/api", errorApiLoc) r.Any(basePath+"/api", errorApiLoc)