626fc700d2
♻️ refactor(outbox): 抽离通用事件总线，并完成 event_type-only 收口 - ✨ 新增 `infra` 层通用 `EventBus` / `EventContract`，统一事件发布与消费协议 - 🔄 将聊天持久化链路调整为通过 `service/events` 注册 handler 并发布事件，进一步解耦业务逻辑与异步处理流程 - 🧹 移除 `chat_history_async` 旧适配实现，以及基于 `biz_type` 的兼容分发逻辑 - 📝 更新 Outbox 异步持久化决策记录，明确保留方案 A，并正式启用方案 B - 📚 同步更新 README 中关于 Outbox + Kafka 可靠异步链路的说明 - 🚚 当前 `outbox + kafka` 已与项目业务链路完全解耦，沉淀为通用、可靠性更强的消息队列能力；后续将参考消息队列的典型使用方式，逐步扩展到更多业务场景 - ✨ 补充跨不同分类事务管理器中的 `agent dao` 注册与接入支持
105 lines
3.7 KiB
Go
105 lines
3.7 KiB
Go
package events
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"errors"
|
||
|
||
"github.com/LoveLosita/smartflow/backend/dao"
|
||
kafkabus "github.com/LoveLosita/smartflow/backend/infra/kafka"
|
||
outboxinfra "github.com/LoveLosita/smartflow/backend/infra/outbox"
|
||
"github.com/LoveLosita/smartflow/backend/model"
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
const (
|
||
// EventTypeChatHistoryPersistRequested 是“聊天消息持久化请求”的业务事件类型。
|
||
//
|
||
// 命名策略：
|
||
// 1. 只描述业务语义，不包含 outbox/kafka 等实现词；
|
||
// 2. 作为新路由键长期保留，后续协议变化优先走 event_version；
|
||
// 3. 旧路由键仅作兼容，不再作为新发布默认值。
|
||
EventTypeChatHistoryPersistRequested = "chat.history.persist.requested"
|
||
)
|
||
|
||
// RegisterChatHistoryPersistHandler 注册“聊天消息持久化”消费者处理器。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只负责聊天事件，不处理其他业务事件；
|
||
// 2. 只负责注册，不负责总线启停；
|
||
// 3. 通过 outbox 通用事务入口把“业务写入 + consumed 推进”合并为一个事务；
|
||
// 4. 当前版本仅注册新路由键（chat.history.persist.requested），不再注册旧兼容键。
|
||
func RegisterChatHistoryPersistHandler(
|
||
bus *outboxinfra.EventBus,
|
||
outboxRepo *outboxinfra.Repository,
|
||
repoManager *dao.RepoManager,
|
||
) error {
|
||
// 1. 依赖校验：任何一个关键依赖为空都无法安全处理消息。
|
||
if bus == nil {
|
||
return errors.New("event bus is nil")
|
||
}
|
||
if outboxRepo == nil {
|
||
return errors.New("outbox repository is nil")
|
||
}
|
||
if repoManager == nil {
|
||
return errors.New("repo manager is nil")
|
||
}
|
||
|
||
// 2. 定义统一处理器：
|
||
// 2.1 解析 payload；
|
||
// 2.2 调用 outbox 通用消费事务；
|
||
// 2.3 在事务回调中复用 RepoManager.WithTx 执行业务 DAO 写入。
|
||
handler := func(ctx context.Context, envelope kafkabus.Envelope) error {
|
||
var payload model.ChatHistoryPersistPayload
|
||
if unmarshalErr := json.Unmarshal(envelope.Payload, &payload); unmarshalErr != nil {
|
||
// 2.1 payload 非法属于不可恢复错误，直接标 dead，避免无意义重试。
|
||
_ = outboxRepo.MarkDead(ctx, envelope.OutboxID, "解析聊天持久化载荷失败: "+unmarshalErr.Error())
|
||
return nil
|
||
}
|
||
|
||
// 2.2 使用 outbox 通用消费事务，保证“业务写入 + consumed 状态推进”原子一致。
|
||
return outboxRepo.ConsumeAndMarkConsumed(ctx, envelope.OutboxID, func(tx *gorm.DB) error {
|
||
// 2.2.1 基于同一个 tx 构造 RepoManager，复用你现有跨包事务模型。
|
||
txM := repoManager.WithTx(tx)
|
||
// 2.2.2 在同事务内写入聊天历史与会话计数。
|
||
return txM.Agent.SaveChatHistoryInTx(
|
||
ctx,
|
||
payload.UserID,
|
||
payload.ConversationID,
|
||
payload.Role,
|
||
payload.Message,
|
||
)
|
||
})
|
||
}
|
||
|
||
// 3. 注册新路由键（主路由）。
|
||
if err := bus.RegisterEventHandler(EventTypeChatHistoryPersistRequested, handler); err != nil {
|
||
return err
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// PublishChatHistoryPersistRequested 发布“聊天消息持久化请求”事件。
|
||
//
|
||
// 设计目的：
|
||
// 1. 让业务层只传 DTO，不重复拼事件元数据；
|
||
// 2. 统一消息键策略（conversation_id 作为 MessageKey/AggregateID）；
|
||
// 3. 发布失败时显式返回 error，由调用方决定是否降级到同步写库。
|
||
func PublishChatHistoryPersistRequested(
|
||
ctx context.Context,
|
||
publisher outboxinfra.EventPublisher,
|
||
payload model.ChatHistoryPersistPayload,
|
||
) error {
|
||
if publisher == nil {
|
||
return errors.New("event publisher is nil")
|
||
}
|
||
return publisher.Publish(ctx, outboxinfra.PublishRequest{
|
||
EventType: EventTypeChatHistoryPersistRequested,
|
||
EventVersion: outboxinfra.DefaultEventVersion,
|
||
MessageKey: payload.ConversationID,
|
||
AggregateID: payload.ConversationID,
|
||
Payload: payload,
|
||
})
|
||
}
|