Init
This commit is contained in:
@@ -0,0 +1,137 @@
|
||||
package services
|
||||
|
||||
import (
|
||||
"cs-agent/internal/models"
|
||||
"cs-agent/internal/pkg/enums"
|
||||
"cs-agent/internal/repositories"
|
||||
"encoding/json"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/mlogclub/simple/sqls"
|
||||
"github.com/mlogclub/simple/web/params"
|
||||
)
|
||||
|
||||
var ChannelMessageOutboxService = newChannelMessageOutboxService()
|
||||
|
||||
func newChannelMessageOutboxService() *channelMessageOutboxService {
|
||||
return &channelMessageOutboxService{}
|
||||
}
|
||||
|
||||
type channelMessageOutboxService struct {
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) Get(id int64) *models.ChannelMessageOutbox {
|
||||
return repositories.ChannelMessageOutboxRepository.Get(sqls.DB(), id)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) Take(where ...interface{}) *models.ChannelMessageOutbox {
|
||||
return repositories.ChannelMessageOutboxRepository.Take(sqls.DB(), where...)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) Find(cnd *sqls.Cnd) []models.ChannelMessageOutbox {
|
||||
return repositories.ChannelMessageOutboxRepository.Find(sqls.DB(), cnd)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) FindOne(cnd *sqls.Cnd) *models.ChannelMessageOutbox {
|
||||
return repositories.ChannelMessageOutboxRepository.FindOne(sqls.DB(), cnd)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) FindPageByParams(params *params.QueryParams) (list []models.ChannelMessageOutbox, paging *sqls.Paging) {
|
||||
return repositories.ChannelMessageOutboxRepository.FindPageByParams(sqls.DB(), params)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) FindPageByCnd(cnd *sqls.Cnd) (list []models.ChannelMessageOutbox, paging *sqls.Paging) {
|
||||
return repositories.ChannelMessageOutboxRepository.FindPageByCnd(sqls.DB(), cnd)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) Count(cnd *sqls.Cnd) int64 {
|
||||
return repositories.ChannelMessageOutboxRepository.Count(sqls.DB(), cnd)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) Create(t *models.ChannelMessageOutbox) error {
|
||||
return repositories.ChannelMessageOutboxRepository.Create(sqls.DB(), t)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) Update(t *models.ChannelMessageOutbox) error {
|
||||
return repositories.ChannelMessageOutboxRepository.Update(sqls.DB(), t)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) Updates(id int64, columns map[string]interface{}) error {
|
||||
return repositories.ChannelMessageOutboxRepository.Updates(sqls.DB(), id, columns)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) UpdateColumn(id int64, name string, value interface{}) error {
|
||||
return repositories.ChannelMessageOutboxRepository.UpdateColumn(sqls.DB(), id, name, value)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) Delete(id int64) {
|
||||
repositories.ChannelMessageOutboxRepository.Delete(sqls.DB(), id)
|
||||
}
|
||||
|
||||
// GetByMessageID retrieves the outbox entry by message ID and channel type.
|
||||
func (s *channelMessageOutboxService) GetByMessageID(channelType string, messageID int64) *models.ChannelMessageOutbox {
|
||||
return repositories.ChannelMessageOutboxRepository.Take(sqls.DB(), "channel_type = ? AND message_id = ?", channelType, messageID)
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) EnqueueWxWorkKFMessage(conversation *models.Conversation, message *models.Message) error {
|
||||
if conversation == nil || message == nil {
|
||||
return nil
|
||||
}
|
||||
if conversation.ExternalSource != enums.ExternalSourceWxWorkKF {
|
||||
return nil
|
||||
}
|
||||
if message.SenderType != enums.IMSenderTypeAgent && message.SenderType != enums.IMSenderTypeAI {
|
||||
return nil
|
||||
}
|
||||
if message.MessageType != enums.IMMessageTypeText && message.MessageType != enums.IMMessageTypeHTML {
|
||||
return nil
|
||||
}
|
||||
if existing := s.GetByMessageID(enums.ChannelTypeWxWorkKF, message.ID); existing != nil {
|
||||
return nil
|
||||
}
|
||||
|
||||
payload, err := json.Marshal(map[string]any{
|
||||
"conversationId": conversation.ID,
|
||||
"messageId": message.ID,
|
||||
"messageType": message.MessageType,
|
||||
"content": strings.TrimSpace(message.Content),
|
||||
"payload": strings.TrimSpace(message.Payload),
|
||||
"senderId": message.SenderID,
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
now := time.Now()
|
||||
return s.Create(&models.ChannelMessageOutbox{
|
||||
ChannelType: enums.ChannelTypeWxWorkKF,
|
||||
ConversationID: conversation.ID,
|
||||
MessageID: message.ID,
|
||||
Payload: string(payload),
|
||||
SendStatus: string(enums.ChannelMessageOutboxStatusPending),
|
||||
AuditFields: models.AuditFields{
|
||||
CreatedAt: now,
|
||||
CreateUserID: message.UpdateUserID,
|
||||
CreateUserName: message.UpdateUserName,
|
||||
UpdatedAt: now,
|
||||
UpdateUserID: message.UpdateUserID,
|
||||
UpdateUserName: message.UpdateUserName,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
func (s *channelMessageOutboxService) ListPending(channelType string, limit int) []models.ChannelMessageOutbox {
|
||||
if limit <= 0 {
|
||||
limit = 20
|
||||
}
|
||||
cnd := sqls.NewCnd().
|
||||
Eq("channel_type", strings.TrimSpace(channelType)).
|
||||
In("send_status", []string{
|
||||
string(enums.ChannelMessageOutboxStatusPending),
|
||||
string(enums.ChannelMessageOutboxStatusFailed),
|
||||
}).
|
||||
Asc("id").
|
||||
Limit(limit)
|
||||
return s.Find(cnd)
|
||||
}
|
||||
Reference in New Issue
Block a user