修复「1.2.2 用户管理-用户列表勾选单一用户会全选」 close #216

2022-01-23 17:53:22 +08:00
parent 29c066ca3a
commit d35b348a33
130 changed files with 5467 additions and 4554 deletions
--- a/server/service/user_group.go
+++ b/server/service/user_group.go
@ -0,0 +1,115 @@
+package service
+
+import (
+	"context"
+	"errors"
+
+	"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 {
+}
+
+func (service userGroupService) DeleteById(userGroupId string) error {
+	return env.GetDB().Transaction(func(tx *gorm.DB) error {
+		c := context.WithValue(context.TODO(), constant.DB, 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(name string, members []string) (model.UserGroup, error) {
+	var err error
+	_, err = repository.UserGroupRepository.FindByName(context.TODO(), name)
+	if err == nil {
+		return model.UserGroup{}, constant.ErrNameAlreadyUsed
+	}
+
+	if !errors.Is(gorm.ErrRecordNotFound, err) {
+		return model.UserGroup{}, err
+	}
+
+	userGroupId := utils.UUID()
+	userGroup := model.UserGroup{
+		ID:      userGroupId,
+		Created: utils.NowJsonTime(),
+		Name:    name,
+	}
+
+	return userGroup, env.GetDB().Transaction(func(tx *gorm.DB) error {
+		c := context.WithValue(context.TODO(), constant.DB, tx)
+		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) {
+	var userGroup model.UserGroup
+	userGroup, err = repository.UserGroupRepository.FindByName(context.TODO(), name)
+	if err == nil && userGroup.ID != userGroupId {
+		return constant.ErrNameAlreadyUsed
+	}
+
+	if !errors.Is(gorm.ErrRecordNotFound, err) {
+		return err
+	}
+
+	return env.GetDB().Transaction(func(tx *gorm.DB) error {
+		c := context.WithValue(context.TODO(), constant.DB, 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
+	})
+}