Contents

require 'json'

module Attender
  class Agent
    Signal.trap(:INT) { exit 0 }

    def initialize(url, port)
      @url = url.nil? ? 'localhost' : url
      @port = port.nil? ? '8500' : port
      @timeout = 60
      @index = nil
      @response_queue = Queue.new
      STDOUT.sync = true
      Thread.abort_on_exception = true
    end

    def run
      getter = Thread.new do
        path = '/v1/health/state/passing'
        loop do
          @response_queue << get(path)
        end
      end

      putter = Thread.new do
        response = nil
        loop do
          unless @response_queue.empty?
            prev_response = response
            response = @response_queue.pop
            put(response, prev_response)
          end
          sleep rand * 0.1
        end
      end

      getter.join
      putter.join
    end

    def get(path, lp = true)
      http = Net::HTTP.new(@url, @port)
      http.read_timeout = @timeout
      path+= "?wait=#{(@timeout-1).to_s}s&index=#{@index}" unless @index.nil? if lp
      response = http.get(path)

      code = response.code
      unless code == '200'
        raise "Invalid status code #{code}"
      end

      @index = response['x-consul-index']
      if @index.nil?
        raise 'Missing x-consul-index'
      end

      JSON.parse(response.body)
    end

    def put(response, prev_response)
      return if prev_response.nil?

      response.select { |v| !prev_response.include?(v) }.each do |check|
        put_format(check, '+')
      end

      prev_response.select { |v| !response.include?(v) }.each do |check|
        put_format(check, '-')
      end
    end

    private

    def put_format(check, type)
      puts "#{type} #{check['Node']} #{check['CheckID']}"
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
attender-0.0.1	lib/attender/agent.rb