perms-system-server/internal/logic/role/bindRolePermsLogic.go

package role

import (
	"context"
	"time"

	authHelper "perms-system-server/internal/logic/auth"
	"perms-system-server/internal/model/roleperm"
	"perms-system-server/internal/response"
	"perms-system-server/internal/svc"
	"perms-system-server/internal/types"

	"github.com/zeromicro/go-zero/core/logx"
	"github.com/zeromicro/go-zero/core/stores/sqlx"
)

type BindRolePermsLogic struct {
	logx.Logger
	ctx    context.Context
	svcCtx *svc.ServiceContext
}

func NewBindRolePermsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *BindRolePermsLogic {
	return &BindRolePermsLogic{
		Logger: logx.WithContext(ctx),
		ctx:    ctx,
		svcCtx: svcCtx,
	}
}

func (l *BindRolePermsLogic) BindRolePerms(req *types.BindPermsReq) error {
	role, err := l.svcCtx.SysRoleModel.FindOne(l.ctx, req.RoleId)
	if err != nil {
		return response.ErrNotFound("角色不存在")
	}

	if err := authHelper.RequireProductAdminFor(l.ctx, role.ProductCode); err != nil {
		return err
	}

	if len(req.PermIds) > 0 {
		seen := make(map[int64]bool, len(req.PermIds))
		uniqueIds := make([]int64, 0, len(req.PermIds))
		for _, id := range req.PermIds {
			if !seen[id] {
				seen[id] = true
				uniqueIds = append(uniqueIds, id)
			}
		}
		req.PermIds = uniqueIds
	}

	if len(req.PermIds) > 0 {
		perms, err := l.svcCtx.SysPermModel.FindByIds(l.ctx, req.PermIds)
		if err != nil {
			return err
		}
		if len(perms) != len(req.PermIds) {
			return response.ErrBadRequest("包含无效的权限ID")
		}
		for _, p := range perms {
			if p.ProductCode != role.ProductCode {
				return response.ErrBadRequest("不能绑定其他产品的权限")
			}
		}
	}

	if err := l.svcCtx.SysRolePermModel.TransactCtx(l.ctx, func(ctx context.Context, session sqlx.Session) error {
		if err := l.svcCtx.SysRolePermModel.DeleteByRoleIdTx(ctx, session, req.RoleId); err != nil {
			return err
		}
		if len(req.PermIds) == 0 {
			return nil
		}
		now := time.Now().Unix()
		data := make([]*roleperm.SysRolePerm, 0, len(req.PermIds))
		for _, permId := range req.PermIds {
			data = append(data, &roleperm.SysRolePerm{
				RoleId:     req.RoleId,
				PermId:     permId,
				CreateTime: now,
				UpdateTime: now,
			})
		}
		return l.svcCtx.SysRolePermModel.BatchInsertWithTx(ctx, session, data)
	}); err != nil {
		return err
	}

	affectedUserIds, _ := l.svcCtx.SysUserRoleModel.FindUserIdsByRoleId(l.ctx, req.RoleId)
	l.svcCtx.UserDetailsLoader.BatchDel(l.ctx, affectedUserIds, role.ProductCode)
	return nil
}