Golang 事务消息队列：基于 SQLite 的轻量级消息队列解决方案

最新推荐文章于 2025-04-20 21:15:54 发布

gospace

最新推荐文章于 2025-04-20 21:15:54 发布

阅读量486

点赞数 7

文章标签： golang sqlite

本文链接：https://blog.csdn.net/weixin_42324398/article/details/147290568

版权

事务消息队列：基于 SQLite 的轻量级消息队列解决方案

什么是消息队列？

消息队列是一种异步通信机制，用于在分布式系统中传递消息。它就像一个"邮局"，发送方将消息放入队列，接收方从队列中取出消息进行处理。这种模式带来了以下好处：

解耦：生产者和消费者不需要直接通信
削峰：可以缓冲突发流量
异步：提高系统响应速度
可靠性：消息持久化，确保不丢失

什么是事务消息队列？

事务消息队列的核心是保证消息发布的原子性。在传统消息队列中，如果事务回滚，或者事务完成，但程序在发送消息前崩溃，消息可能异常。而事务消息队列通过以下机制解决这些问题：

原子性：消息的发送和业务操作在同一个事务中，要么都成功，要么都失败
回滚机制：如果事务失败，消息会自动回滚，不会发送
持久性：消息一旦提交就永久保存，即使程序崩溃也不会丢失

实践案例：使用 NSQite 实现用户行为分析系统

假设我们有一个内容平台，需要记录用户的行为数据（如浏览、点赞、收藏等）用于后续分析。这些行为数据的特点是：

数据量大但实时性要求不高
处理失败可以重试
不影响核心业务流程

使用消息队列后，我们可以：

用户行为发生时立即返回成功
行为数据进入消息队列
后台服务异步处理数据（数据清洗、统计分析等）

使用 nsqite 的代码示例：

// 定义消息处理器
type UserActionHandler struct{}

func (h *UserActionHandler) HandleMessage(message *nsqite.Message) error {
    var action struct {
        UserID    string `json:"user_id"`
        Action    string `json:"action"`
        ContentID string `json:"content_id"`
        Timestamp string `json:"timestamp"`
    }
    if err := json.Unmarshal(message.Body, &action); err != nil {
        return err
    }
    // 数据清洗和统计分析
    return analyzeUserAction(action)
}

func main() {
    db, err := gorm.Open(sqlite.Open("user_actions.db"), &gorm.Config{})
    if err != nil {
        log.Fatal(err)
    }
    // 设置 GORM 数据库连接
    nsqite.SetGorm(db)

    const topic = "user_actions"
    // 创建生产者
    p := nsqite.NewProducer()
    // 创建消费者，设置最大重试次数为 5
    c := nsqite.NewConsumer(topic, "consumer1", nsqite.WithConsumerMaxAttempts(5))
    // 添加 5 个并发处理器
    c.AddConcurrentHandlers(&UserActionHandler{}, 5)

    // 在事务中发布消息
    db.Transaction(func(tx *gorm.DB) error {
        // 业务操作
        if err := doSomeBusiness(tx); err != nil {
            return err
        }
        // 发布消息
        action := map[string]interface{}{
            "user_id":    "123",
            "action":     "view",
            "content_id": "456",
            "timestamp":  time.Now().Format(time.RFC3339),
        }
        body, _ := json.Marshal(action)
        return p.PublishTx(tx, topic, body)
    })

	time.Sleep(time.Second * 5)
}