MiaoStars/cloudreve
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(workflow): import files from external storage

Browse Source
This commit is contained in:
Aaron Liu
2025-05-20 10:45:16 +08:00
parent 5d72faf688
commit a10a008ed7
32 changed files with 1071 additions and 609 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
inventory/file.go
View File

@@ -120,6 +120,7 @@ type (
Source string
Size int64
UploadSessionID uuid.UUID
Importing bool
}
RelocateEntityParameter struct {
@@ -716,6 +717,11 @@ func (f *fileClient) CreateFile(ctx context.Context, root *ent.File, args *Creat
SetParent(root).
SetIsSymbolic(args.IsSymbolic).
SetStoragePoliciesID(args.StoragePolicyID)
if args.EntityParameters != nil && args.EntityParameters.Importing {
stm.SetSize(args.EntityParameters.Size)
}
newFile, err := stm.Save(ctx)
if err != nil {
return nil, nil, nil, fmt.Errorf("failed to create file: %v", err)
@@ -730,6 +736,12 @@ func (f *fileClient) CreateFile(ctx context.Context, root *ent.File, args *Creat
if err != nil {
return nil, nil, storageDiff, fmt.Errorf("failed to create default entity: %v", err)
}
if args.EntityParameters.Importing {
if err := f.client.File.UpdateOne(newFile).SetPrimaryEntity(defaultEntity.ID).Exec(ctx); err != nil {
return nil, nil, storageDiff, fmt.Errorf("failed to set primary entity: %v", err)
}
}
}
// Create metadata if needed
@@ -848,7 +860,7 @@ func (f *fileClient) CreateEntity(ctx context.Context, file *ent.File, args *Ent
stm.SetUpdatedAt(*args.ModifiedAt)
}
if args.UploadSessionID != uuid.Nil {
if args.UploadSessionID != uuid.Nil && !args.Importing {
stm.SetUploadSessionID(args.UploadSessionID)
}