rabbitmq——heartbeat

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

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

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

1). 服务端的设置

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

例如:

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

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

2). 客户端的设置

根据AMQP协议，rabbitmq会经过connection.tune信令将heartbeat检测时间间隔告知客户端，客户端能够根据须要从新设置该值，并经过Connection.tune-ok信令将时间间隔再告诉给rabbitmq，rabbitmq会以客户端的时间做为该tcp链接上heartbeat检测的间隔时间。

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

若是启用了rabbitmq_management的话，从web控制台能够看到客户端的检测的时间间隔。

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;
           ...

 

 

收发检测的时候利用了inet模块的getstat，查看socket的统计信息

 

recv_oct: 查看socket上接收的字节数

send_oct: 查看socket上发送的字节数

inet详细见这里: http://www.erlang.org/doc/man/inet.html