nyne/nysoure
1
0
Fork 0
mirror of https://github.com/wgh136/nysoure.git synced 2025-12-16 23:51:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add tag field to file and uploading file models, and update related functions

Browse Source
This commit is contained in:
nyne
2025-11-22 17:42:00 +08:00
parent 070b9c7656
commit 1c23bf1d6e
5 changed files with 46 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
server/dao/file.go
View File

@@ -10,7 +10,7 @@ import (
"gorm.io/gorm/clause"
)
func CreateUploadingFile(filename string, description string, fileSize int64, blockSize int64, tempPath string, resourceID, storageID, userID uint) (*model.UploadingFile, error) {
func CreateUploadingFile(filename string, description string, fileSize int64, blockSize int64, tempPath string, resourceID, storageID, userID uint, tag string) (*model.UploadingFile, error) {
blocksCount := (fileSize + blockSize - 1) / blockSize
uf := &model.UploadingFile{
Filename: filename,
@@ -22,6 +22,7 @@ func CreateUploadingFile(filename string, description string, fileSize int64, bl
TargetResourceID: resourceID,
TargetStorageID: storageID,
UserID: userID,
Tag: tag,
}
if err := db.Create(uf).Error; err != nil {
return nil, err
@@ -73,7 +74,7 @@ func GetUploadingFilesOlderThan(time time.Time) ([]model.UploadingFile, error) {
return files, nil
}
func CreateFile(filename string, description string, resourceID uint, storageID *uint, storageKey string, redirectUrl string, size int64, userID uint, hash string) (*model.File, error) {
func CreateFile(filename string, description string, resourceID uint, storageID *uint, storageKey string, redirectUrl string, size int64, userID uint, hash string, tag string) (*model.File, error) {
if storageID == nil && redirectUrl == "" {
return nil, errors.New("storageID and redirectUrl cannot be both empty")
}
@@ -89,6 +90,7 @@ func CreateFile(filename string, description string, resourceID uint, storageID
Size: size,
UserID: userID,
Hash: hash,
Tag: tag,
}
err := db.Transaction(func(tx *gorm.DB) error {
@@ -171,7 +173,7 @@ func DeleteFile(id string) error {
return nil
}
func UpdateFile(id string, filename string, description string) (*model.File, error) {
func UpdateFile(id string, filename string, description string, tag string) (*model.File, error) {
f := &model.File{}
if err := db.Where("uuid = ?", id).First(f).Error; err != nil {
return nil, err
@@ -182,6 +184,9 @@ func UpdateFile(id string, filename string, description string) (*model.File, er
if description != "" {
f.Description = description
}
if tag != "" {
f.Tag = tag
}
if err := db.Save(f).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, model.NewNotFoundError("file not found")