Module: Octopus::Migration::ClassMethods

Defined in:
lib/octopus/migration.rb

Instance Method Summary collapse

Instance Method Details

#shards ⇒ Object 



42
43
44
45
46
47
48
49
50
51
52
53
54
55
 
# File 'lib/octopus/migration.rb', line 42

def shards
  shards = Set.new

  if (groups = (current_group_specified ? current_group : Octopus.config[:default_migration_group]))
    Array.wrap(groups).each do |group|
      group_shards = connection.shards_for_group(group)
      shards.merge(group_shards) if group_shards
    end
  elsif (shard = current_shard)
    shards.merge(Array.wrap(shard))
  end

  shards.to_a.presence || [:master]
end

#using(*args) ⇒ Object 



27
28
29
30
31
32
 
# File 'lib/octopus/migration.rb', line 27

def using(*args)
  return self unless connection.is_a?(Octopus::Proxy)

  self.current_shard = args
  self
end

#using_group(*groups) ⇒ Object 



34
35
36
37
38
39
40
 
# File 'lib/octopus/migration.rb', line 34

def using_group(*groups)
  return self unless connection.is_a?(Octopus::Proxy)

  self.current_group = groups
  self.current_group_specified = true
  self
end