Experiment with Authentication and Websockets

middleware/ws.go

@@ -0,0 +1,20 @@
+package middleware
+
+import (
+	"github.com/gofiber/contrib/websocket"
+	"github.com/gofiber/fiber/v2"
+)
+
+type WsMiddleware struct {
+}
+
+func NewWsMiddleware() WsMiddleware {
+	return WsMiddleware{}
+}
+
+func (wm WsMiddleware) IsWebscoketUpgrade(ctx *fiber.Ctx) error {
+	if websocket.IsWebSocketUpgrade(ctx) {
+		return ctx.Next()
+	}
+	return ctx.SendStatus(fiber.StatusUpgradeRequired)
+}