You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
34 lines
860 B
34 lines
860 B
package scripts
|
|
|
|
import (
|
|
"context"
|
|
model "github.com/cloudreve/Cloudreve/v3/models"
|
|
"github.com/cloudreve/Cloudreve/v3/pkg/util"
|
|
)
|
|
|
|
type UserStorageCalibration int
|
|
|
|
type storageResult struct {
|
|
Total uint64
|
|
}
|
|
|
|
// Run 运行脚本校准所有用户容量
|
|
func (script UserStorageCalibration) Run(ctx context.Context) {
|
|
// 列出所有用户
|
|
var res []model.User
|
|
model.DB.Model(&model.User{}).Find(&res)
|
|
|
|
// 逐个检查容量
|
|
for _, user := range res {
|
|
// 计算正确的容量
|
|
var total storageResult
|
|
model.DB.Model(&model.File{}).Where("user_id = ?", user.ID).Select("sum(size) as total").Scan(&total)
|
|
// 更新用户的容量
|
|
if user.Storage != total.Total {
|
|
util.Log().Info("Calibrate used storage for user %q, from %d to %d.", user.Email,
|
|
user.Storage, total.Total)
|
|
}
|
|
model.DB.Model(&user).Update("storage", total.Total)
|
|
}
|
|
}
|