Add configuration for trusted proxies (#141)

2022-10-05 08:11:07 +08:00
parent b20bf6deab
commit e98d249cf7
5 changed files with 10 additions and 4 deletions
--- a/router/router_server_ws.go
+++ b/router/router_server_ws.go
@@ -32,7 +32,7 @@ func getServerWebsocket(c *gin.Context) {
 	ctx, cancel := context.WithCancel(c.Request.Context())
 	defer cancel()

-	handler, err := websocket.GetHandler(s, c.Writer, c.Request)
+	handler, err := websocket.GetHandler(s, c.Writer, c.Request, c)
 	if err != nil {
 		NewServerError(err, s).Abort(c)
 		return