module Octopus::Migrator
Public Class Methods
included(base)
click to toggle source
# File lib/octopus/migration.rb, line 64 def self.included(base) unless Octopus.atleast_rails52? base.extend(ClassMethods) base.class_eval do class << self alias_method :migrate_without_octopus, :migrate alias_method :migrate, :migrate_with_octopus alias_method :up_without_octopus, :up alias_method :up, :up_with_octopus alias_method :down_without_octopus, :down alias_method :down, :down_with_octopus alias_method :run_without_octopus, :run alias_method :run, :run_with_octopus alias_method :rollback_without_octopus, :rollback alias_method :rollback, :rollback_with_octopus end end end base.send :alias_method, :run_without_octopus, :run base.send :alias_method, :run, :run_with_octopus base.send :alias_method, :migrate_without_octopus, :migrate base.send :alias_method, :migrate, :migrate_with_octopus base.send :alias_method, :migrations_without_octopus, :migrations base.send :alias_method, :migrations, :migrations_with_octopus end
Public Instance Methods
connection()
click to toggle source
# File lib/octopus/migration.rb, line 129 def connection ActiveRecord::Base.connection end
migrate_with_octopus(&block)
click to toggle source
# File lib/octopus/migration.rb, line 110 def migrate_with_octopus(&block) return migrate_without_octopus(&block) unless connection.is_a?(Octopus::Proxy) shards = migrations.map(&:shards).flatten.map(&:to_s) connection.send_queries_to_multiple_shards(shards) do migrate_without_octopus(&block) end rescue ActiveRecord::UnknownMigrationVersionError => e raise unless migrations(true).detect { |m| m.version == e.version } end
migrations_with_octopus(shard_agnostic = true)
click to toggle source
# File lib/octopus/migration.rb, line 120 def migrations_with_octopus(shard_agnostic = true) migrations = migrations_without_octopus return migrations if !connection.is_a?(Octopus::Proxy) || shard_agnostic migrations.select { |m| m.shards.include?(connection.current_shard.to_sym) } end
run_with_octopus(&block)
click to toggle source
Post RAILS 5.2 Migration
methods
# File lib/octopus/migration.rb, line 100 def run_with_octopus(&block) return run_without_octopus(&block) unless connection.is_a?(Octopus::Proxy) shards = migrations.map(&:shards).flatten.map(&:to_s) connection.send_queries_to_multiple_shards(shards) do run_without_octopus(&block) end rescue ActiveRecord::UnknownMigrationVersionError => e raise unless migrations(true).detect { |m| m.version == e.version } end