diff --git a/server/chat_handler.go b/server/chat_handler.go index 1ebb7bed..e84da722 100644 --- a/server/chat_handler.go +++ b/server/chat_handler.go @@ -63,18 +63,18 @@ func (s *Server) ChatHandle(c *gin.Context) { return } logger.Info("Receive a message: ", string(message)) - replyMessage(client, "当前 TOKEN 无效,请使用合法的 TOKEN 登录!", false) - replyMessage(client, "![](images/wx.png)", false) - //ctx, cancel := context.WithCancel(context.Background()) - //s.ReqCancelFunc[sessionId] = cancel - //// 回复消息 - //err = s.sendMessage(ctx, session, chatRole, string(message), client, false) - //if err != nil { - // logger.Error(err) - //} else { - // replyChunkMessage(client, types.WsMessage{Type: types.WsEnd, IsHelloMsg: false}) - // logger.Info("回答完毕: " + string(message)) - //} + //replyMessage(client, "当前 TOKEN 无效,请使用合法的 TOKEN 登录!", false) + //replyMessage(client, "![](images/wx.png)", false) + ctx, cancel := context.WithCancel(context.Background()) + s.ReqCancelFunc[sessionId] = cancel + // 回复消息 + err = s.sendMessage(ctx, session, chatRole, string(message), client, false) + if err != nil { + logger.Error(err) + } else { + replyChunkMessage(client, types.WsMessage{Type: types.WsEnd, IsHelloMsg: false}) + logger.Info("回答完毕: " + string(message)) + } } }()