21 lines
392 B
Go
21 lines
392 B
Go
package middlewares
|
|
|
|
import (
|
|
"github.com/gofiber/contrib/websocket"
|
|
"github.com/gofiber/fiber/v2"
|
|
)
|
|
|
|
type WSMiddleware struct{}
|
|
|
|
func NewWSMiddleware() *WSMiddleware {
|
|
return &WSMiddleware{}
|
|
}
|
|
|
|
func (wsMiddleware *WSMiddleware) Upgrade(ctx *fiber.Ctx) error {
|
|
if websocket.IsWebSocketUpgrade(ctx) {
|
|
ctx.Locals("allowed", true)
|
|
return ctx.Next()
|
|
}
|
|
return fiber.ErrUpgradeRequired
|
|
}
|