Sha256: e715a0e0fe31241c5dec6131a2f31c48250940eacbdffc942422a2e24f4fa406

Contents?: true

Size: 1.03 KB

Versions: 1

Compression:

Stored size: 1.03 KB

Contents

module Healthety
  extend self

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

    instance_eval(&block)

    start
  end

  def server(server)
    @server = server
  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(@server, @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, Time.now.utc)
          sleep worker.interval
        end
      end
    end

    @threads.map(&:join)
  end

  def message
    <<-EOM
=> Workers starting to send to #{@server}:#{@port}
=> Ctrl-C to stop
EOM
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

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