Class: Rack::MongoLog

Inherits:
Object
  • Object
show all
Defined in:
lib/rack/mongolog.rb

Instance Method Summary collapse

Constructor Details

#initialize(app, options = {}) ⇒ MongoLog

Returns a new instance of MongoLog.



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/rack/mongolog.rb', line 6

def initialize(app, options={})
  @app = app
  
  @maps = default_map
  @collection = options[:collection] || 'mlog'

  @maps.update(options[:map]) if options.key?(:map)

  if not options.key?(:uri) and ENV.key?('MONGOHQ_URL')
    options[:uri] = ENV['MONGOHQ_URL']
  end

  # Setup the MongoDB Connection
  uri = URI.parse(options[:uri])
  conn = Mongo::Connection.from_uri(options[:uri])
  @db = conn.db(uri.path.gsub(/^\//, ''))
end

Instance Method Details

#call(env) ⇒ Object 



44
45
46
47
48
49
50
51
52
53
54
55
 
# File 'lib/rack/mongolog.rb', line 44

def call(env)
  status, header, body = @app.call(env)
  kwargs = {}

  @maps.each do |key, block|
    kwargs[key] = block.call(env, status, header, body)
  end

  @db[@collection].save(kwargs)

  [status, header, body]
end

#default_map ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/rack/mongolog.rb', line 24

def default_map
  {
    :method => lambda { |env, status, headers, body| env['REQUEST_METHOD'] },
    :path => lambda { |env, status, headers, body| env['PATH_INFO'] },
    :status => lambda { |env, status, headers, body| status },
    :ip => lambda { |env, status, headers, body| env['REMOTE_ADDR'] },

    :user_agent => lambda { |env, status, headers, body| env['HTTP_USER_AGENT'] },
    :referer => lambda { |env, status, headers, body| env['HTTP_REFERER'] },
    :language => lambda { |env, status, headers, body| env['HTTP_ACCEPT_LANGUAGE'] },

    :redirect => lambda { |env, status, headers, body| ((status == 301) or (status == 302))? headers['Location'] : nil },

    :time => lambda { |env, status, headers, body| Time.now },

    :secure? => lambda { |env, status, headers, body| env['HTTPS'] == 'on' },
    :ajax? => lambda { |env, status, headers, body| env['HTTP_X_REQUESTED_WITH'] == 'XMLHttpRequest' }
  }
end