init

bbs-go/server/internal/services/user_role_service.go

@@ -0,0 +1,99 @@
+package services
+
+import (
+	"bbs-go/internal/cache"
+	"bbs-go/internal/models"
+	"bbs-go/internal/repositories"
+	"strings"
+
+	"github.com/mlogclub/simple/common/dates"
+	"github.com/mlogclub/simple/sqls"
+	"github.com/mlogclub/simple/web/params"
+	"gorm.io/gorm"
+)
+
+var UserRoleService = newUserRoleService()
+
+func newUserRoleService() *userRoleService {
+	return &userRoleService{}
+}
+
+type userRoleService struct {
+}
+
+func (s *userRoleService) Get(id int64) *models.UserRole {
+	return repositories.UserRoleRepository.Get(sqls.DB(), id)
+}
+
+func (s *userRoleService) Take(where ...interface{}) *models.UserRole {
+	return repositories.UserRoleRepository.Take(sqls.DB(), where...)
+}
+
+func (s *userRoleService) Find(cnd *sqls.Cnd) []models.UserRole {
+	return repositories.UserRoleRepository.Find(sqls.DB(), cnd)
+}
+
+func (s *userRoleService) FindOne(cnd *sqls.Cnd) *models.UserRole {
+	return repositories.UserRoleRepository.FindOne(sqls.DB(), cnd)
+}
+
+func (s *userRoleService) FindPageByParams(params *params.QueryParams) (list []models.UserRole, paging *sqls.Paging) {
+	return repositories.UserRoleRepository.FindPageByParams(sqls.DB(), params)
+}
+
+func (s *userRoleService) FindPageByCnd(cnd *sqls.Cnd) (list []models.UserRole, paging *sqls.Paging) {
+	return repositories.UserRoleRepository.FindPageByCnd(sqls.DB(), cnd)
+}
+
+func (s *userRoleService) Count(cnd *sqls.Cnd) int64 {
+	return repositories.UserRoleRepository.Count(sqls.DB(), cnd)
+}
+
+func (s *userRoleService) Delete(id int64) {
+	repositories.UserRoleRepository.Delete(sqls.DB(), id)
+}
+
+func (s *userRoleService) UpdateUserRoles(userId int64, roleIds []int64) error {
+	err := sqls.DB().Transaction(func(tx *gorm.DB) error {
+		var roles []models.Role
+		if len(roleIds) > 0 {
+			roles = repositories.RoleRepository.Find(tx, sqls.NewCnd().In("id", roleIds))
+		}
+
+		var roleCodes []string
+		for _, role := range roles {
+			roleCodes = append(roleCodes, role.Code)
+		}
+
+		if err := tx.Delete(&models.UserRole{}, "user_id = ?", userId).Error; err != nil {
+			return err
+		}
+		if len(roles) == 0 {
+			return repositories.UserRepository.UpdateColumn(tx, userId, "roles", "")
+		} else {
+			for _, role := range roles {
+				if err := repositories.UserRoleRepository.Create(tx, &models.UserRole{
+					UserId:     userId,
+					RoleId:     role.Id,
+					CreateTime: dates.NowTimestamp(),
+				}); err != nil {
+					return err
+				}
+			}
+			return repositories.UserRepository.UpdateColumn(tx, userId, "roles", strings.Join(roleCodes, ","))
+		}
+	})
+	if err != nil {
+		return err
+	}
+	cache.UserCache.Invalidate(userId)
+	return nil
+}
+
+func (s *userRoleService) GetUserRoleIds(userId int64) (roleIds []int64) {
+	list := s.Find(sqls.NewCnd().Eq("user_id", userId))
+	for _, userRole := range list {
+		roleIds = append(roleIds, userRole.RoleId)
+	}
+	return roleIds
+}