取插入ID

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 批量删除角色信息

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 修改用户信息