76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
package roleservicelogic
|
|
|
|
import (
|
|
ers "code.30cm.net/wanderland/library-go/errors"
|
|
"context"
|
|
"fmt"
|
|
"github.com/zeromicro/go-zero/core/stores/sqlx"
|
|
|
|
"ark-permission/gen_result/pb/permission"
|
|
"ark-permission/internal/svc"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type UpdateRoleLogic struct {
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
logx.Logger
|
|
}
|
|
|
|
func NewUpdateRoleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateRoleLogic {
|
|
return &UpdateRoleLogic{
|
|
ctx: ctx,
|
|
svcCtx: svcCtx,
|
|
Logger: logx.WithContext(ctx),
|
|
}
|
|
}
|
|
|
|
// UpdateRole 更新角色
|
|
func (l *UpdateRoleLogic) UpdateRole(in *permission.UpdateRoleReq) (*permission.RoleResp, error) {
|
|
if in.GetRoleId() == "" {
|
|
return nil, ers.InvalidFormat("failed to get role id")
|
|
}
|
|
|
|
// 暴露出 transaction 的統包，可以實現跨表跟同表
|
|
err := l.svcCtx.Conn.TransactCtx(l.ctx, func(ctx context.Context, session sqlx.Session) error {
|
|
if in.GetStatus() != 0 {
|
|
if in.GetStatus() == permission.PermissionStatus_PERMISSION_STATUS_NONE {
|
|
return fmt.Errorf("failed to get role id")
|
|
}
|
|
err := l.svcCtx.RoleRepo.TransUpdateStatusByRoleID(
|
|
ctx, session, int64(in.GetStatus().Number()), in.GetRoleId())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
if in.GetDisplayName() != "" {
|
|
err := l.svcCtx.RoleRepo.TransUpdateDisplayNameByRoleID(
|
|
ctx, session, in.GetDisplayName(), in.GetRoleId())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
})
|
|
if err != nil {
|
|
return nil, ers.DBError(err.Error())
|
|
}
|
|
|
|
result, err := l.svcCtx.RoleRepo.FindOneByRoleId(l.ctx, in.GetRoleId())
|
|
if err != nil {
|
|
return nil, ers.DBError(err.Error())
|
|
}
|
|
|
|
return &permission.RoleResp{
|
|
Id: result.Id,
|
|
Status: permission.PermissionStatus(result.Status),
|
|
DisplayName: result.DisplayName,
|
|
RoleId: result.RoleId,
|
|
CreateTime: result.CreateTime,
|
|
UpdateTime: result.UpdateTime,
|
|
}, nil
|
|
}
|