From 27b8c055128a1cfd359d889416f494d60be4535c Mon Sep 17 00:00:00 2001
From: piupuer <piupuer@gmail.com>
Date: Thu, 10 Oct 2024 11:55:55 +0800
Subject: [PATCH] [fix]bug

---
 internal/data/action.go     | 5 ++++-
 internal/data/role.go       | 4 +++-
 internal/data/user.go       | 3 +++
 internal/data/user_group.go | 4 +++-
 4 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/internal/data/action.go b/internal/data/action.go
index 6552c8d..5bb5440 100644
--- a/internal/data/action.go
+++ b/internal/data/action.go
@@ -156,6 +156,7 @@ func (ro actionRepo) CodeExists(ctx context.Context, code string) (err error) {
 		if m.ID == constant.UI0 {
 			err = biz.ErrRecordNotFound(ctx)
 			log.
+				WithContext(ctx).
 				WithError(err).
 				Error("invalid code: %s", code)
 			return
@@ -171,7 +172,9 @@ func (ro actionRepo) WordExists(ctx context.Context, word string) (ok bool) {
 	for _, item := range arr {
 		m := db.GetByCol("word", item)
 		if m.ID == constant.UI0 {
-			log.Error("invalid word: %s", item)
+			log.
+				WithContext(ctx).
+				Error("invalid word: %s", item)
 			return
 		}
 	}
diff --git a/internal/data/role.go b/internal/data/role.go
index 836579e..bf2387e 100644
--- a/internal/data/role.go
+++ b/internal/data/role.go
@@ -129,7 +129,9 @@ func (ro roleRepo) WordExists(ctx context.Context, word string) (ok bool) {
 	for _, item := range arr {
 		m := db.GetByCol("word", item)
 		if m.ID == constant.UI0 {
-			log.Error("invalid word: %s", item)
+			log.
+				WithContext(ctx).
+				Error("invalid word: %s", item)
 			return
 		}
 	}
diff --git a/internal/data/user.go b/internal/data/user.go
index 764bcf7..03b813a 100644
--- a/internal/data/user.go
+++ b/internal/data/user.go
@@ -42,6 +42,7 @@ func (ro userRepo) GetByUsername(ctx context.Context, username string) (item *bi
 	if err != nil || m.ID == constant.UI0 {
 		err = biz.ErrRecordNotFound(ctx)
 		log.
+			WithContext(ctx).
 			WithError(err).
 			Error("invalid username: %s", username)
 		return
@@ -197,6 +198,7 @@ func (ro userRepo) Update(ctx context.Context, item *biz.UpdateUser) (err error)
 			if mRole.ID == constant.UI0 {
 				err = biz.ErrRecordNotFound(ctx)
 				log.
+					WithContext(ctx).
 					WithError(err).
 					Error("invalid roleId: %s", v)
 				return
@@ -311,6 +313,7 @@ func (ro userRepo) GetByCode(ctx context.Context, code string) (item *biz.User,
 	if err != nil || m.ID == constant.UI0 {
 		err = biz.ErrRecordNotFound(ctx)
 		log.
+			WithContext(ctx).
 			WithError(err).
 			Error("invalid code: %s", code)
 		return
diff --git a/internal/data/user_group.go b/internal/data/user_group.go
index 9fbc6d3..ffb236e 100644
--- a/internal/data/user_group.go
+++ b/internal/data/user_group.go
@@ -186,7 +186,9 @@ func (ro userGroupRepo) WordExists(ctx context.Context, word string) (ok bool) {
 	for _, item := range arr {
 		m := db.GetByCol("word", item)
 		if m.ID == constant.UI0 {
-			log.Error("invalid word: %s", item)
+			log.
+				WithContext(ctx).
+				Error("invalid word: %s", item)
 			return
 		}
 	}