rogee/quyun
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add upload timeout

Browse Source
This commit is contained in:
Rogee
2025-05-06 11:13:49 +08:00
parent 72ab4ad7ec
commit bb689c62e1
4 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/app/jobs/download_from_alioss.go
View File

@@ -84,7 +84,7 @@ func (w *DownloadFromAliOSSWorker) Work(ctx context.Context, job *Job[DownloadFr
}
log.Infof("Starting download for file: %s", media.Path)
if err := w.oss.Download(ctx, media.Path, dst, ali.WithInternal(true)); err != nil {
if err := w.oss.Download(ctx, media.Path, dst, ali.WithInternal()); err != nil {
log.Errorf("Error downloading file: %v", err)
return err
}

2
backend/app/jobs/video_extract_head_image.go
View File

@@ -101,7 +101,7 @@ func (w *VideoExtractHeadImageWorker) Work(ctx context.Context, job *Job[VideoEx
}
// upload to oss
if err := w.oss.Upload(ctx, output, imageMedia.Path, ali.WithInternal(true)); err != nil {
if err := w.oss.Upload(ctx, output, imageMedia.Path, ali.WithInternal()); err != nil {
log.Errorf("Error uploading image to OSS: %v", err)
return JobCancel(err)
}

2
backend/app/jobs/video_store_short.go
View File

@@ -101,7 +101,7 @@ func (w *VideoStoreShortWorker) Work(ctx context.Context, job *Job[VideoStoreSho
}
// upload to oss
if err := w.oss.Upload(ctx, job.Args.FilePath, filePath, ali.WithInternal(true)); err != nil {
if err := w.oss.Upload(ctx, job.Args.FilePath, filePath, ali.WithInternal()); err != nil {
log.Errorf("Error uploading file to OSS: %v", err)
return err
}

9
backend/providers/ali/oss_client.go
View File

@@ -22,9 +22,9 @@ func WithExpire(expire time.Duration) OSSOption {
}
}
func WithInternal(internal bool) OSSOption {
func WithInternal() OSSOption {
return func(o *OSSOptions) {
o.internal = internal
o.internal = true
}
}
@@ -152,7 +152,10 @@ func (c *OSSClient) Upload(ctx context.Context, input, dst string, opts ...OSSOp
client = c.internalClient
}
if _, err := client.PutObjectFromFile(ctx, request, input); err != nil {
ossFuncs := []func(*oss.Options){
oss.OpReadWriteTimeout(time.Minute * 20),
}
if _, err := client.PutObjectFromFile(ctx, request, input, ossFuncs...); err != nil {
return err
}
return nil