Contents

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

module Contrast
  module Api
    module Communication
      # Keeps track of the state of connections to SpeedRacer.
      class ConnectionStatus
        def initialize
          @last_success = nil
          @last_failure = nil
          @startup_messages_sent = false
        end

        # Whether we have sent startup message to SpeedRacer. True after successfully sending startup messages to
        # SpeedRacer and reset to false if we lose connection.
        #
        # @return [Boolean]
        def startup_messages_sent?
          @startup_messages_sent
        end

        # The last message sent was successful or not
        #
        # @return [Boolean]
        def connected?
          @last_success && (@last_failure.nil? || @last_success > @last_failure)
        end

        # The current state of the SpeedRacer is active with a successful message sent
        def success!
          @startup_messages_sent = true
          @last_success = Time.now.to_f
        end

        # The SpeedRacer may be in some sort of error state
        def failure!
          @startup_messages_sent = false
          @last_failure = Time.now.to_f
        end
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
contrast-agent-4.14.1	lib/contrast/api/communication/connection_status.rb
contrast-agent-4.14.0	lib/contrast/api/communication/connection_status.rb