RabbitMQ 的 Heartbeat 总结

540次阅读

共计 3329 个字符，预计需要花费 9 分钟才能阅读完成。

heartbeat 通常用来检测通信的对端是否存活（未正常关闭 socket 连接而异常 crash）。其基本原理是检测对应的 socket 连接上数据的收发是否正常，如果一段时间内没有收发数据，则向对端发送一个心跳检测包，如果一段时间内没有回应则认为心跳超时，即认为对端可能异常 crash 了。

rabbitmq 也不例外，heatbeat 在客户端和服务端之间用于检测对端是否正常，即客户端与服务端之间的 tcp 链接是否正常。

1. heartbeat 检测时间间隔的设置

1). 服务端的设置

heartbeat 检测时间间隔可在配置文件 rabbitmq.config 中增加配置项 {heartbeat,Timeout} 进行配置，其中 Timeout 指定时间间隔，单位为秒。

例如:

RabbitMQ 的 Heartbeat 总结

如果没有进行配置，默认的时间间隔为 600 秒（最新版本 3.2.2 修改为 580 秒）

在 rabbit.app 中有 heartbeat 的配置项。

2). 客户端的设置

根据 AMQP 协议，rabbitmq 会通过 connection.tune 信令将 heartbeat 检测时间间隔告知客户端，客户端可以根据需要重新设置该值，并通过 Connection.tune-ok 信令将时间间隔再告诉给 rabbitmq，rabbitmq 会以客户端的时间作为该 tcp 连接上 heartbeat 检测的间隔时间。

RabbitMQ 的 Heartbeat 总结

这里要注意的是：如果时间间隔配置为 0，则表示不启用 heartbeat 检测。

如果启用了 rabbitmq_management 的话，从 web 控制台可以看到客户端的检测的时间间隔。

RabbitMQ 的 Heartbeat 总结

2. heartbeat 的实现

rabbitmq 在收到来自客户端的 connection.tune-ok 信令后，启用心跳检测，rabbitmq 会为每个 tcp 连接创建两个进程用于心跳检测，一个进程定时检测 tcp 连接上是否有数据发送（这里的发送是指 rabbitmq 发送数据给客户端），如果一段时间内没有数据发送给客户端，则发送一个心跳包给客户端，然后循环进行下一次检测；另一个进程定时检测 tcp 连接上是否有数据的接收，如果一段时间内没有收到任何数据，则判定为心跳超时，最终会关闭 tcp 连接。另外，rabbitmq 的流量控制机制可能会暂停 heartbeat 检测，这里不展开描述。

涉及的源码：

start(SupPid, Sock, SendTimeoutSec,
SendFun, ReceiveTimeoutSec, ReceiveFun) ->
%% 数据发送检测进程
{ok, Sender} = start_heartbeater(SendTimeoutSec, SupPid, Sock,
SendFun, heartbeat_sender,
start_heartbeat_sender),
%% 数据接收检测进程
{ok, Receiver} = start_heartbeater(ReceiveTimeoutSec, SupPid,
Sock, ReceiveFun,
heartbeat_receiver,
start_heartbeat_receiver),
{Sender, Receiver}.

start_heartbeat_sender(Sock, TimeoutSec, SendFun) ->
%% the ‘div 2’ is there so that we don’t end up waiting for
%% nearly 2 * TimeoutSec before sending a heartbeat in the
%% boundary case
heartbeater({Sock, TimeoutSec * 1000 div 2, send_oct, 0,
fun () -> SendFun(), continue end}).

start_heartbeat_receiver(Sock, TimeoutSec, ReceiveFun) ->
%% we check for incoming data every interval, and time out after
%% two checks with no change. As a result we will time out
%% between 2 and 3 intervals after the last data has been
%% received
heartbeater({Sock, TimeoutSec * 1000, recv_oct, 1,
fun () -> ReceiveFun(), stop end}).

heartbeater({Sock, TimeoutMillisec,
StatName, Threshold, Handler} = Params,
Deb,
{StatVal, SameCount} = State) ->
Recurse = fun (State1) -> heartbeater(Params, Deb, State1) end,
receive
…
%% 定时检测
after TimeoutMillisec ->
case rabbit_net:getstat(Sock, [StatName]) of
{ok, [{StatName, NewStatVal}]} ->
%% 收发数据有变化
if NewStatVal =/= StatVal ->
%% 重新开始检测
Recurse({NewStatVal, 0});
%% 未达到指定次数, 发送为 0, 接收为 1
SameCount < Threshold ->
%% 计数加 1, 再次检测
Recurse({NewStatVal, SameCount + 1});
%%heartbeat 超时
true ->
%% 对于发送检测超时, 向客户端发送 heartbeat 包
%% 对于接收检测超时, 向父进程发送超时通知
%% 由父进程触发 tcp 关闭等操作
case Handler() of
%% 接收检测超时
stop -> ok;
%% 发送检测超时
continue -> Recurse({NewStatVal, 0})
end;
…