Kaynağa Gözat

wip:日志输出修改

wangkaiyue 2 yıl önce
ebeveyn
işleme
1fb30b385c

+ 2 - 2
internal/controller/chatHistory.go

@@ -6,7 +6,7 @@ import (
 	"app.yhyue.com/moapp/jybase/encrypt"
 	"context"
 	"fmt"
-	"github.com/gogf/gf/v2/os/glog"
+	"github.com/gogf/gf/v2/frame/g"
 )
 
 var (
@@ -27,7 +27,7 @@ func (c *cChatHistory) Method(ctx context.Context, req *v1.ChatHistoryReq) (res
 	}
 	history, hasNext, err := model.ChatHistory.GetMessage(session.PositionId, req.PageNum, req.PageSize, req.PrevId)
 	if err != nil {
-		glog.Error(ctx, "%d查询聊天记录异常,error:%s", session.PositionId, err)
+		g.Log().Error(ctx, "%d查询聊天记录异常,error:%s", session.PositionId, err)
 		res.ErrorCode = -1
 		res.ErrorMsg = "数据查询异常"
 		return

+ 2 - 3
internal/controller/chatWs.go

@@ -4,7 +4,6 @@ import (
 	"aiChat/internal/model"
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/net/ghttp"
-	"github.com/gogf/gf/v2/os/glog"
 )
 
 // ChatWs 聊天websocket请求
@@ -17,7 +16,7 @@ var ChatWs = func(r *ghttp.Request) {
 	// 创建ws链接
 	ws, err := r.WebSocket()
 	if err != nil {
-		glog.Error(wsChat.Ctx, session.PositionId, "建立连接出错", err)
+		g.Log().Error(wsChat.Ctx, session.PositionId, "建立连接出错", err)
 		r.Exit()
 	}
 	if session.PositionId == 0 {
@@ -33,7 +32,7 @@ var ChatWs = func(r *ghttp.Request) {
 		_, msg, err := ws.ReadMessage()
 
 		if err != nil {
-			glog.Info(wsChat.Ctx, session.PositionId, "接收消息出错", err)
+			g.Log().Info(wsChat.Ctx, session.PositionId, "接收消息出错", err)
 			_ = ws.Close()
 			return
 		}

+ 5 - 6
internal/model/chatHistory.go

@@ -5,7 +5,6 @@ import (
 	"fmt"
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/os/gctx"
-	"github.com/gogf/gf/v2/os/glog"
 	"github.com/gogf/gf/v2/util/gconv"
 	"time"
 )
@@ -51,15 +50,15 @@ func (m *cChatHistory) CacheSave(msg *ChatRecord) {
 		val := gconv.Maps(tmp)
 		r, err := g.Model("ai_message_history").Data(val).Insert()
 		if err != nil {
-			glog.Error(ctx, "插入聊天记录异常,error:%s\ndata:%v", err, val)
+			g.Log().Error(ctx, "插入聊天记录异常,error:%s\ndata:%v", err, val)
 			g.Dump(val)
 		}
 		affect, _ := r.RowsAffected()
 		if len(tmp) == gconv.Int(affect) {
-			glog.Error(ctx, "插入聊天记录异常 共%d条 插入%d条,error:%s\ndata:%v", len(tmp), affect, err, val)
+			g.Log().Error(ctx, "插入聊天记录异常 共%d条 插入%d条,error:%s\ndata:%v", len(tmp), affect, err, val)
 			g.Dump(val)
 		} else {
-			glog.Info(ctx, "插入%d条聊天记录成功", affect)
+			g.Log().Info(ctx, "插入%d条聊天记录成功", affect)
 		}
 	}
 }
@@ -70,11 +69,11 @@ func (m *cChatHistory) Save(ctx context.Context, msgs ...*ChatRecord) (id int64)
 	r, err := g.Model("ai_message_history").Data(val).Insert()
 
 	if err != nil {
-		glog.Error(ctx, "插入聊天记录异常,error:%v\ndata:%v", err, val)
+		g.Log().Error(ctx, "插入聊天记录异常,error:%v\ndata:%v", err, val)
 		g.Dump(val)
 	} else {
 		if id, err = r.LastInsertId(); err != nil {
-			glog.Error(ctx, "插入聊天记录异常")
+			g.Log().Error(ctx, "插入聊天记录异常")
 			g.Dump(val)
 		}
 	}

+ 1 - 2
internal/model/ws.go

@@ -12,7 +12,6 @@ import (
 	"github.com/gogf/gf/v2/encoding/gjson"
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/net/ghttp"
-	"github.com/gogf/gf/v2/os/glog"
 	"github.com/gogf/gf/v2/util/gconv"
 	"io"
 	"time"
@@ -41,7 +40,7 @@ func (m *WsChat) Handle(ws *ghttp.WebSocket, msg []byte) {
 	}
 	req, from := &QuestionReq{}, 0
 	if err := gjson.Unmarshal(msg, req); err != nil {
-		glog.Errorf(m.Ctx, "%d 接收消息Unmarshal出错:%v", jSession.PositionId, err)
+		g.Log().Errorf(m.Ctx, "%d 接收消息Unmarshal出错:%v", jSession.PositionId, err)
 		return
 	}
 	questionId := ChatHistory.Save(m.Ctx, &ChatRecord{