112 lines
2.8 KiB
Go
112 lines
2.8 KiB
Go
// Package ws -----------------------------
|
||
// @file : client.go
|
||
// @author : JJXu
|
||
// @contact : wavingbear@163.com
|
||
// @time : 2022/10/21 18:18:05
|
||
// -------------------------------------------
|
||
package ws
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"github.com/gorilla/websocket"
|
||
uuid "github.com/satori/go.uuid"
|
||
"go.uber.org/zap"
|
||
"log"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
var (
|
||
// 注册事件最大等待时间
|
||
limitRegisterWaitTime = time.Second * 6
|
||
limitReadTime = time.Second * 5
|
||
)
|
||
|
||
// NewClient 创建客户端实例
|
||
//
|
||
// param userId 用户id
|
||
// param uid 用户uuid
|
||
// param conn 客户端websocket连接对象
|
||
// return *Client
|
||
func NewClient(userId int64, uid string, conn *websocket.Conn, room *ChatRoom) *Client {
|
||
uidobj, _ := uuid.NewV4()
|
||
var v = &Client{
|
||
Room: room,
|
||
UserId: userId,
|
||
Uuid: uid,
|
||
ClientId: strings.Replace(uidobj.String(), "-", "", -1),
|
||
Conn: conn,
|
||
Send: make(chan []byte, 500),
|
||
}
|
||
return v
|
||
}
|
||
|
||
type Client struct {
|
||
Room *ChatRoom `json:"-" `
|
||
UserId int64 `json:"userId" ` //用户id
|
||
Uuid string `json:"uuid"` //画家uid
|
||
ClientId string `json:"clientId"` //为用户不同设备分配不同的客户端ID
|
||
Conn *websocket.Conn `json:"-"`
|
||
Send chan []byte
|
||
SessionId string `json:"sessionId"` //会话ID，同一个用户多客户端登录，会话ID相同
|
||
Waiter bool `json:"waiter"` //是否是客服
|
||
}
|
||
|
||
func (c *Client) Reading(ctx context.Context, handleFunc ...func(sourceData []byte, cli *Client)) {
|
||
defer func() {
|
||
c.Room.UnRegister <- c
|
||
ctx.Done()
|
||
return
|
||
}()
|
||
//c.Conn.SetReadLimit(maxMessageSize)
|
||
c.Conn.SetReadDeadline(time.Now().Add(pongWait))
|
||
//接收到ping命令后，更新读取时间
|
||
c.Conn.SetPongHandler(func(string) error {
|
||
c.Conn.SetReadDeadline(time.Now().Add(pongWait))
|
||
return nil
|
||
})
|
||
for {
|
||
msgType, byteData, err := c.Conn.ReadMessage()
|
||
if msgType == -1 {
|
||
return
|
||
}
|
||
if err != nil {
|
||
if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseAbnormalClosure) {
|
||
log.Println("ws连接已关闭", zap.Error(err))
|
||
}
|
||
break
|
||
}
|
||
if handleFunc != nil {
|
||
handleFunc[0](byteData, c)
|
||
} else {
|
||
HandleMessage(byteData, c)
|
||
}
|
||
}
|
||
}
|
||
func (c *Client) WriteWait() {
|
||
ticker := time.NewTicker(pingPeriod)
|
||
defer func() {
|
||
ticker.Stop()
|
||
c.Conn.Close()
|
||
}()
|
||
|
||
for {
|
||
select {
|
||
case msg, ok := <-c.Send:
|
||
fmt.Printf("发送消息:%+v\n", string(msg))
|
||
c.Conn.WriteMessage(websocket.TextMessage, msg)
|
||
fmt.Println("发送消息结束")
|
||
if !ok {
|
||
// 聊天室关闭了管道
|
||
c.Conn.WriteControl(websocket.CloseMessage, []byte{}, time.Now().Add(writeWait))
|
||
return
|
||
}
|
||
case <-ticker.C:
|
||
if err := c.Conn.WriteControl(websocket.PingMessage, nil, time.Now().Add(pongWait)); err != nil {
|
||
return
|
||
}
|
||
}
|
||
}
|
||
}
|