RnD-Soft/lusnoc

require 'timeouter'
require 'lusnoc/exceptions'
require 'lusnoc/helper'

module Lusnoc
  class Watcher

    include Helper

    def initialize(base_url,
                   timeout: 0,
                   eclass: Lusnoc::TimeoutError,
                   emessage: 'watch timeout')
      @base_url = base_url
      @timeout  = timeout
      @eclass   = eclass
      @emessage = emessage
    end

    # run Consul blocking request in a loop with timeout support.
    # break condition yielded by block call with response body
    def run(max_consul_wait: nil)
      logger.debug "Watch #{@base_url} with #{@timeout.inspect} timeout"
      last_x_consul_index = 1

      Timeouter.loop!(@timeout, eclass: @eclass, message: @emessage) do |t|
        wait_condition = build_wait_condition(@base_url, t.left, max_consul_wait)
        url = "#{@base_url}?index=#{last_x_consul_index}#{wait_condition}"

        resp = Lusnoc.http_get(url, timeout: t.left)
        return true if yield(resp.body)

        logger.debug "Watch #{@base_url} response: #{resp.body}"

        index = [Integer(resp['x-consul-index']), 1].max
        last_x_consul_index = (index < last_x_consul_index ? 1 : index)
        sleep 0.4
      end
    end

    def build_wait_condition(_url, time_left, max_consul_wait)
      if time_left
        max = [time_left.to_i, max_consul_wait.to_i].max
        "&wait=#{max}s"
      elsif max_consul_wait
        "&wait=#{max_consul_wait.to_i}s"
      else
        ''
      end
    end

  end
end