52 lines
1.2 KiB
Go
52 lines
1.2 KiB
Go
package app
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/geo-platform/tenant-api/internal/shared/messaging/rabbitmq"
|
|
)
|
|
|
|
type kolAssistJob struct {
|
|
TaskID string `json:"task_id"`
|
|
TenantID int64 `json:"tenant_id"`
|
|
}
|
|
|
|
type KolAssistJob = kolAssistJob
|
|
|
|
func publishKolAssistTask(ctx context.Context, client *rabbitmq.Client, job kolAssistJob) error {
|
|
if client == nil {
|
|
return fmt.Errorf("rabbitmq client is not configured")
|
|
}
|
|
|
|
payload, err := json.Marshal(job)
|
|
if err != nil {
|
|
return fmt.Errorf("marshal kol assist job: %w", err)
|
|
}
|
|
|
|
if err := client.PublishKolAssistTask(ctx, payload); err != nil {
|
|
return fmt.Errorf("publish kol assist task: %w", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func decodeKolAssistTask(payload []byte) (*kolAssistJob, error) {
|
|
var job kolAssistJob
|
|
if err := json.Unmarshal(payload, &job); err != nil {
|
|
return nil, fmt.Errorf("decode kol assist job: %w", err)
|
|
}
|
|
if job.TaskID == "" || job.TenantID <= 0 {
|
|
return nil, fmt.Errorf("kol assist job is incomplete")
|
|
}
|
|
return &job, nil
|
|
}
|
|
|
|
func DecodeKolAssistTask(payload []byte) (*KolAssistJob, error) {
|
|
job, err := decodeKolAssistTask(payload)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return job, nil
|
|
}
|