mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-17 16:56:38 +08:00
30 lines
803 B
Go
30 lines
803 B
Go
package utils
|
|
|
|
import (
|
|
"chatplus/core/types"
|
|
logger2 "chatplus/logger"
|
|
"encoding/json"
|
|
)
|
|
|
|
var logger = logger2.GetLogger()
|
|
|
|
// ReplyChunkMessage 回复客户片段端消息
|
|
func ReplyChunkMessage(client *types.WsClient, message types.WsMessage) {
|
|
msg, err := json.Marshal(message)
|
|
if err != nil {
|
|
logger.Errorf("Error for decoding json data: %v", err.Error())
|
|
return
|
|
}
|
|
err = client.Send(msg)
|
|
if err != nil {
|
|
logger.Errorf("Error for reply message: %v", err.Error())
|
|
}
|
|
}
|
|
|
|
// ReplyMessage 回复客户端一条完整的消息
|
|
func ReplyMessage(ws *types.WsClient, message interface{}) {
|
|
ReplyChunkMessage(ws, types.WsMessage{Type: types.WsStart})
|
|
ReplyChunkMessage(ws, types.WsMessage{Type: types.WsMiddle, Content: message})
|
|
ReplyChunkMessage(ws, types.WsMessage{Type: types.WsEnd})
|
|
}
|