Module: DbCharmer::ActiveRecord::Migration::MultiDbMigrations
- Defined in:
- lib/db_charmer/active_record/migration/multi_db_migrations.rb
Defined Under Namespace
Modules: ClassMethods
Class Method Summary collapse
Instance Method Summary collapse
- #migrate_with_db_wrapper(direction) ⇒ Object
- #on_db(db_name, &block) ⇒ Object
- #record_on_db(db_name, block) ⇒ Object
- #replay_commands_on_db(name, commands) ⇒ Object
Class Method Details
.append_features(base) ⇒ Object
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# File 'lib/db_charmer/active_record/migration/multi_db_migrations.rb', line 6 def self.append_features(base) return false if base < self super base.extend const_get("ClassMethods") if const_defined?("ClassMethods") base.class_eval do if DbCharmer.rails31? alias_method_chain :migrate, :db_wrapper else class << self alias_method_chain :migrate, :db_wrapper end end end end |
Instance Method Details
#migrate_with_db_wrapper(direction) ⇒ Object
108 109 110 111 112 113 114 115 116 117 118 |
# File 'lib/db_charmer/active_record/migration/multi_db_migrations.rb', line 108 def migrate_with_db_wrapper(direction) if names = self.class.multi_db_names names.each do |multi_db_name| on_db(multi_db_name) do migrate_without_db_wrapper(direction) end end else migrate_without_db_wrapper(direction) end end |
#on_db(db_name, &block) ⇒ Object
136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 |
# File 'lib/db_charmer/active_record/migration/multi_db_migrations.rb', line 136 def on_db(db_name, &block) if @connection.is_a?(::ActiveRecord::Migration::CommandRecorder) record_on_db(db_name, block) return end s = "Switching connection to " if db_name.is_a?(Hash) s << db_name[:connection_name] s << ", schema #{db_name[:schema_name]}" if db_name[:schema_name] else s << db_name.inspect end announce s # Switch connection old_connection, old_proxy = @connection, ::ActiveRecord::Base.db_charmer_connection_proxy db_name = nil if db_name == :default ::ActiveRecord::Base.switch_connection_to(db_name, DbCharmer.connections_should_exist?) # Yield the block ::ActiveRecord::Base.connection_pool.with_connection do |conn| @connection = conn yield end ensure @connection = old_connection # Switch it back ::ActiveRecord::Base.verify_active_connections! announce "Switching connection back" ::ActiveRecord::Base.switch_connection_to(old_proxy) end |
#record_on_db(db_name, block) ⇒ Object
120 121 122 123 124 125 126 |
# File 'lib/db_charmer/active_record/migration/multi_db_migrations.rb', line 120 def record_on_db(db_name, block) recorder = ::ActiveRecord::Migration::CommandRecorder.new(DbCharmer::ConnectionFactory.connect(db_name)) old_recorder, @connection = @connection, recorder block.call old_recorder.record :on_db, [db_name, @connection] @connection = old_recorder end |
#replay_commands_on_db(name, commands) ⇒ Object
128 129 130 131 132 133 134 |
# File 'lib/db_charmer/active_record/migration/multi_db_migrations.rb', line 128 def replay_commands_on_db(name, commands) on_db(name) do commands.each do |cmd, args| send(cmd, *args) end end end |