12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- package chat
- import (
- "context"
- "log"
- "net/http"
- "sync"
- "github.com/gogf/gf/v2/net/ghttp"
- "github.com/gorilla/websocket"
- v1 "cris/api/chat/v1"
- )
- var clients = make(map[*websocket.Conn]bool)
- var broadcast = make(chan string)
- var mu sync.Mutex
- type chat struct{}
- func New() *chat {
- return &chat{}
- }
- func (s *chat) Chat(ctx context.Context, r *ghttp.Request, req *v1.ChatReq) {
-
- upgrader := websocket.Upgrader{
- CheckOrigin: func(r *http.Request) bool {
- return true
- },
- ReadBufferSize: 1024,
- WriteBufferSize: 1024,
- }
-
- conn, err := upgrader.Upgrade(r.Response.Writer, r.Request, nil)
- if err != nil {
- log.Println("WebSocket upgrade failed:", err)
- return
- }
-
- mu.Lock()
- clients[conn] = true
- mu.Unlock()
-
- defer func() {
- mu.Lock()
- delete(clients, conn)
- mu.Unlock()
- conn.Close()
- }()
- for {
- _, msg, err := conn.ReadMessage()
- if err != nil {
- log.Println("Error reading message:", err)
- break
- }
- broadcast <- string(msg)
- }
- }
- func handleBroadcast() {
- for {
- msg := <-broadcast
- mu.Lock()
- for client := range clients {
- err := client.WriteMessage(websocket.TextMessage, []byte(msg))
- if err != nil {
- log.Println("Error writing message:", err)
- client.Close()
- delete(clients, client)
- }
- }
- mu.Unlock()
- }
- }
- func init() {
- go handleBroadcast()
- }
|