add jobs
This commit is contained in:
23
backend/app/events/publishers/post_created.go
Normal file
23
backend/app/events/publishers/post_created.go
Normal file
@@ -0,0 +1,23 @@
|
||||
package publishers
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
|
||||
"backend/app/events"
|
||||
|
||||
"git.ipao.vip/rogeecn/atom/contracts"
|
||||
)
|
||||
|
||||
var _ contracts.EventPublisher = (*UserRegister)(nil)
|
||||
|
||||
type PostCreated struct {
|
||||
ID int64 `json:"id"`
|
||||
}
|
||||
|
||||
func (e *PostCreated) Marshal() ([]byte, error) {
|
||||
return json.Marshal(e)
|
||||
}
|
||||
|
||||
func (e *PostCreated) Topic() string {
|
||||
return events.TopicPostCreated.String()
|
||||
}
|
||||
99
backend/app/events/subscribers/post_created.go
Normal file
99
backend/app/events/subscribers/post_created.go
Normal file
@@ -0,0 +1,99 @@
|
||||
package subscribers
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
|
||||
"backend/app/events"
|
||||
"backend/app/events/publishers"
|
||||
"backend/app/http/posts"
|
||||
"backend/app/jobs"
|
||||
"backend/database/fields"
|
||||
"backend/providers/job"
|
||||
|
||||
"git.ipao.vip/rogeecn/atom/contracts"
|
||||
"github.com/ThreeDotsLabs/watermill/message"
|
||||
"github.com/pkg/errors"
|
||||
"github.com/samber/lo"
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
var _ contracts.EventHandler = (*PostCreated)(nil)
|
||||
|
||||
// @provider(event)
|
||||
type PostCreated struct {
|
||||
log *logrus.Entry `inject:"false"`
|
||||
|
||||
postSvc *posts.Service
|
||||
job *job.Job
|
||||
}
|
||||
|
||||
func (e *PostCreated) Prepare() error {
|
||||
e.log = logrus.WithField("module", "events.subscribers.user_register")
|
||||
return nil
|
||||
}
|
||||
|
||||
// PublishToTopic implements contracts.EventHandler.
|
||||
func (e *PostCreated) PublishToTopic() string {
|
||||
return events.TopicProcessed.String()
|
||||
}
|
||||
|
||||
// Topic implements contracts.EventHandler.
|
||||
func (e *PostCreated) Topic() string {
|
||||
return events.TopicPostCreated.String()
|
||||
}
|
||||
|
||||
// Handler implements contracts.EventHandler.
|
||||
func (e *PostCreated) Handler(msg *message.Message) ([]*message.Message, error) {
|
||||
var payload publishers.PostCreated
|
||||
err := json.Unmarshal(msg.Payload, &payload)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
e.log.Infof("received event %s", msg.Payload)
|
||||
|
||||
post, err := e.postSvc.GetPostByID(context.Background(), payload.ID)
|
||||
if err != nil {
|
||||
return nil, errors.Wrapf(err, "failed to get user by id: %d", payload.ID)
|
||||
}
|
||||
|
||||
video, ok := lo.Find(post.Assets.Data, func(asset fields.MediaAsset) bool {
|
||||
return asset.Type == fields.MediaAssetTypeVideo
|
||||
})
|
||||
|
||||
if !ok {
|
||||
e.log.Warnf("post %d %s has no video asset", post.ID, post.Title)
|
||||
return nil, nil
|
||||
}
|
||||
|
||||
_ = video
|
||||
|
||||
job, err := e.job.Client()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// cut video
|
||||
_, err = job.Insert(context.Background(), jobs.VideoCut{
|
||||
PostID: post.ID,
|
||||
Hash: video.Hash,
|
||||
TenantID: post.TenantID,
|
||||
UserID: post.UserID,
|
||||
}, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// extract audio
|
||||
_, err = job.Insert(context.Background(), jobs.VideoExtractAudio{
|
||||
PostID: post.ID,
|
||||
Hash: video.Hash,
|
||||
TenantID: post.TenantID,
|
||||
UserID: post.UserID,
|
||||
}, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return nil, nil
|
||||
}
|
||||
@@ -18,6 +18,8 @@ const (
|
||||
TopicProcessed Topic = "event:processed"
|
||||
// TopicUserRegister is a Topic of type UserRegister.
|
||||
TopicUserRegister Topic = "user:register"
|
||||
// TopicPostCreated is a Topic of type PostCreated.
|
||||
TopicPostCreated Topic = "post:created"
|
||||
)
|
||||
|
||||
var ErrInvalidTopic = fmt.Errorf("not a valid Topic, try [%s]", strings.Join(_TopicNames, ", "))
|
||||
@@ -25,6 +27,7 @@ var ErrInvalidTopic = fmt.Errorf("not a valid Topic, try [%s]", strings.Join(_To
|
||||
var _TopicNames = []string{
|
||||
string(TopicProcessed),
|
||||
string(TopicUserRegister),
|
||||
string(TopicPostCreated),
|
||||
}
|
||||
|
||||
// TopicNames returns a list of possible string values of Topic.
|
||||
@@ -39,6 +42,7 @@ func TopicValues() []Topic {
|
||||
return []Topic{
|
||||
TopicProcessed,
|
||||
TopicUserRegister,
|
||||
TopicPostCreated,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -57,6 +61,7 @@ func (x Topic) IsValid() bool {
|
||||
var _TopicValue = map[string]Topic{
|
||||
"event:processed": TopicProcessed,
|
||||
"user:register": TopicUserRegister,
|
||||
"post:created": TopicPostCreated,
|
||||
}
|
||||
|
||||
// ParseTopic attempts to convert a string to a Topic.
|
||||
|
||||
@@ -5,6 +5,7 @@ package events
|
||||
//
|
||||
// Processed = "event:processed"
|
||||
// UserRegister = "user:register"
|
||||
// PostCreated = "post:created"
|
||||
//
|
||||
// )
|
||||
type Topic string
|
||||
|
||||
@@ -174,12 +174,13 @@ func (ctl *Controller) Create(ctx fiber.Ctx, claim *jwt.Claims, tenantSlug strin
|
||||
Title: body.Title,
|
||||
Description: body.Description,
|
||||
Content: body.Content,
|
||||
Stage: fields.PostStagePending,
|
||||
Status: fields.PostStatusPending,
|
||||
Price: body.Price,
|
||||
Discount: body.Discount,
|
||||
Assets: body.Assets,
|
||||
Tags: body.Tags,
|
||||
|
||||
Stage: fields.PostStagePending,
|
||||
Status: fields.PostStatusPending,
|
||||
}
|
||||
|
||||
if err := ctl.svc.Create(ctx.Context(), tenant, user, post); err != nil {
|
||||
|
||||
52
backend/app/jobs/VideoCut.go
Normal file
52
backend/app/jobs/VideoCut.go
Normal file
@@ -0,0 +1,52 @@
|
||||
package jobs
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
_ "git.ipao.vip/rogeecn/atom"
|
||||
_ "git.ipao.vip/rogeecn/atom/contracts"
|
||||
. "github.com/riverqueue/river"
|
||||
)
|
||||
|
||||
var (
|
||||
_ JobArgs = VideoCut{}
|
||||
_ JobArgsWithInsertOpts = VideoCut{}
|
||||
)
|
||||
|
||||
type VideoCut struct {
|
||||
PostID int64 `json:"post_id"`
|
||||
Hash string `json:"hash"`
|
||||
TenantID int64 `json:"tenant_id"`
|
||||
UserID int64 `json:"user_id"`
|
||||
}
|
||||
|
||||
// InsertOpts implements JobArgsWithInsertOpts.
|
||||
func (s VideoCut) InsertOpts() InsertOpts {
|
||||
return InsertOpts{
|
||||
Queue: QueueDefault,
|
||||
Priority: PriorityDefault,
|
||||
UniqueOpts: UniqueOpts{
|
||||
ByArgs: true,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (VideoCut) Kind() string {
|
||||
return "VideoCut"
|
||||
}
|
||||
|
||||
var _ Worker[VideoCut] = (*VideoCutWorker)(nil)
|
||||
|
||||
// @provider(job)
|
||||
type VideoCutWorker struct {
|
||||
WorkerDefaults[VideoCut]
|
||||
}
|
||||
|
||||
func (w *VideoCutWorker) NextRetry(job *Job[VideoCut]) time.Time {
|
||||
return time.Now().Add(5 * time.Second)
|
||||
}
|
||||
|
||||
func (w *VideoCutWorker) Work(ctx context.Context, job *Job[VideoCut]) error {
|
||||
return nil
|
||||
}
|
||||
52
backend/app/jobs/VideoExtractAudio.go
Normal file
52
backend/app/jobs/VideoExtractAudio.go
Normal file
@@ -0,0 +1,52 @@
|
||||
package jobs
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
|
||||
_ "git.ipao.vip/rogeecn/atom"
|
||||
_ "git.ipao.vip/rogeecn/atom/contracts"
|
||||
. "github.com/riverqueue/river"
|
||||
)
|
||||
|
||||
var (
|
||||
_ JobArgs = VideoExtractAudio{}
|
||||
_ JobArgsWithInsertOpts = VideoExtractAudio{}
|
||||
)
|
||||
|
||||
type VideoExtractAudio struct {
|
||||
PostID int64 `json:"post_id"`
|
||||
Hash string `json:"hash"`
|
||||
TenantID int64 `json:"tenant_id"`
|
||||
UserID int64 `json:"user_id"`
|
||||
}
|
||||
|
||||
// InsertOpts implements JobArgsWithInsertOpts.
|
||||
func (s VideoExtractAudio) InsertOpts() InsertOpts {
|
||||
return InsertOpts{
|
||||
Queue: QueueDefault,
|
||||
Priority: PriorityDefault,
|
||||
UniqueOpts: UniqueOpts{
|
||||
ByArgs: true,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (VideoExtractAudio) Kind() string {
|
||||
return "VideoExtractAudio"
|
||||
}
|
||||
|
||||
var _ Worker[VideoExtractAudio] = (*VideoExtractAudioWorker)(nil)
|
||||
|
||||
// @provider(job)
|
||||
type VideoExtractAudioWorker struct {
|
||||
WorkerDefaults[VideoExtractAudio]
|
||||
}
|
||||
|
||||
func (w *VideoExtractAudioWorker) NextRetry(job *Job[VideoExtractAudio]) time.Time {
|
||||
return time.Now().Add(5 * time.Minute)
|
||||
}
|
||||
|
||||
func (w *VideoExtractAudioWorker) Work(ctx context.Context, job *Job[VideoExtractAudio]) error {
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user