qinglong-go/api/ws/ws.go

package ws

import (
	"github.com/gin-gonic/gin"
	"github.com/gobwas/ws"
	"github.com/gobwas/ws/wsutil"
	"github.com/huoxue1/qinglong-go/utils/res"
)

func Api(group *gin.RouterGroup) {
	group.GET("/info", info())
	group.GET("/:id/:name/websocket", wsHandle())
}

func info() gin.HandlerFunc {
	return func(ctx *gin.Context) {
		ctx.JSON(200, gin.H{"websocket": true, "origins": []string{"*:*"}, "cookie_needed": false, "entropy": int64(3563341155)})
	}
}

func wsHandle() gin.HandlerFunc {
	return func(ctx *gin.Context) {
		conn, _, _, err := ws.UpgradeHTTP(ctx.Request, ctx.Writer)
		if err != nil {
			ctx.JSON(502, res.Err(502, err))
			return
		}
		writer := wsutil.NewWriter(conn, ws.StateServerSide, ws.OpText)
		writer.Write([]byte("pong"))
		writer.Flush()
	}
}