104 lines
2.3 KiB
Go
104 lines
2.3 KiB
Go
package logic
|
|
|
|
import (
|
|
"admin/internal/dao"
|
|
"admin/internal/model"
|
|
"admin/validator"
|
|
"context"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"admin/admin"
|
|
"admin/internal/svc"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
"pkg.local/modelbase"
|
|
"pkg.local/utils"
|
|
)
|
|
|
|
type RoleItemsLogic struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
logx.Logger
|
|
BaseLogic
|
|
}
|
|
|
|
func NewRoleItemsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RoleItemsLogic {
|
|
return &RoleItemsLogic{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
Logger: logx.WithContext(ctx),
|
|
}
|
|
}
|
|
|
|
func (l *RoleItemsLogic) RoleItems(in *admin.RoleItemsRequest) (*admin.Response, error) {
|
|
var v validator.RoleItemsValidator
|
|
if fail := l.checkParams(in, &v); fail != nil {
|
|
return fail, nil
|
|
}
|
|
modelObj := model.RoleModel{}.Init()
|
|
w := modelbase.Params{
|
|
Page: int(in.Page),
|
|
Size: int(in.Size),
|
|
Order: "id asc",
|
|
Eq: map[string]string{"service_id": strconv.FormatInt(in.ServiceId, 10)},
|
|
}
|
|
|
|
w.Like = make(map[string]string)
|
|
if in.Status > utils.NumberZero {
|
|
w.Eq["status"] = strconv.Itoa(int(in.Status))
|
|
}
|
|
if len(in.Name) > utils.NumberZero {
|
|
w.Like["name LIKE ? "] = "%%" + in.Name + "%%"
|
|
}
|
|
|
|
var data []dao.Role
|
|
err := modelObj.Items(w, &data)
|
|
if err != nil {
|
|
l.Logger.Error(err)
|
|
return nil, err
|
|
}
|
|
|
|
var items []dao.RoleItems
|
|
rulesInfos := l.getRulesInfo(data)
|
|
for i := utils.NumberZero; i < len(data); i++ {
|
|
var rules []string
|
|
for _, rule := range strings.Split(data[i].Rules, utils.DecollatorComma) {
|
|
if _, ok := rulesInfos[rule]; ok {
|
|
rules = append(rules, rulesInfos[rule])
|
|
}
|
|
}
|
|
items = append(items, dao.RoleItems{
|
|
Role: data[i],
|
|
RuleItems: rules,
|
|
})
|
|
}
|
|
|
|
return l.ok(items), nil
|
|
}
|
|
|
|
func (l *RoleItemsLogic) getRulesInfo(data []dao.Role) map[string]string {
|
|
var ids []string
|
|
for i := utils.NumberZero; i < len(data); i++ {
|
|
ids = append(ids, strings.Split(data[i].Rules, utils.DecollatorComma)...)
|
|
}
|
|
|
|
w := modelbase.Params{
|
|
In: map[string][]string{"id in ?": ids},
|
|
Other: map[string]string{"type != ?": strconv.Itoa(dao.TypeDir)},
|
|
}
|
|
var items []dao.AuthorityNames
|
|
retData := make(map[string]string)
|
|
|
|
err := model.AuthorityModel{}.Init().Items(w, &items)
|
|
if err != nil {
|
|
return retData
|
|
}
|
|
|
|
for i := utils.NumberZero; i < len(items); i++ {
|
|
retData[strconv.Itoa(int(items[i].Id))] = items[i].Name
|
|
}
|
|
|
|
return retData
|
|
}
|