128 lines
3.3 KiB
Go
128 lines
3.3 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
|
|
"next-terminal/server/constant"
|
|
"next-terminal/server/env"
|
|
"next-terminal/server/model"
|
|
"next-terminal/server/repository"
|
|
"next-terminal/server/utils"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type userGroupService struct {
|
|
baseService
|
|
}
|
|
|
|
func (service userGroupService) DeleteById(userGroupId string) error {
|
|
return env.GetDB().Transaction(func(tx *gorm.DB) error {
|
|
c := service.Context(tx)
|
|
// 删除用户组
|
|
if err := repository.UserGroupRepository.DeleteById(c, userGroupId); err != nil {
|
|
return err
|
|
}
|
|
// 删除用户组与用户的关系
|
|
if err := repository.UserGroupMemberRepository.DeleteByUserGroupId(c, userGroupId); err != nil {
|
|
return err
|
|
}
|
|
// 删除用户组与资产的关系
|
|
if err := repository.ResourceSharerRepository.DeleteByUserGroupId(c, userGroupId); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
})
|
|
}
|
|
|
|
func (service userGroupService) Create(ctx context.Context, name string, members []string) (model.UserGroup, error) {
|
|
exist, err := repository.UserGroupRepository.ExistByName(ctx, name)
|
|
if err != nil {
|
|
return model.UserGroup{}, err
|
|
}
|
|
|
|
if exist {
|
|
return model.UserGroup{}, constant.ErrNameAlreadyUsed
|
|
}
|
|
|
|
userGroupId := utils.UUID()
|
|
userGroup := model.UserGroup{
|
|
ID: userGroupId,
|
|
Created: utils.NowJsonTime(),
|
|
Name: name,
|
|
}
|
|
|
|
if service.InTransaction(ctx) {
|
|
return userGroup, service.create(ctx, userGroup, members, userGroupId)
|
|
} else {
|
|
return userGroup, env.GetDB().Transaction(func(tx *gorm.DB) error {
|
|
c := service.Context(tx)
|
|
return service.create(c, userGroup, members, userGroupId)
|
|
})
|
|
}
|
|
}
|
|
|
|
func (service userGroupService) create(c context.Context, userGroup model.UserGroup, members []string, userGroupId string) error {
|
|
if err := repository.UserGroupRepository.Create(c, &userGroup); err != nil {
|
|
return err
|
|
}
|
|
if len(members) > 0 {
|
|
for _, member := range members {
|
|
userGroupMember := model.UserGroupMember{
|
|
ID: utils.Sign([]string{userGroupId, member}),
|
|
UserId: member,
|
|
UserGroupId: userGroupId,
|
|
}
|
|
if err := repository.UserGroupMemberRepository.Create(c, &userGroupMember); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (service userGroupService) Update(userGroupId string, name string, members []string) (err error) {
|
|
dbUserGroup, err := repository.UserGroupRepository.FindById(context.TODO(), userGroupId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if dbUserGroup.Name != name {
|
|
// 修改了名称
|
|
exist, err := repository.UserGroupRepository.ExistByName(context.TODO(), name)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if exist {
|
|
return constant.ErrNameAlreadyUsed
|
|
}
|
|
}
|
|
|
|
return env.GetDB().Transaction(func(tx *gorm.DB) error {
|
|
c := service.Context(tx)
|
|
userGroup := model.UserGroup{
|
|
ID: userGroupId,
|
|
Name: name,
|
|
}
|
|
if err := repository.UserGroupRepository.Update(c, &userGroup); err != nil {
|
|
return err
|
|
}
|
|
if err := repository.UserGroupMemberRepository.DeleteByUserGroupId(c, userGroupId); err != nil {
|
|
return err
|
|
}
|
|
if len(members) > 0 {
|
|
for _, member := range members {
|
|
userGroupMember := model.UserGroupMember{
|
|
ID: utils.Sign([]string{userGroupId, member}),
|
|
UserId: member,
|
|
UserGroupId: userGroupId,
|
|
}
|
|
if err := repository.UserGroupMemberRepository.Create(c, &userGroupMember); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
}
|