110 lines
2.8 KiB
Go
110 lines
2.8 KiB
Go
package jobs
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"quyun/v2/app/services"
|
|
"quyun/v2/database/models"
|
|
"quyun/v2/pkg/fields"
|
|
"quyun/v2/pkg/utils"
|
|
"quyun/v2/providers/ali"
|
|
"quyun/v2/providers/app"
|
|
"quyun/v2/providers/job"
|
|
|
|
"github.com/pkg/errors"
|
|
. "github.com/riverqueue/river"
|
|
"github.com/samber/lo"
|
|
log "github.com/sirupsen/logrus"
|
|
_ "go.ipao.vip/atom"
|
|
"go.ipao.vip/atom/contracts"
|
|
"go.ipao.vip/gen/types"
|
|
)
|
|
|
|
var _ contracts.JobArgs = (*PublishDraftPosts)(nil)
|
|
|
|
type PublishDraftPosts struct {
|
|
MediaHash string `json:"media_hash"`
|
|
}
|
|
|
|
func (s PublishDraftPosts) InsertOpts() InsertOpts {
|
|
return InsertOpts{
|
|
Queue: QueueDefault,
|
|
Priority: PriorityDefault,
|
|
}
|
|
}
|
|
|
|
func (s PublishDraftPosts) Kind() string { return "publish_draft_posts" }
|
|
func (a PublishDraftPosts) UniqueID() string { return a.Kind() }
|
|
|
|
var _ Worker[PublishDraftPosts] = (*PublishDraftPostsWorker)(nil)
|
|
|
|
// @provider(job)
|
|
type PublishDraftPostsWorker struct {
|
|
WorkerDefaults[PublishDraftPosts]
|
|
|
|
oss *ali.OSSClient
|
|
job *job.Job
|
|
app *app.Config
|
|
}
|
|
|
|
func (w *PublishDraftPostsWorker) NextRetry(job *Job[PublishDraftPosts]) time.Time {
|
|
return time.Now().Add(30 * time.Second)
|
|
}
|
|
|
|
func (w *PublishDraftPostsWorker) Work(ctx context.Context, job *Job[PublishDraftPosts]) error {
|
|
log := log.WithField("job", job.Args.Kind())
|
|
|
|
log.Infof("[Start] Working on job with strings: %+v", job.Args)
|
|
defer log.Infof("[End] Finished %s", job.Args.Kind())
|
|
|
|
media, err := services.Media.GetByHash(ctx, job.Args.MediaHash)
|
|
if err != nil {
|
|
log.Errorf("Error getting media by ID: %v", err)
|
|
return JobCancel(err)
|
|
}
|
|
|
|
relationMedias, err := services.Media.GetRelations(ctx, media.Hash)
|
|
if err != nil {
|
|
log.Errorf("Error getting relation medias: %v", err)
|
|
return JobCancel(err)
|
|
}
|
|
|
|
assets := lo.FilterMap(relationMedias, func(media *models.Media, _ int) (fields.MediaAsset, bool) {
|
|
return fields.MediaAsset{
|
|
Type: media.MimeType,
|
|
Media: media.ID,
|
|
Metas: lo.ToPtr(media.Metas.Data()),
|
|
}, media.MimeType != "image/jpeg"
|
|
})
|
|
assets = append(assets, fields.MediaAsset{
|
|
Type: media.MimeType,
|
|
Media: media.ID,
|
|
Metas: lo.ToPtr(media.Metas.Data()),
|
|
})
|
|
|
|
// publish a draft posts
|
|
post := &models.Post{
|
|
Status: fields.PostStatusDraft,
|
|
Title: utils.FormatTitle(media.Name),
|
|
Description: "",
|
|
Content: "",
|
|
Price: 0,
|
|
Discount: 100,
|
|
Views: 0,
|
|
Likes: 0,
|
|
Tags: types.NewJSONType([]string{}),
|
|
Assets: types.NewJSONType(assets),
|
|
HeadImages: types.NewJSONType(lo.FilterMap(relationMedias, func(media *models.Media, _ int) (int64, bool) {
|
|
return media.ID, media.MimeType == "image/jpeg"
|
|
})),
|
|
}
|
|
if err := post.Create(ctx); err != nil {
|
|
log.Errorf("Error creating post: %v", err)
|
|
return errors.Wrap(err, "create post")
|
|
}
|
|
log.Infof("Post created successfully with ID: %d", post.ID)
|
|
|
|
return nil
|
|
}
|