Module: ActiveRecordShards::ActualMigrationExtension

Defined in:
lib/active_record_shards/migration.rb

Overview

ok, so some ‘splaining to do. Rails 3.1 puts the migrate() method on the instance of the migration, where it should have been. But this makes our monkey patch incompatible. So we’re forced to either include or extend this.

Instance Method Summary collapse

Instance Method Details

#migrate_with_forced_shard(direction) ⇒ Object 



88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
 
# File 'lib/active_record_shards/migration.rb', line 88

def migrate_with_forced_shard(direction)
  if migration_shard.blank?
    raise RuntimeError, "#{self.name}: Can't run migrations without a shard spec: this may be :all, :none,
             or a specific shard (for data-fixups).  please call shard(arg) in your migration."
  end

  shard = ActiveRecord::Base.current_shard_selection.shard

  if shard.nil?
    return if migration_shard != :none
  else
    return if migration_shard == :none
    return if migration_shard != :all && migration_shard.to_s != shard.to_s
  end

  migrate_without_forced_shard(direction)
end

#migration_shard ⇒ Object 



106
107
108
 
# File 'lib/active_record_shards/migration.rb', line 106

def migration_shard
  self.class.migration_shard
end