heartbeat.rb

# frozen_string_literal: true

module Appsignal
  class Heartbeat
    class << self
      def transmitter
        @transmitter ||= Appsignal::Transmitter.new(
          "#{Appsignal.config[:logging_endpoint]}/heartbeats/json"
        )
      end
    end

    attr_reader :name, :id

    def initialize(name:)
      @name = name
      @id = SecureRandom.hex(8)
    end

    def start
      transmit_event("start")
    end

    def finish
      transmit_event("finish")
    end

    private

    def event(kind)
      {
        :name => name,
        :id => @id,
        :kind => kind,
        :timestamp => Time.now.utc.to_i
      }
    end

    def transmit_event(kind)
      unless Appsignal.active?
        Appsignal.internal_logger.debug("AppSignal not active, not transmitting heartbeat event")
        return
      end

      response = self.class.transmitter.transmit(event(kind))

      if response.code.to_i >= 200 && response.code.to_i < 300
        Appsignal.internal_logger.trace("Transmitted heartbeat `#{name}` (#{id}) #{kind} event")
      else
        Appsignal.internal_logger.error(
          "Failed to transmit heartbeat event: status code was #{response.code}"
        )
      end
    rescue => e
      Appsignal.internal_logger.error("Failed to transmit heartbeat event: #{e}")
    end
  end

  def self.heartbeat(name)
    heartbeat = Appsignal::Heartbeat.new(:name => name)
    output = nil

    if block_given?
      heartbeat.start
      output = yield
    end

    heartbeat.finish
    output
  end
end