package controllers

import (
	"fmt"

	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{}

func WsServer(c *gin.Context) {
	cn, err := upgrader.Upgrade(c.Writer, c.Request, nil)
	defer cn.Close()

	if err != nil {
		// 响应错误
	}

	for {
		_, message, err := cn.ReadMessage()
		if err != nil {
			fmt.Printf("read msg error:%s\n", err.Error())
		}

		fmt.Printf("read msg: %s\v\n", string(message))
	}
}