Sha256: 9b4be9e2c91bee24efd07e214a0bb935cb5d64bd809195734fe19d7e89ee706c

Contents?: true

Size: 1.58 KB

Versions: 26

Compression:

Stored size: 1.58 KB

Contents

require "thread"
require "amq/protocol/client"
require "amq/protocol/frame"

module Bunny
  # Periodically sends heartbeats, keeping track of the last publishing activity.
  #
  # @private
  class HeartbeatSender

    #
    # API
    #

    def initialize(transport, logger)
      @transport = transport
      @logger    = logger
      @mutex     = Monitor.new

      @last_activity_time = Time.now
    end

    def start(period = 30)
      @mutex.synchronize do
        # calculate interval as half the given period plus
        # some compensation for Ruby's implementation inaccuracy
        # (we cannot get at the nanos level the Java client uses, and
        # our approach is simplistic). MK.
        @interval = [(period / 2) - 1, 0.4].max

        @thread = Thread.new(&method(:run))
      end
    end

    def stop
      @mutex.synchronize { @thread.exit }
    end

    def signal_activity!
      @last_activity_time = Time.now
    end

    protected

    def run
      begin
        loop do
          self.beat

          sleep @interval
        end
      rescue IOError => ioe
        @logger.error "I/O error in the hearbeat sender: #{ioe.message}"
        stop
      rescue Exception => e
        @logger.error "Error in the hearbeat sender: #{e.message}"
        stop
      end
    end

    def beat
      now = Time.now

      if now > (@last_activity_time + @interval)
        @logger.debug { "Sending a heartbeat, last activity time: #{@last_activity_time}, interval (s): #{@interval}" }
        @transport.write_without_timeout(AMQ::Protocol::HeartbeatFrame.encode)
      end
    end
  end
end

Version data entries

26 entries across 26 versions & 1 rubygems

Version Path
bunny-2.6.0 lib/bunny/heartbeat_sender.rb
bunny-2.5.1 lib/bunny/heartbeat_sender.rb
bunny-2.5.0 lib/bunny/heartbeat_sender.rb
bunny-2.4.0 lib/bunny/heartbeat_sender.rb
bunny-2.3.1 lib/bunny/heartbeat_sender.rb
bunny-2.3.0 lib/bunny/heartbeat_sender.rb