From 291c251fd7e0ee39c9e823f1c291e9ca0f871fbd Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Thu, 31 Aug 2023 16:50:44 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8F=96=E6=8F=92=E5=85=A5ID?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 features/sys_role/service/repo_sys_role.go | 13 ++++++++++++-
 features/sys_user/service/repo_sys_user.go | 13 ++++++++++++-
 2 files changed, 24 insertions(+), 2 deletions(-)

diff --git a/features/sys_role/service/repo_sys_role.go b/features/sys_role/service/repo_sys_role.go
index 17ad309d..7fea81c0 100644
--- a/features/sys_role/service/repo_sys_role.go
+++ b/features/sys_role/service/repo_sys_role.go
@@ -295,8 +295,19 @@ func (r *RepoSysRole) InsertRole(sysRole model.SysRole) string {
 		log.Errorf("insert row : %v", err.Error())
 		return ""
 	}
+	//取插入ID
+	if rows > 0 {
+		results, err := datasource.RawDB("", "SELECT MAX(role_id) AS str FROM sys_role;", nil)
+		if err != nil {
+			log.Errorf("query err %v", err)
+			return ""
+		}
+		if len(results) > 0 {
+			return fmt.Sprintf("%v", results[0]["str"])
+		}
+	}
 
-	return fmt.Sprint(rows)
+	return ""
 }
 
 // DeleteRoleByIds 批量删除角色信息
diff --git a/features/sys_user/service/repo_sys_user.go b/features/sys_user/service/repo_sys_user.go
index 5cdf55ba..1132f2cd 100644
--- a/features/sys_user/service/repo_sys_user.go
+++ b/features/sys_user/service/repo_sys_user.go
@@ -441,8 +441,19 @@ func (r *RepoSysUser) InsertUser(sysUser sysUserModel.SysUser) string {
 		log.Errorf("insert row : %v", err.Error())
 		return ""
 	}
+	//取插入ID
+	if rows > 0 {
+		results, err := datasource.RawDB("", "SELECT MAX(id) AS str FROM user;", nil)
+		if err != nil {
+			log.Errorf("query err %v", err)
+			return ""
+		}
+		if len(results) > 0 {
+			return fmt.Sprintf("%v", results[0]["str"])
+		}
+	}
 
-	return fmt.Sprint(rows)
+	return ""
 }
 
 // UpdateUser 修改用户信息