89 lines
1.8 KiB
Go
89 lines
1.8 KiB
Go
package websocket
|
||
|
||
//
|
||
//import (
|
||
// "log"
|
||
// "net/http"
|
||
// "sync"
|
||
//
|
||
// "github.com/gogf/gf/v2/net/ghttp"
|
||
// "github.com/gorilla/websocket"
|
||
//)
|
||
//
|
||
//// 定义 WebSocket 升级器
|
||
//var upgrader = websocket.Upgrader{
|
||
// CheckOrigin: func(r *http.Request) bool {
|
||
// return true
|
||
// },
|
||
//}
|
||
//
|
||
//// 存储客户端连接
|
||
//var (
|
||
// clients = make(map[*websocket.Conn]bool)
|
||
// clientsMu sync.Mutex // 保护 clients map 的互斥锁
|
||
// broadcast = make(chan string, 256) // 广播信息通道，带缓冲
|
||
//)
|
||
//
|
||
//// 初始化广播处理（只启动一次）
|
||
//var once sync.Once
|
||
//
|
||
//// HandlerConnection 处理websocket连接
|
||
//func HandlerConnectionV1(r *ghttp.Request) {
|
||
// // 启动广播处理 goroutine（只启动一次）
|
||
// once.Do(func() {
|
||
// go handleBroadcast()
|
||
// })
|
||
//
|
||
// ws, err := upgrader.Upgrade(r.Response.Writer, r.Request, nil)
|
||
// if err != nil {
|
||
// log.Printf("upgrade error: %v", err)
|
||
// return
|
||
// }
|
||
// defer func() {
|
||
// clientsMu.Lock()
|
||
// delete(clients, ws)
|
||
// clientsMu.Unlock()
|
||
// ws.Close()
|
||
// }()
|
||
//
|
||
// // 注册新客户端
|
||
// clientsMu.Lock()
|
||
// clients[ws] = true
|
||
// clientsMu.Unlock()
|
||
//
|
||
// for {
|
||
// var message string
|
||
// err := ws.ReadJSON(&message)
|
||
// if err != nil {
|
||
// log.Printf("read error: %v", err)
|
||
// clientsMu.Lock()
|
||
// delete(clients, ws)
|
||
// clientsMu.Unlock()
|
||
// break
|
||
// }
|
||
// // 将消息发送到广播通道
|
||
// select {
|
||
// case broadcast <- message:
|
||
// default:
|
||
// log.Printf("broadcast channel full, dropping message")
|
||
// }
|
||
// }
|
||
//}
|
||
//
|
||
//// 将广播通道给所有的用户
|
||
//func handleBroadcast() {
|
||
// for {
|
||
// message := <-broadcast
|
||
// clientsMu.Lock()
|
||
// for client := range clients {
|
||
// err := client.WriteJSON(message)
|
||
// if err != nil {
|
||
// log.Printf("write error: %v", err)
|
||
// client.Close()
|
||
// delete(clients, client)
|
||
// }
|
||
// }
|
||
// clientsMu.Unlock()
|
||
// }
|
||
//}
|