a6c1e5d077
后端： 1. 服务级 outbox 基础设施全量落地——新增 service route / service catalog / route registry，重构 outbox engine、repository、event bus 和 model，按 `event_type -> service -> table/topic/group` 统一写入与投递，保留 `agent` 兼容壳但不再依赖共享 outbox 2. Kafka 投递、消费与启动装配同步切换——更新 kafka config、consumer、envelope，接入服务级 topic 与 consumer group，并同步调整 mysql 初始化、start/main/router 装配，保证各服务 relay / consumer 独立装配 3. 业务事件处理器按服务归属重接新 bus——`active-scheduler` 触发链路，以及 `agent` / `memory` / `notification` / `task` 相关 outbox handler 统一切到新路由注册与服务目录，避免新流量回流共享表 4. 同步更新《微服务四步迁移与第二阶段并行开发计划》，把阶段 1 改成当前基线并补齐结构图、阶段快照、风险回退和多代理执行口径
106 lines
3.5 KiB
Go
106 lines
3.5 KiB
Go
package events
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"errors"
|
||
"strconv"
|
||
"time"
|
||
|
||
"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 (
|
||
// EventTypeChatTokenUsageAdjustRequested 是“会话 token 账本增量调整”事件类型。
|
||
//
|
||
// 命名约束：
|
||
// 1. 仅表达业务语义，不泄露 outbox/kafka 实现细节；
|
||
// 2. 作为稳定路由键长期保留，后续演进优先通过 event_version。
|
||
EventTypeChatTokenUsageAdjustRequested = "chat.token.usage.adjust.requested"
|
||
)
|
||
|
||
// RegisterChatTokenUsageAdjustHandler 注册“会话 token 账本增量调整”消费者。
|
||
//
|
||
// 职责边界：
|
||
// 1. 只处理 token 调整事件，不处理聊天正文落库；
|
||
// 2. 通过 outbox 统一消费事务入口，保证“业务成功 + consumed 推进”原子一致；
|
||
// 3. 非法载荷直接标记 dead，避免无意义重试。
|
||
func RegisterChatTokenUsageAdjustHandler(
|
||
bus OutboxBus,
|
||
outboxRepo *outboxinfra.Repository,
|
||
repoManager *dao.RepoManager,
|
||
) error {
|
||
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")
|
||
}
|
||
eventOutboxRepo, err := scopedOutboxRepoForEvent(outboxRepo, EventTypeChatTokenUsageAdjustRequested)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
handler := func(ctx context.Context, envelope kafkabus.Envelope) error {
|
||
var payload model.ChatTokenUsageAdjustPayload
|
||
if unmarshalErr := json.Unmarshal(envelope.Payload, &payload); unmarshalErr != nil {
|
||
_ = eventOutboxRepo.MarkDead(ctx, envelope.OutboxID, "解析会话 token 调整载荷失败: "+unmarshalErr.Error())
|
||
return nil
|
||
}
|
||
|
||
if payload.UserID <= 0 || payload.TokensDelta <= 0 || payload.ConversationID == "" {
|
||
_ = eventOutboxRepo.MarkDead(ctx, envelope.OutboxID, "会话 token 调整载荷无效: user_id/conversation_id/tokens_delta 非法")
|
||
return nil
|
||
}
|
||
|
||
return eventOutboxRepo.ConsumeAndMarkConsumed(ctx, envelope.OutboxID, func(tx *gorm.DB) error {
|
||
txM := repoManager.WithTx(tx)
|
||
return txM.Agent.AdjustTokenUsageInTx(ctx, payload.UserID, payload.ConversationID, payload.TokensDelta)
|
||
})
|
||
}
|
||
|
||
return bus.RegisterEventHandler(EventTypeChatTokenUsageAdjustRequested, handler)
|
||
}
|
||
|
||
// PublishChatTokenUsageAdjustRequested 发布“会话 token 账本增量调整”事件。
|
||
//
|
||
// 说明：
|
||
// 1. 只保证“写入 outbox 成功”，不等待消费完成；
|
||
// 2. 业务层只传 DTO，不关心 outbox/kafka 协议细节。
|
||
func PublishChatTokenUsageAdjustRequested(
|
||
ctx context.Context,
|
||
publisher outboxinfra.EventPublisher,
|
||
payload model.ChatTokenUsageAdjustPayload,
|
||
) error {
|
||
if publisher == nil {
|
||
return errors.New("event publisher is nil")
|
||
}
|
||
if payload.UserID <= 0 {
|
||
return errors.New("invalid user_id")
|
||
}
|
||
if payload.TokensDelta <= 0 {
|
||
return errors.New("invalid tokens_delta")
|
||
}
|
||
if payload.ConversationID == "" {
|
||
return errors.New("invalid conversation_id")
|
||
}
|
||
if payload.TriggeredAt.IsZero() {
|
||
payload.TriggeredAt = time.Now()
|
||
}
|
||
|
||
return publisher.Publish(ctx, outboxinfra.PublishRequest{
|
||
EventType: EventTypeChatTokenUsageAdjustRequested,
|
||
EventVersion: outboxinfra.DefaultEventVersion,
|
||
MessageKey: payload.ConversationID,
|
||
AggregateID: strconv.Itoa(payload.UserID) + ":" + payload.ConversationID,
|
||
Payload: payload,
|
||
})
|
||
}
|