fix version 1

module/base/cloud/internal/logic/disk/quota.go

@@ -0,0 +1,43 @@
+package disk
+
+import (
+	"bsm/full/module/base/cloud/internal/impl"
+	"bsm/full/module/base/cloud/internal/models"
+	"git.apinb.com/bsm-sdk/core/errcode"
+	"git.apinb.com/bsm-sdk/core/printer"
+)
+
+// defaultMaxStorage 默认容量配额100GB，与 Space.Get 创建默认空间时的取值保持一致
+const defaultMaxStorage int64 = 100 * 1024 * 1024 * 1024
+
+// checkQuota 校验「已用空间 + 本次新增」是否超出配额，超出返回明确错误
+func checkQuota(passportID uint, addSize int64) error {
+	if addSize <= 0 {
+		return nil
+	}
+
+	// 配额取用户空间记录的 max_storage，无空间记录时使用默认配额
+	maxStorage := defaultMaxStorage
+	var space models.CloudSpace
+	if err := impl.DBService.Where("passport_id = ?", passportID).First(&space).Error; err == nil && space.MaxStorage > 0 {
+		maxStorage = space.MaxStorage
+	}
+
+	// 已用空间按云盘文件实际占用统计
+	var usedStorage int64
+	row := impl.DBService.Model(&models.CloudDiskFile{}).
+		Joins("JOIN cloud_disk_dirs ON cloud_disk_files.directory_id = cloud_disk_dirs.id").
+		Where("cloud_disk_dirs.passport_id = ?", passportID).
+		Select("COALESCE(SUM(cloud_disk_files.size), 0)").
+		Row()
+	if err := row.Scan(&usedStorage); err != nil {
+		printer.Error("Query used storage error: %v", err)
+		return errcode.ErrDB
+	}
+
+	if usedStorage+addSize > maxStorage {
+		return errcode.ErrResourceExhausted
+	}
+
+	return nil
+}