Module: DbCharmer::ActiveRecord::ConnectionSwitching

Defined in:
lib/db_charmer/active_record/connection_switching.rb

Instance Method Summary collapse

Instance Method Details

#coerce_to_connection_proxy(conn, should_exist = true) ⇒ Object 



51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
 
# File 'lib/db_charmer/active_record/connection_switching.rb', line 51

def coerce_to_connection_proxy(conn, should_exist = true)
  # Return nil if given no connection specification
  return nil if conn.nil?

  # For sharded proxies just use them as-is
  return conn if conn.respond_to?(:set_real_connection)

  # For connection proxies and objects that could be coerced into a proxy just call the coercion method
  return conn.db_charmer_connection_proxy if conn.respond_to?(:db_charmer_connection_proxy)

  # For plain AR connection adapters, just use them as-is
  return conn if conn.kind_of?(::ActiveRecord::ConnectionAdapters::AbstractAdapter)

  # For connection names, use connection factory to create new connections
  if conn.kind_of?(Symbol) || conn.kind_of?(String)
    return DbCharmer::ConnectionFactory.connect(conn, should_exist)
  end

  # For connection configs (hashes), create connections
  if conn.kind_of?(Hash)
    conn = conn.symbolize_keys
    raise ArgumentError, "Missing required :connection_name parameter" unless conn[:connection_name]
    return DbCharmer::ConnectionFactory.connect_to_db(conn[:connection_name], conn)
  end

  # Fails for unsupported connection types
  raise "Unsupported connection type: #{conn.class}"
end

#establish_real_connection_if_exists(name, should_exist = false) ⇒ Object 



4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
 
# File 'lib/db_charmer/active_record/connection_switching.rb', line 4

def establish_real_connection_if_exists(name, should_exist = false)
  name = name.to_s

  # Check environment name
  config = configurations[DbCharmer.env]
  unless config
    error = "Invalid environment name (does not exist in database.yml): #{DbCharmer.env}. Please set correct Rails.env or DbCharmer.env."
    raise ArgumentError, error
  end

  # Check connection name
  config = config[name]
  unless config
    if should_exist
      raise ArgumentError, "Invalid connection name (does not exist in database.yml): #{DbCharmer.env}/#{name}"
    end
    return # No need to establish connection - they do not want us to
  end

  # Pass connection name with config
  config[:connection_name] = name
  establish_connection(config)
end

#hijack_connection! ⇒ Object 



29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
 
# File 'lib/db_charmer/active_record/connection_switching.rb', line 29

def hijack_connection!
  return if self.respond_to?(:connection_with_magic)
  class << self
    # Make sure we check our accessors before going to the default connection retrieval method
    def connection_with_magic
      db_charmer_remapped_connection || db_charmer_model_connection_proxy || connection_without_magic
    end
    alias_method_chain :connection, :magic

    def connection_pool_with_magic
      if connection.respond_to?(:abstract_connection_class)
        abstract_connection_class = connection.abstract_connection_class
        connection_handler.retrieve_connection_pool(abstract_connection_class) || connection_pool_without_magic
      else
        connection_pool_without_magic
      end
    end
    alias_method_chain :connection_pool, :magic
  end
end

#switch_connection_to(conn, should_exist = true) ⇒ Object 



81
82
83
84
85
86
87
88
89
90
 
# File 'lib/db_charmer/active_record/connection_switching.rb', line 81

def switch_connection_to(conn, should_exist = true)
  new_conn = coerce_to_connection_proxy(conn, should_exist)

  if db_charmer_connection_proxy.respond_to?(:set_real_connection)
    db_charmer_connection_proxy.set_real_connection(new_conn)
  end

  self.db_charmer_connection_proxy = new_conn
  self.hijack_connection!
end