Sha256: 07b8d38c96dda27adc00aeca8d968859dcb04748494eaeb246b6175fe54a6658

Contents?: true

Size: 1021 Bytes

Versions: 1

Compression:

Stored size: 1021 Bytes

Contents

module Healthety
  extend self

  def workers(&block)
    @workers = []
    @threads = []

    instance_eval(&block)

    start
  end

  def host(host)
    @host = host
  end

  def port(port)
    @port = port
  end

  def worker(name, &block)
    @workers << Worker.new(name, &block)
  end

  # Makes the methods defined in the helper modules given in `extensions`
  # available to the workers.
  def helpers(*extensions)
    Worker.class_eval { include(*extensions) } if extensions.any?
  end

  def start
    puts message
    transmission = Transmission.new(@host, @port)

    # Catch Ctrl-C and terminate all worker threads.
    trap("INT") { @threads.map(&:kill) }

    @workers.each do |worker|
      @threads << Thread.new do
        loop do
          worker.perform
          transmission.send(worker.name, worker.value)
          sleep worker.interval
        end
      end
    end

    @threads.map(&:join)
  end

  def message
    <<-EOM
=> Workers sending to #{@host}:#{@port}
=> Ctrl-C to stop
EOM
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version Path
healthety-0.0.6 lib/healthety/healthety.rb