Go 每日一库之 watermill

最新推荐文章于 2024-06-28 08:15:15 发布

darjun

最新推荐文章于 2024-06-28 08:15:15 发布

阅读量1.6k

点赞数

文章标签： go 消息队列 watermill

本文链接：https://blog.csdn.net/darjun/article/details/104594254

版权

watermill是Go语言的异步消息解决方案，支持消息重传、保存消息，提供监控和限流等中间件。通过路由功能，可以实现消息处理的监控、重试、统计等，并内置多种订阅-发布实现，如AMQP。通过示例展示了如何使用watermill进行消息发布和订阅，以及如何设置中间件进行消息处理。

摘要由CSDN通过智能技术生成

简介

在上一篇文章Go 每日一库之 message-bus中，我们介绍了一款小巧、实现简单的异步通信库。作为学习，message-bus确实不错。但是在实际使用上，message-bus的功能就有点捉襟见肘了。例如，message-bus将消息发送到订阅者管道之后就不管了，这样如果订阅者处理压力较大，会在管道中堆积太多消息，一旦订阅者异常退出，这些消息将会全部丢失！另外，message-bus不负责保存消息，如果订阅者后启动，之前发布的消息，这个订阅者是无法收到的。这些问题，我们将要介绍的watermill都能解决！

watermill是 Go 语言的一个异步消息解决方案，它支持消息重传、保存消息，后启动的订阅者也能收到前面发布的消息。watermill内置了多种订阅-发布实现，包括Kafka/RabbitMQ，甚至还支持HTTP/MySQL binlog。当然也可以编写自己的订阅-发布实现。此外，它还提供了监控、限流等中间件。

快速使用

watermill内置了很多订阅-发布实现，最简单、直接的要属GoChannel。我们就以这个实现为例介绍watermill的特性。

安装：

$ go get github.com/ThreeDotsLabs/watermill

使用：

package main

import (
  "context"
  "log"
  "time"

  "github.com/ThreeDotsLabs/watermill"
  "github.com/ThreeDotsLabs/watermill/message"
  "github.com/ThreeDotsLabs/watermill/pubsub/gochannel"
)

func main() {
  pubSub := gochannel.NewGoChannel(
    gochannel.Config{},
    watermill.NewStdLogger(false, false),
  )

  messages, err := pubSub.Subscribe(context.Background(), "example.topic")
  if err != nil {
    panic(err)
  }

  go process(messages)

  publishMessages(pubSub)
}

func publishMessages(publisher message.Publisher) {
  for {
    msg := message.NewMessage(watermill.NewUUID(), []byte("Hello, world!"))

    if err := publisher.Publish("example.topic", msg); err != nil {
      panic(err)
    }

    time.Sleep(time.Second)
  }
}

func process(messages <-chan *message.Message) {
  for msg := range messages {
    log.Printf("received message: %s, payload: %s", msg.UUID, string(msg.Payload))
    msg.Ack()
  }
}

首先，我们创建一个GoChannel对象，它是一个消息管理器。可以调用其Subscribe订阅某个主题（topic）的消息，调用其Publish()以某个主题发布消息。Subscribe()方法会返回一个<-chan *message.Message，一旦该主题有消息发布，GoChannel就会将消息发送到该管道中。订阅者只需监听此管道，接收消息进行处理。在上面的例子中，我们启动了一个消息处理的goroutine，持续从管道中读取消息，然后打印输出。主goroutine在一个死循环中每隔 1s 发布一次消息。