xyxie пре 4 месеци
родитељ
комит
a22fad05cd
3 измењених фајлова са 8 додато и 3 уклоњено
  1. 1 0
      models/db.go
  2. 2 0
      models/system/sys_role.go
  3. 5 3
      services/sys_role.go

+ 1 - 0
models/db.go

@@ -47,4 +47,5 @@ func initEdbData() {
 
 func initSystem() {
 	orm.RegisterModel(new(system.Admin))
+	orm.RegisterModel(new(system.SysRole))
 }

+ 2 - 0
models/system/sys_role.go

@@ -9,6 +9,7 @@ type SysRole struct {
 	RoleName     string `description:"角色名称"`
 	RoleType     string `description:"角色类型"`
 	RoleTypeCode string `description:"角色类型编码"`
+	RoleLevel    int    `description:"角色等级:0-表示一级角色,每个账号只能绑定一个一级角色 ,1-表示二级角色,每个账号可以绑定多个二级 角色"`
 	CreateTime   string `description:"创建时间"`
 	ModifyTime   string
 }
@@ -18,6 +19,7 @@ type AddRoleReq struct {
 	RoleName     string `description:"角色名称"`
 	RoleType     string `description:"角色类型"`
 	RoleTypeCode string `description:"角色类型编码"`
+	RoleLevel    int    `description:"角色等级:0-表示一级角色,每个账号只能绑定一个一级角色 ,1-表示二级角色,每个账号可以绑定多个二级 角色"`
 	CreateTime   string `description:"创建时间"`
 	ModifyTime   string
 	BusinessCode string `description:"商家编码"`

+ 5 - 3
services/sys_role.go

@@ -7,7 +7,7 @@ import (
 	"fmt"
 )
 
-// 新增或者更新用户
+// 新增或者更新角色
 func AddOrUpdateRole(req *system.AddRoleReq) (err error, isSendEmail bool) {
 	defer func() {
 		if err != nil {
@@ -17,8 +17,7 @@ func AddOrUpdateRole(req *system.AddRoleReq) (err error, isSendEmail bool) {
 	}()
 	isSendEmail = true
 	isAdd := true
-	roleInfo := new(system.SysRole)
-	roleInfo, err = system.GetSysRoleByRoleId(req.RoleId)
+	roleInfo, err := system.GetSysRoleByRoleId(req.RoleId)
 	if err != nil {
 		if err.Error() != utils.ErrNoRow() {
 			err = fmt.Errorf("获取数据失败,%v", err)
@@ -27,12 +26,15 @@ func AddOrUpdateRole(req *system.AddRoleReq) (err error, isSendEmail bool) {
 	}
 	if err == nil && roleInfo.RoleId > 0 {
 		isAdd = false
+	} else {
+		roleInfo = new(system.SysRole)
 	}
 	// 角色信息
 	roleInfo.RoleId = req.RoleId
 	roleInfo.RoleName = req.RoleName
 	roleInfo.RoleType = req.RoleType
 	roleInfo.RoleTypeCode = req.RoleTypeCode
+	roleInfo.RoleLevel = req.RoleLevel
 	roleInfo.ModifyTime = req.ModifyTime
 	roleInfo.CreateTime = req.CreateTime
 	if isAdd {