44 lines
1.4 KiB
Go
44 lines
1.4 KiB
Go
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
|
||
}
|