Class: ApiHitLogger::Middleware

Inherits:
Object
  • Object
show all
Defined in:
lib/api_hit_logger/middleware.rb

Instance Method Summary collapse

Constructor Details

#initialize(app) ⇒ Middleware

Returns a new instance of Middleware.



7
8
9
 
# File 'lib/api_hit_logger/middleware.rb', line 7

def initialize(app)
  @app = app
end

Instance Method Details

#call(env) ⇒ Object 



11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
 
# File 'lib/api_hit_logger/middleware.rb', line 11

def call(env)
  request = Rack::Request.new(env)
  start_time = Time.now
  status = nil

  begin
    # Call the downstream app (Rails)
    status, headers, response = @app.call(env)

    # Log success with status
    log_api_hit(request, status, start_time, Time.now)
    [status, headers, response]

  rescue => e
    # In case of exception, set status to 500 (Internal Server Error)
    status = 500

    log_api_hit(request, status, start_time, Time.now, error: e.message)

    raise e # re-raise to let Rails handle it
  end
end