Feat: file download in service level

This commit is contained in:
HFO4
2019-11-26 20:59:57 +08:00
parent a734493b65
commit 4156a71adf
10 changed files with 99 additions and 30 deletions

View File

@@ -17,6 +17,9 @@ type File struct {
FolderID uint `gorm:"index:folder_id"`
PolicyID uint
Dir string `gorm:"size:65536"`
// 关联模型
Policy Policy `gorm:"PRELOAD:false,association_autoupdate:false"`
}
// Create 创建文件记录
@@ -41,3 +44,12 @@ func (folder *Folder) GetChildFile() ([]File, error) {
result := DB.Where("folder_id = ?", folder.ID).Find(&files)
return files, result.Error
}
// GetPolicy 获取文件所属策略
// TODO:test
func (file *File) GetPolicy() *Policy {
if file.Policy.Model.ID == 0 {
file.Policy, _ = GetPolicyByID(file.PolicyID)
}
return &file.Policy
}