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