eta_menu_sync/init_serve/ws.go

package init_serve

import (
	"fmt"
	socketio "github.com/googollee/go-socket.io"
	"eta/eta_menu_sync/global"
)

func InitWs() {
	// 创建一个 Socket.IO 服务器实例
	server := socketio.NewServer(nil)

	// 处理 Socket.IO 连接
	server.OnConnect("/", func(s socketio.Conn) error {
		s.SetContext("")
		fmt.Println("connected:", s.ID())
		return nil
	})

	// 处理 Socket.IO 断开连接
	server.OnDisconnect("/", func(s socketio.Conn, reason string) {
		fmt.Println("disconnected:", s.ID(), reason)
	})

	// 处理 Socket.IO 消息
	server.OnEvent("/", "message", func(s socketio.Conn, msg string) {
		fmt.Println("message:", msg)
		s.Emit("response", "hello")
	})

	global.Ws = server
}