mirror of
https://github.com/SamyRai/tercul-backend.git
synced 2025-12-27 05:11:34 +00:00
20da2199ba
- Fixes a potential panic in the background job queue by changing `log.Fatalf` to `log.Printf`, allowing for more graceful error handling.
- Implements all `panic("not implemented")` methods in the mock repositories for `Like`, `Work`, and `User`, enabling more robust testing.
- Consolidates duplicated `WorkAnalytics` and `TranslationAnalytics` structs into a central `internal/domain/analytics` package to reduce code duplication and improve maintainability.
- Corrects build errors that arose during testing, including an unused import and an incorrect struct field name in a mock repository.
70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package sync
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"time"
|
|
|
|
"github.com/hibiken/asynq"
|
|
)
|
|
|
|
// enqueueTask is a generic function to enqueue tasks with payload
|
|
func enqueueTask[T any](client *asynq.Client, taskType string, payload T, delay time.Duration) error {
|
|
data, err := json.Marshal(payload)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
task := asynq.NewTask(taskType, data)
|
|
_, err = client.Enqueue(task, asynq.ProcessIn(delay))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// EnqueueFullSync enqueues a full sync task
|
|
func EnqueueFullSync(client *asynq.Client) error {
|
|
payload := SyncPayload{Full: true}
|
|
if err := enqueueTask(client, TaskFullSync, payload, DefaultFullSyncDelay); err != nil {
|
|
return err
|
|
}
|
|
log.Println("Enqueued full sync task.")
|
|
return nil
|
|
}
|
|
|
|
// EnqueueEntitySync enqueues an entity sync task
|
|
func EnqueueEntitySync(client *asynq.Client, className string) error {
|
|
payload := EntitySyncPayload{ClassName: className}
|
|
if err := enqueueTask(client, TaskEntitySync, payload, DefaultEntityDelay); err != nil {
|
|
return err
|
|
}
|
|
log.Printf("Enqueued entity sync task for class %s.", className)
|
|
return nil
|
|
}
|
|
|
|
// EnqueueEdgeSync enqueues an edge sync task
|
|
func EnqueueEdgeSync(client *asynq.Client, batchSize, offset int) error {
|
|
payload := EdgeSyncPayload{
|
|
BatchSize: batchSize,
|
|
Offset: offset,
|
|
}
|
|
if err := enqueueTask(client, TaskEdgeSync, payload, DefaultEdgeDelay); err != nil {
|
|
return err
|
|
}
|
|
log.Printf("Enqueued edge sync task (offset %d, batch size %d).", offset, batchSize)
|
|
return nil
|
|
}
|
|
|
|
// RegisterQueueHandlers registers all sync job handlers with the Asynq server
|
|
func RegisterQueueHandlers(srv *asynq.Server, syncJob *SyncJob) {
|
|
mux := asynq.NewServeMux()
|
|
mux.HandleFunc(TaskFullSync, syncJob.HandleFullSync)
|
|
mux.HandleFunc(TaskEntitySync, syncJob.HandleEntitySync)
|
|
mux.HandleFunc(TaskEdgeSync, syncJob.HandleEdgeSync)
|
|
if err := srv.Run(mux); err != nil {
|
|
log.Printf("Failed to start asynq server: %v", err)
|
|
}
|
|
}
|