912a6d8cfe
refactor: ♻️ 调整 Outbox 写入时序并移除 Kafka 首包同步投递逻辑 * 将 `outbox` 表写入逻辑后置到 LLM 请求之后，减少主链路阻塞 * 删除 Codex 生成的 Kafka 首包同步投递抽象逻辑，简化消息发送流程 * 优化 SSE 首字到达时间，整体降低约 1s 延迟 * 当前在请求 LLM 之前的流程全部为 Redis 操作，显著降低 IO 开销 docs: 📊 保留 SSE 链路性能打点逻辑 * 保留原有 SSE 全链路打点计时代码，便于后续性能排查与分析 * 当前默认注释，如需使用可手动启用进行性能调试
46 lines
969 B
Go
46 lines
969 B
Go
package kafka
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
segmentkafka "github.com/segmentio/kafka-go"
|
|
)
|
|
|
|
type Producer struct {
|
|
writer *segmentkafka.Writer
|
|
}
|
|
|
|
func NewProducer(cfg Config) (*Producer, error) {
|
|
if len(cfg.Brokers) == 0 {
|
|
return nil, errors.New("kafka brokers 未配置")
|
|
}
|
|
writer := &segmentkafka.Writer{
|
|
Addr: segmentkafka.TCP(cfg.Brokers...),
|
|
Balancer: &segmentkafka.Hash{},
|
|
RequiredAcks: segmentkafka.RequireOne,
|
|
Async: false,
|
|
}
|
|
return &Producer{writer: writer}, nil
|
|
}
|
|
|
|
// Enqueue 将消息写入 Kafka。
|
|
func (p *Producer) Enqueue(ctx context.Context, topic, key string, value []byte) error {
|
|
if p == nil || p.writer == nil {
|
|
return errors.New("kafka producer 未初始化")
|
|
}
|
|
msg := segmentkafka.Message{
|
|
Topic: topic,
|
|
Key: []byte(key),
|
|
Value: value,
|
|
}
|
|
return p.writer.WriteMessages(ctx, msg)
|
|
}
|
|
|
|
func (p *Producer) Close() error {
|
|
if p == nil || p.writer == nil {
|
|
return nil
|
|
}
|
|
return p.writer.Close()
|
|
}
|