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 



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

def coerce_to_connection_proxy(conn, should_exist = true)
  #Rails.logger.debug "coerce_to_connection_proxy: conn class=#{conn.class}"
  return nil if conn.nil?

  if conn.kind_of?(Symbol) || conn.kind_of?(String)
    return DbCharmer::ConnectionFactory.connect(conn, should_exist)
  end

  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

  if conn.respond_to?(:db_charmer_connection_proxy)
    return conn.db_charmer_connection_proxy
  end

  if conn.kind_of?(::ActiveRecord::ConnectionAdapters::AbstractAdapter) || conn.kind_of?(DbCharmer::Sharding::StubConnection)
    return conn
  end

  raise "Unsupported connection type: #{conn.class}"
end

#establish_real_connection_if_exists(name, should_exist = false) ⇒ Object 



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

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 



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

def hijack_connection!

  # do not hijack anything if the model is not handled by db_charmer
  return unless self.respond_to?(:db_charmer_extended)

  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_connection_proxy || connection_without_magic
    end
    alias_method_chain :connection, :magic

    def connection_pool_with_magic
      abstract_connection_class = connection.abstract_connection_class rescue nil # respond_to? doesn't work on connection_proxy...
      if 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 



82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
 
# File 'lib/db_charmer/active_record/connection_switching.rb', line 82

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

  if db_charmer_connection_proxy.is_a?(DbCharmer::Sharding::StubConnection)
    db_charmer_connection_proxy.set_real_connection(new_conn)
  end

  # Set schema name as table_name_prefix
  set_schema_table_name_prefix(conn) if self.respond_to? :set_schema_table_name_prefix


  self.db_charmer_connection_proxy = new_conn
  self.hijack_connection!

  # self.reset_column_information
end