85 lines
1.9 KiB
Go
85 lines
1.9 KiB
Go
package worker
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"feedsystem_video_go/internal/middleware/rabbitmq"
|
|
rediscache "feedsystem_video_go/internal/middleware/redis"
|
|
"feedsystem_video_go/internal/video"
|
|
"log"
|
|
|
|
amqp "github.com/rabbitmq/amqp091-go"
|
|
)
|
|
|
|
type PopularityWorker struct {
|
|
ch *amqp.Channel
|
|
cache *rediscache.Client
|
|
queue string
|
|
}
|
|
|
|
func NewPopularityWorker(ch *amqp.Channel, cache *rediscache.Client, queue string) *PopularityWorker {
|
|
return &PopularityWorker{ch: ch, cache: cache, queue: queue}
|
|
}
|
|
|
|
func (w *PopularityWorker) Run(ctx context.Context) error {
|
|
if w == nil || w.ch == nil || w.cache == nil {
|
|
return errors.New("popularity worker is not initialized")
|
|
}
|
|
if w.queue == "" {
|
|
return errors.New("queue is required")
|
|
}
|
|
|
|
deliveries, err := w.ch.Consume(
|
|
w.queue,
|
|
"",
|
|
false,
|
|
false,
|
|
false,
|
|
false,
|
|
nil,
|
|
)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for {
|
|
select {
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
case d, ok := <-deliveries:
|
|
if !ok {
|
|
return errors.New("deliveries channel closed")
|
|
}
|
|
w.handleDelivery(ctx, d)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (w *PopularityWorker) handleDelivery(ctx context.Context, d amqp.Delivery) {
|
|
if err := w.process(ctx, d.Body); err != nil {
|
|
retryCount := rabbitmq.GetRetryCount(d)
|
|
if retryCount >= rabbitmq.MaxRetryCount {
|
|
log.Printf("popularity worker: max retries exceeded (%d), moving to DLX: %v", retryCount, err)
|
|
_ = d.Ack(false)
|
|
return
|
|
}
|
|
log.Printf("popularity worker: failed (retry %d/%d): %v", retryCount+1, rabbitmq.MaxRetryCount, err)
|
|
_ = d.Nack(false, true)
|
|
return
|
|
}
|
|
_ = d.Ack(false)
|
|
}
|
|
|
|
func (w *PopularityWorker) process(ctx context.Context, body []byte) error {
|
|
var evt rabbitmq.PopularityEvent
|
|
if err := json.Unmarshal(body, &evt); err != nil {
|
|
return nil
|
|
}
|
|
if evt.VideoID == 0 || evt.Change == 0 {
|
|
return nil
|
|
}
|
|
video.UpdatePopularityCache(ctx, w.cache, evt.VideoID, evt.Change)
|
|
return nil
|
|
}
|