Contents

# Worker used to report metrics to various services

module RestFtpDaemon
  class ReporterWorker < Worker

  protected

    def worker_init
      # Load corker conf
      config_section :reporter

      # Other configuration options
      @feature_newrelic = Conf.feature?(:newrelic)

      # Check that everything is OK
      return "invalid timer"      unless @config[:timer].to_i > 0
      return false
    end

    def worker_after
      # Sleep for a few seconds
      worker_status WORKER_STATUS_WAITING
      sleep @config[:timer]
    end

    def worker_process
      # Announce we are working
      worker_status WORKER_STATUS_REPORTING

      # Report metrics
      do_metrics
    end

  private

    def do_metrics
      # Get common metrics and dump them to logs
      log_debug "begin metrics sample"
      metrics = Metrics.sample

      # Skip following if no valid metrics collected
      unless metrics.is_a? Hash
        log_error "unable to collect metrics"
        return
      end
      log_info "collected metrics (newrelic: #{@feature_newrelic.inspect})", metrics

      # Transpose metrics to NewRelic metrics
      report_newrelic(metrics) if @feature_newrelic
    end

    def report_newrelic metrics
      metrics_newrelic = {}
      metrics.each do |group, pairs|
        pairs.each do |key, value|
          name = "rftpd/#{group}/#{key}"
          ::NewRelic::Agent.record_metric(name, value)
          metrics_newrelic[name] = value
        end
      end

      # Don't dump metrics unless we're debugging
      newrelic_app_name = Conf.at(:newrelic, :app_name)
      msg_newrelic = "reported metrics to NewRelic [#{newrelic_app_name}]"
      if @config[:debug]
        log_debug msg_newrelic, metrics_newrelic
      else
        log_info msg_newrelic
      end

    end

  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
rest-ftp-daemon-0.434.0	lib/rest-ftp-daemon/workers/reporter.rb