marco/mchess-server
1
0
Fork 0
Code Issues 5 Pull Requests Packages Projects Releases Wiki Activity

Rate limit websocket connection

Browse Source
This commit is contained in:
Marco 2024-05-12 15:42:40 +02:00
parent 4207a0ed72
commit 97ad45e505
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
api/websocket/connection.go → api/handler/websocket.go
View File

@ -1,4 +1,4 @@
package websocket package handler
import ( import (
"context" "context"
@ -21,6 +21,8 @@ var upgrader = gorillaws.Upgrader{
} }
func RegisterWebSocketConnection(c *gin.Context) { func RegisterWebSocketConnection(c *gin.Context) {
limiter.Take()
log.Println(c.Request) log.Println(c.Request)
conn, err := upgrader.Upgrade(c.Writer, c.Request, nil) conn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
if err != nil { if err != nil {
@ -31,6 +33,8 @@ func RegisterWebSocketConnection(c *gin.Context) {
} }
func waitForAndHandlePlayerID(ctx context.Context, conn *gorillaws.Conn) { func waitForAndHandlePlayerID(ctx context.Context, conn *gorillaws.Conn) {
limiter.Take()
msgType, msg, err := conn.ReadMessage() msgType, msg, err := conn.ReadMessage()
if err != nil { if err != nil {
errorMessage := fmt.Sprintf("Reading from websocket connection did not work: %s", err) errorMessage := fmt.Sprintf("Reading from websocket connection did not work: %s", err)
@ -72,5 +76,6 @@ func waitForAndHandlePlayerID(ctx context.Context, conn *gorillaws.Conn) {
} }
func ConnectWsForGame(c *gin.Context) { func ConnectWsForGame(c *gin.Context) {
limiter.Take()
} }

5
main.go
View File

@ -4,7 +4,6 @@ import (
"flag" "flag"
"log" "log"
"mchess_server/api/handler" "mchess_server/api/handler"
"mchess_server/api/websocket"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
) )
@ -26,10 +25,10 @@ func main() {
router := gin.Default() router := gin.Default()
router.GET("/api/hostPrivate", handler.HostGameHandler) router.GET("/api/hostPrivate", handler.HostGameHandler)
router.POST("/api/joinPrivate", handler.JoinPrivateGame) router.POST("/api/joinPrivate", handler.JoinPrivateGame)
router.GET("/api/ws", websocket.RegisterWebSocketConnection) router.GET("/api/ws", handler.RegisterWebSocketConnection)
router.GET("/api/getLobbyForPassphrase/:phrase", handler.GetLobbyForPassphraseHandler) router.GET("/api/getLobbyForPassphrase/:phrase", handler.GetLobbyForPassphraseHandler)
router.GET("/api/registerWsForGame/:id", websocket.ConnectWsForGame) router.GET("/api/registerWsForGame/:id", handler.ConnectWsForGame)
router.POST("/api/joinGame/:id", handler.JoinGameHandler) router.POST("/api/joinGame/:id", handler.JoinGameHandler)
if debugMode { if debugMode {