feat():learning后台管理项目初始化

internal/services/user_project_quota_service.go

@@ -0,0 +1,79 @@
+package services
+
+import (
+	"errors"
+	"goalfymax-admin/internal/models"
+	"goalfymax-admin/internal/storage"
+)
+
+type UserProjectQuotaService interface {
+	Create(input *models.UserProjectQuota) (*models.UserProjectQuota, error)
+	Update(id uint, input *models.UserProjectQuota) (*models.UserProjectQuota, error)
+	Delete(id uint) error
+	GetByID(id uint) (*models.UserProjectQuota, error)
+	GetByUserID(userID string) (*models.UserProjectQuota, error)
+	List(filter storage.UserProjectQuotaFilter) ([]models.UserProjectQuota, int64, error)
+}
+
+type userProjectQuotaService struct {
+	store storage.UserProjectQuotaStorage
+}
+
+func NewUserProjectQuotaService(store storage.UserProjectQuotaStorage) UserProjectQuotaService {
+	return &userProjectQuotaService{store: store}
+}
+
+func (s *userProjectQuotaService) validate(q *models.UserProjectQuota) error {
+	if q.UserID == "" {
+		return errors.New("user_id 不能为空")
+	}
+	if q.ProjectLimit < 0 || q.CoderVMLimit < 0 || q.BrowserVMLimit < 0 || q.ProcessLimit < 0 {
+		return errors.New("配额上限不能为负数")
+	}
+	return nil
+}
+
+func (s *userProjectQuotaService) Create(input *models.UserProjectQuota) (*models.UserProjectQuota, error) {
+	if err := s.validate(input); err != nil {
+		return nil, err
+	}
+	if _, err := s.store.GetByUserID(input.UserID); err == nil {
+		return nil, errors.New("该用户配额已存在")
+	}
+	if err := s.store.Create(input); err != nil {
+		return nil, err
+	}
+	return input, nil
+}
+
+func (s *userProjectQuotaService) Update(id uint, input *models.UserProjectQuota) (*models.UserProjectQuota, error) {
+	if err := s.validate(input); err != nil {
+		return nil, err
+	}
+	exist, err := s.store.GetByID(id)
+	if err != nil {
+		return nil, errors.New("记录不存在")
+	}
+	// user_id 不建议修改，如需修改可放开以下行：exist.UserID = input.UserID
+	exist.ProjectLimit = input.ProjectLimit
+	exist.CoderVMLimit = input.CoderVMLimit
+	exist.BrowserVMLimit = input.BrowserVMLimit
+	exist.ProcessLimit = input.ProcessLimit
+	exist.Enabled = input.Enabled
+	exist.Description = input.Description
+	if err := s.store.Update(exist); err != nil {
+		return nil, err
+	}
+	return exist, nil
+}
+
+func (s *userProjectQuotaService) Delete(id uint) error { return s.store.Delete(id) }
+func (s *userProjectQuotaService) GetByID(id uint) (*models.UserProjectQuota, error) {
+	return s.store.GetByID(id)
+}
+func (s *userProjectQuotaService) GetByUserID(userID string) (*models.UserProjectQuota, error) {
+	return s.store.GetByUserID(userID)
+}
+func (s *userProjectQuotaService) List(filter storage.UserProjectQuotaFilter) ([]models.UserProjectQuota, int64, error) {
+	return s.store.List(filter)
+}