wehub-resource-sync
f36e2104d8
tests / ragflow_tests_infinity (push) Has been cancelled
tests / ragflow_tests_elasticsearch (push) Has been cancelled
sep-tests / ragflow_preflight (push) Has been cancelled
sep-tests / ragflow_tests_infinity (push) Has been cancelled
sep-tests / ragflow_tests_elasticsearch (push) Has been cancelled
tests / ragflow_preflight (push) Has been cancelled
32 lines
724 B
Go
32 lines
724 B
Go
package service
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"ragflow/internal/common"
|
|
"ragflow/internal/engine"
|
|
)
|
|
|
|
type TaskPublisher interface {
|
|
PublishTaskMessage(subject string, msg common.TaskMessage) error
|
|
}
|
|
|
|
type MessageQueueTaskPublisher struct{}
|
|
|
|
func NewMessageQueueTaskPublisher() *MessageQueueTaskPublisher {
|
|
return &MessageQueueTaskPublisher{}
|
|
}
|
|
|
|
func (p *MessageQueueTaskPublisher) PublishTaskMessage(subject string, msg common.TaskMessage) error {
|
|
msgQueueEngine := engine.GetMessageQueueEngine()
|
|
if msgQueueEngine == nil {
|
|
return fmt.Errorf("message queue engine is not initialized")
|
|
}
|
|
payload, err := json.Marshal(msg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return msgQueueEngine.PublishTask(subject, payload)
|
|
}
|