Contents

require 'statsd'
require 'faraday'




module Saddle
  module Middleware
    module Logging

      # Public: Wraps request with statsd logging
      # Expects statsd_path in request options.  However, if using saddle and no statsd_path is specified
      # will read call_chain and action and use them to construct a statsd_path
      class StatsdLogger < Faraday::Middleware
        attr_accessor :graphite_host, :graphite_port, :namespace

        def initialize(app, graphite_host, graphite_port=nil, namespace=nil)
          super(app)
          @graphite_host = graphite_host
          @graphite_port = graphite_port
          @namespace = namespace
        end

        def statsd
          @statsd ||= begin
            client = ::Statsd.new(@graphite_host, @graphite_port)
            client.namespace = @namespace if @namespace
            client
          end
        end

        def call(env)
          # Try to build up a path for the STATSD logging
          if env[:request][:statsd_path]
            statsd_path = env[:request][:statsd_path]
          elsif env[:request][:saddle]
            statsd_path_components = [
              'saddle',
              ActiveSupport::Inflector.underscore(env[:request][:saddle][:client].name),
            ]
            if env[:request][:saddle][:call_chain] && env[:request][:saddle][:action]
              statsd_path_components += env[:request][:saddle][:call_chain]
              statsd_path_components << env[:request][:saddle][:action]
            else
              statsd_path_components << 'raw'
              statsd_path_components << "#{env[:url].host}#{env[:url].path}"
            end
            statsd_path = statsd_path_components.join('.')
          end

          # If we have a path, wrap the call
          if statsd_path
            self.statsd.time(statsd_path) do
              @app.call(env)
            end
          else
            @app.call(env)
          end
        end
      end

    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
saddle-0.0.31	lib/saddle/middleware/logging/statsd.rb
saddle-0.0.30	lib/saddle/middleware/logging/statsd.rb
saddle-0.0.27	lib/saddle/middleware/logging/statsd.rb
saddle-0.0.26	lib/saddle/middleware/logging/statsd.rb
saddle-0.0.25	lib/saddle/middleware/logging/statsd.rb