feat(kol): AI assist service + worker
This commit is contained in:
@@ -0,0 +1,51 @@
|
||||
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
|
||||
}
|
||||
Reference in New Issue
Block a user