package main
import (
	"fmt"
	"net/http"
	"github.com/gin-gonic/gin"
	"github.com/olahol/melody"
)
func main() {
	r := gin.Default()
	// 創(chuàng)建Melody實(shí)例
	m := melody.New()
	// 注冊WebSocket處理函數(shù)
	r.GET("/ws", func(c *gin.Context) {
		err := m.HandleRequest(c.Writer, c.Request)
		if err != nil {
			fmt.Println(err)
		}
	})
	// 處理WebSocket連接
	m.HandleConnect(func(s *melody.Session) {
		fmt.Println("WebSocket connected")
	})
	// 處理WebSocket消息
	m.HandleMessage(func(s *melody.Session, msg []byte) {
		fmt.Println("Received message:", string(msg))
		// 發(fā)送消息給所有連接的客戶端
		m.Broadcast(msg)
	})
	// 處理WebSocket斷開連接
	m.HandleDisconnect(func(s *melody.Session) {
		fmt.Println("WebSocket disconnected")
	})
	// 啟動HTTP服務(wù)器
	fmt.Println("WebSocket server listening on :8080")
	http.ListenAndServe(":8080", r)
}

package main
import (
	"github.com/gin-gonic/gin"
	"github.com/olahol/melody"
	"net/http"
)
func main() {
	r := gin.Default()
	m := melody.New()
	r.GET("/", func(c *gin.Context) {
		http.ServeFile(c.Writer, c.Request, "index.html")
	})
	r.GET("/ws", func(c *gin.Context) {
		m.HandleRequest(c.Writer, c.Request)
	})
	m.HandleMessage(func(s *melody.Session, msg []byte) {
		m.Broadcast(msg)
	})
	r.Run(":5000")
}

go get github.com/gin-gonic/gin
go get github.com/olahol/melody