Contents

# Copyright (c) 2021 Contrast Security, Inc. See https://www.contrastsecurity.com/enduser-terms-0317a for more details.
# frozen_string_literal: true

require 'contrast/components/interface'
require 'contrast/agent/worker_thread'

module Contrast
  module Agent
    # The ServiceHeartbeat functions to keep the Contrast Service alive and
    # ensure that it maintains this Agent's ApplicationContext.
    class ServiceHeartbeat < WorkerThread
      include Contrast::Components::Interface
      access_component :logging

      # Spec recommends 30 seconds, we're going with 15.
      REFRESH_INTERVAL_SEC = 15

      def start_thread!
        return if running?

        @_thread = Contrast::Agent::Thread.new do
          logger.info('Starting heartbeat thread.')
          loop do
            begin
              Contrast::Agent.messaging_queue.send_event_eventually(poll_message)
            end
            sleep REFRESH_INTERVAL_SEC
          end
        end
      end

      def poll_message
        @_poll_message ||= Contrast::Api::Dtm::Poll.new
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
contrast-agent-4.6.0	lib/contrast/agent/service_heartbeat.rb
contrast-agent-4.5.0	lib/contrast/agent/service_heartbeat.rb