Contents

module PumaAutoTune
  class Hook

    def initialize(resource)
      @resource = resource
      @started  = Time.now
      @hooks    = {}
      @wraps    = {}
    end

    def define_hook(name, &block)
      wrap_hook(name)
      @hooks[name] = block
    end
    alias :set :define_hook

    def call(name)
      hook = @hooks[name] or raise "No such hook #{name.inspect}. Available: #{@hooks.keys.inspect}"
      hook.call(self.args)
    end

    def wrap_hook(name, &block)
      if block
        @wraps[name] = block
      else
        if wrap = @wraps[name]
          @hooks[name] = wrap.call(@hooks[name])
        end
      end
    end
    alias :wrap :wrap_hook

    def auto_cycle
      Thread.new do
        loop do
          sleep PumaAutoTune.frequency
          call(:cycle) if @resource.master.running?
        end
      end
    end

    def log(msg, options = {})
      elapsed = (Time.now - @started).ceil
      msg     = ["PumaAutoTune (#{elapsed}s): #{msg}"]

      options[@resource.name] = @resource.amount
      options["current_cluster_size"] = @resource.workers.size
      options.each { |k, v| msg << "puma.#{k.to_s.downcase}=#{v}" }
      puts msg.join(" ")
    end

    def args
      @resource.reset
      [@resource.amount, @resource.master, @resource.workers]
    end
  end
end

require 'puma_auto_tune/defaults/ram/wrappers'
require 'puma_auto_tune/defaults/ram/hooks'

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
puma_auto_tune-0.0.1	lib/puma_auto_tune/hook.rb