opensource/cloudreve
1
0
Fork 0
mirror of https://github.com/cloudreve/cloudreve.git synced 2025-11-01 19:13:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat: after uploading hooks and checks

Browse Source
This commit is contained in:
HFO4
2019-11-18 19:09:56 +08:00
parent 0dddc12609
commit aa17aa8e6a
10 changed files with 105 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
models/user.go
View File

@ -40,7 +40,7 @@ type User struct {
Options string `json:"-",gorm:"size:4096"`
// 关联模型
Group Group
Group Group `gorm:"association_autoupdate:false"`
Policy Policy `gorm:"PRELOAD:false,association_autoupdate:false"`
// 数据库忽略字段
@ -58,7 +58,7 @@ type UserOption struct {
func (user *User) DeductionStorage(size uint64) bool {
if size <= user.Storage {
user.Storage -= size
DB.Save(user)
DB.Model(user).UpdateColumn("storage", gorm.Expr("storage - ?", size))
return true
}
return false
@ -68,7 +68,7 @@ func (user *User) DeductionStorage(size uint64) bool {
func (user *User) IncreaseStorage(size uint64) bool {
if size <= user.GetRemainingCapacity() {
user.Storage += size
DB.Save(user)
DB.Model(user).UpdateColumn("storage", gorm.Expr("storage + ?", size))
return true
}
return false