feat: add download from alioss job

backend/app/jobs/download_from_alioss.go

@@ -2,8 +2,12 @@ package jobs
 
 import (
 	"context"
+	"os"
+	"path/filepath"
 	"time"
 
+	"quyun/providers/ali"
+
 	. "github.com/riverqueue/river"
 	log "github.com/sirupsen/logrus"
 	_ "go.ipao.vip/atom"
@@ -14,8 +18,7 @@ import (
 var _ contracts.JobArgs = (*WechatCallback)(nil)
 
 type DownloadFromAliOSS struct {
-	Bucket string
-	Path   string
+	Path string
 }
 
 func (s DownloadFromAliOSS) InsertOpts() InsertOpts {
@@ -33,6 +36,12 @@ var _ Worker[DownloadFromAliOSS] = (*DownloadFromAliOSSWorker)(nil)
 // @provider(job)
 type DownloadFromAliOSSWorker struct {
 	WorkerDefaults[DownloadFromAliOSS]
+
+	oss *ali.OSSClient
+}
+
+func (w *DownloadFromAliOSSWorker) NextRetry(job *Job[DownloadFromAliOSS]) time.Time {
+	return time.Now().Add(30 * time.Second)
 }
 
 func (w *DownloadFromAliOSSWorker) Work(ctx context.Context, job *Job[DownloadFromAliOSS]) error {
@@ -41,9 +50,37 @@ func (w *DownloadFromAliOSSWorker) Work(ctx context.Context, job *Job[DownloadFr
 	log.Infof("[Start] Working on job with strings: %+v", job.Args)
 	defer log.Infof("[End] Finished %s", job.Args.Kind())
 
+	dst := filepath.Join("/Users/rogee/Projects/self/quyun/backend/fixtures/oss/", job.Args.Path)
+
+	// check is path exist
+	_, err := os.Stat(dst)
+	if os.IsNotExist(err) {
+		log.Infof("File not exists: %s", dst)
+		// midir
+		err := os.MkdirAll(filepath.Dir(dst), os.ModePerm)
+		if err != nil {
+			log.Errorf("Error creating directory: %v", err)
+			return err
+		}
+	}
+
+	if err == nil {
+		log.Infof("File already exists: %s", dst)
+		err := os.Remove(dst)
+		if err != nil {
+			log.Errorf("Error removing file: %v", err)
+			return err
+		}
+
+		return nil
+	}
+
+	if err := w.oss.Download(ctx, job.Args.Path, dst); err != nil {
+		log.Errorf("Error downloading file: %v", err)
+		return err
+	}
+
+	log.Infof("Successfully downloaded file: %s", job.Args.Path)
+
 	return nil
 }
-
-func (w *DownloadFromAliOSSWorker) NextRetry(job *Job[DownloadFromAliOSS]) time.Time {
-	return time.Now().Add(30 * time.Second)
-}

backend/app/jobs/download_from_alioss_test.go

@@ -0,0 +1,58 @@
+package jobs
+
+import (
+	"context"
+	"testing"
+
+	"quyun/app/service/testx"
+	"quyun/providers/ali"
+	"quyun/providers/job"
+
+	. "github.com/riverqueue/river"
+	. "github.com/smartystreets/goconvey/convey"
+	"github.com/stretchr/testify/suite"
+	_ "go.ipao.vip/atom"
+	"go.ipao.vip/atom/contracts"
+	"go.uber.org/dig"
+)
+
+type DownloadFromAliOSSSuiteInjectParams struct {
+	dig.In
+
+	Initials []contracts.Initial `group:"initials"` // nolint:structcheck
+	Job      *job.Job
+	Oss      *ali.OSSClient
+}
+
+type DownloadFromAliOSSSuite struct {
+	suite.Suite
+
+	DownloadFromAliOSSSuiteInjectParams
+}
+
+func Test_DownloadFromAliOSS(t *testing.T) {
+	providers := testx.Default().With(Provide)
+
+	testx.Serve(providers, t, func(p DownloadFromAliOSSSuiteInjectParams) {
+		suite.Run(t, &DownloadFromAliOSSSuite{DownloadFromAliOSSSuiteInjectParams: p})
+	})
+}
+
+func (t *DownloadFromAliOSSSuite) Test_Work() {
+	Convey("test_work", t.T(), func() {
+		Convey("step 1", func() {
+			job := &Job[DownloadFromAliOSS]{
+				Args: DownloadFromAliOSS{
+					Path: "quyun/959e5310105c96e653f10b74e5bdc36b.mp4",
+				},
+			}
+
+			worker := &DownloadFromAliOSSWorker{
+				oss: t.Oss,
+			}
+
+			err := worker.Work(context.Background(), job)
+			So(err, ShouldBeNil)
+		})
+	})
+}

backend/app/jobs/provider.gen.go

@@ -1,6 +1,7 @@
 package jobs
 
 import (
+	"quyun/providers/ali"
 	"quyun/providers/job"
 
 	"github.com/riverqueue/river"
@@ -39,8 +40,11 @@ func Provide(opts ...opt.Option) error {
 	}
 	if err := container.Container.Provide(func(
 		__job *job.Job,
+		oss *ali.OSSClient,
 	) (contracts.Initial, error) {
-		obj := &DownloadFromAliOSSWorker{}
+		obj := &DownloadFromAliOSSWorker{
+			oss: oss,
+		}
 		if err := river.AddWorkerSafely(__job.Workers, obj); err != nil {
 			return nil, err
 		}