Class: RailsMultisite::Middleware

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

Instance Method Summary collapse

Constructor Details

#initialize(app, config = nil) ⇒ Middleware

Returns a new instance of Middleware.



4
5
6
7
 
# File 'lib/rails_multisite/middleware.rb', line 4

def initialize(app, config = nil)
  @app = app
  @db_lookup = config && config[:db_lookup]
end

Instance Method Details

#call(env) ⇒ Object 



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

def call(env)
  host = ConnectionManagement.host(env)
  db = nil
  begin

    unless ConnectionManagement.connection_spec(host: host)
      db = @db_lookup && @db_lookup.call(env)
      if db
        host = nil
      else
        return [404, {}, ["not found"]]
      end
    end

    ActiveRecord::Base.connection_handler.clear_active_connections!
    ConnectionManagement.establish_connection(host: host, db: db)
    CookieSalt.update_cookie_salts(env: env, host: host)
    @app.call(env)
  ensure
    ActiveRecord::Base.connection_handler.clear_active_connections!
  end
end