wsy182/Cloudreve
1
0
Fork 0
mirror of https://github.com/halejohn/Cloudreve.git synced 2026-01-26 09:34:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Feat: mount storage policy for folders in WebDAV

Browse Source
This commit is contained in:
HFO4
2020-02-13 15:27:40 +08:00
parent a0a9686959
commit 44d6ca487c
13 changed files with 76 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
models/user.go
View File

@@ -135,17 +135,20 @@ func (user *User) GetRemainingCapacity() uint64 {
}
// GetPolicyID 获取用户当前的存储策略ID
func (user *User) GetPolicyID() uint {
func (user *User) GetPolicyID(prefer uint) uint {
if prefer == 0 {
prefer = user.OptionsSerialized.PreferredPolicy
}
// 用户未指定时，返回可用的第一个
if user.OptionsSerialized.PreferredPolicy == 0 {
if prefer == 0 {
if len(user.Group.PolicyList) != 0 {
return user.Group.PolicyList[0]
}
return 1
}
// 用户指定时，先检查是否为可用策略列表中的值
if util.ContainsUint(user.Group.PolicyList, user.OptionsSerialized.PreferredPolicy) {
return user.OptionsSerialized.PreferredPolicy
if util.ContainsUint(user.Group.PolicyList, prefer) {
return prefer
}
// 不可用时，返回第一个
if len(user.Group.PolicyList) != 0 {
@@ -205,7 +208,7 @@ func (user *User) AfterFind() (err error) {
}
// 预加载存储策略
user.Policy, _ = GetPolicyByID(user.GetPolicyID())
user.Policy, _ = GetPolicyByID(user.GetPolicyID(0))
return err
}