我在Go语言中有一个RabbitMQ消费者脚本。这是一个来自RabbitMQ tutorial的简单脚本,它使用了streadway/amqp
库。
问题是,如果RabbitMQ服务器停止,消费者脚本就不会退出;并且当RabbitMQ服务器重新启动时,消费者不再接收消息。
是否有一种方法可以检测到消费者连接已断开并重新连接,或者至少终止消费者脚本?
我知道库为连接设置了默认的10秒心跳间隔;有没有可能利用一下?
func main() {
conn, err := amqp.Dial("amqp://guest:guest@localhost:5672/")
failOnError(err, "Failed to connect to RabbitMQ")
defer conn.Close()
ch, err := conn.Channel()
failOnError(err, "Failed to open a channel")
defer ch.Close()
q, err := ch.QueueDeclare(
"test_task_queue", // name
true, // durable
false, // delete when unused
false, // exclusive
false, // no-wait
nil, // arguments
)
failOnError(err, "Failed to declare a queue")
err = ch.Qos(
1, // prefetch count
0, // prefetch size
false, // global
)
failOnError(err, "Failed to set QoS")
msgs, err := ch.Consume(
q.Name, // queue
"", // consumer
false, // auto-ack
false, // exclusive
false, // no-local
false, // no-wait
nil, // args
)
failOnError(err, "Failed to register a consumer")
forever := make(chan bool)
go func() {
for d := range msgs {
log.Printf("Received a message: %s", d.Body)
d.Ack(false)
dot_count := bytes.Count(d.Body, []byte("."))
t := time.Duration(dot_count)
time.Sleep(t * time.Second)
log.Printf("Done")
}
}()
log.Printf(" [*] Waiting for messages. To exit press CTRL+C")
<-forever
}
4条答案
按热度按时间iibxawm41#
amqp.Connection
具有NotifyClose()
方法,该方法返回通道信号传输或协议错误。l7wslrjt2#
有几种方法可以做到这一点:检查传送通道是否关闭或使用
Channel.NotifyClose
。检查传送通道
启动消费者后,您将从传递通道接收。如您所知,接收操作可能采用
x, ok := <-ch
的特殊形式,其中ok
在x
具有零值时为false,原因是通道已关闭(并且为空):这是因为,当AMQP通道关闭或发生错误时,Go通道
<-chan amqp.Delivery
将关闭:[It]继续传送至传回的chan Delivery,直到发生Channel.Cancel、Connection.Close、Channel.Close或AMQP例外状况。
使用
Channel.NotifyClose
这很简单,原理也一样:
NotifyClose注册一个侦听器,用于在服务器以Connection.Close或Channel.Close方法的形式发送通道或连接异常时使用。
NotifyClose
返回的通道与您作为参数传递的通道相同;该方法只在内部注册它,因此您可以执行以下操作:其中,
n
是非零缓冲区大小。确保将缓冲通道传递到NotifyClose
,否则,根据代码的结构,库可能会在发送时阻塞。然后,您可以在
errC
通道上进行接收,并根据错误类型采取措施。简而言之,错误可能是:nil
如果程序故意调用conn.Close()
要了解错误是否可恢复,您可以检查
amqp.Error
的Code
字段和/或Recover
字段,如果发生软异常,该字段将设置为true。下面的函数显示了如何区分错误代码-这是作为附加信息提供的。对于一般情况,只需检查
Error.Recover
:7bsow1i63#
这可能会帮助一些人
第一个
7vux5j2d4#
没有发现go-amqp库实现了连接池的断开和重新连接功能。
github上有一个基于Amqp二次打包的开源代码,已经支持断线后重连和异常重连,代码使用起来也比较简单,每个服务都有一个连接和通道。
Source Code here
示例代码:
希望它能帮助你或给你一些想法