KoCoder
1e41dd833f
Some checks failed
build-docker-imge / Build the docker container (push) Failing after 2m24s
21 lines
380 B
Go
21 lines
380 B
Go
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)
|
|
}
|