Module: ActiveRecord::Tasks::DatabaseTasks

Defined in:
lib/active_record/turntable/active_record_ext/database_tasks.rb

Instance Method Summary collapse

Instance Method Details

#create_all_turntable_cluster ⇒ Object 



6
7
8
9
10
11
 
# File 'lib/active_record/turntable/active_record_ext/database_tasks.rb', line 6

def create_all_turntable_cluster
  each_local_turntable_cluster_configuration { |name, configuration|
    puts "[turntable] *** executing to database: #{configuration['database']}"
    create configuration
  }
end

#create_current_turntable_cluster(environment = env) ⇒ Object 



20
21
22
23
24
25
26
 
# File 'lib/active_record/turntable/active_record_ext/database_tasks.rb', line 20

def create_current_turntable_cluster(environment = env)
  each_current_turntable_cluster_configuration(true, environment) { |name, configuration|
    puts "[turntable] *** executing to database: #{configuration['database']}"
    create configuration
  }
  ActiveRecord::Base.establish_connection environment.to_sym
end

#current_turntable_cluster_configurations(*environments) ⇒ Object 



69
70
71
72
73
74
75
76
77
78
 
# File 'lib/active_record/turntable/active_record_ext/database_tasks.rb', line 69

def current_turntable_cluster_configurations(*environments)
  configurations = []
  environments.each do |environ|
    config = ActiveRecord::Base.configurations[environ]
    %w(shards seq).each do |name|
      configurations += config[name].to_a
    end
  end
  configurations
end

#drop_all_turntable_cluster ⇒ Object 



13
14
15
16
17
18
 
# File 'lib/active_record/turntable/active_record_ext/database_tasks.rb', line 13

def drop_all_turntable_cluster
  each_local_turntable_cluster_configuration { |name, configuration|
    puts "[turntable] *** executing to database: #{configuration['database']}"
    drop configuration
  }
end

#drop_current_turntable_cluster(environment = env) ⇒ Object 



28
29
30
31
32
33
 
# File 'lib/active_record/turntable/active_record_ext/database_tasks.rb', line 28

def drop_current_turntable_cluster(environment = env)
  each_current_turntable_cluster_configuration(true, environment) { |name, configuration|
    puts "[turntable] *** executing to database: #{configuration['database']}"
    drop configuration
  }
end

#each_current_turntable_cluster_configuration(with_test = false, environment = env) ⇒ Object 



46
47
48
49
50
51
52
53
 
# File 'lib/active_record/turntable/active_record_ext/database_tasks.rb', line 46

def each_current_turntable_cluster_configuration(with_test = false, environment = env)
  environments = [environment]
  environments << 'test' if with_test and environment == 'development'

  current_turntable_cluster_configurations(*environments).each do |name, configuration|
    yield(name, configuration) unless configuration['database'].blank?
  end
end

#each_current_turntable_cluster_connected(with_test = false, environment = env) ⇒ Object 



35
36
37
38
39
40
41
42
43
44
 
# File 'lib/active_record/turntable/active_record_ext/database_tasks.rb', line 35

def each_current_turntable_cluster_connected(with_test = false, environment = env)
  each_current_turntable_cluster_configuration(with_test, environment) do |name, configuration|
    ActiveRecord::Base.clear_active_connections!
    ActiveRecord::Base.establish_connection(configuration)
    ActiveRecord::Migration.current_shard = name
    yield(name, configuration)
  end
  ActiveRecord::Base.clear_active_connections!
  ActiveRecord::Base.establish_connection environment.to_sym
end

#each_local_turntable_cluster_configuration ⇒ Object 



55
56
57
58
59
60
61
62
63
64
65
66
67
 
# File 'lib/active_record/turntable/active_record_ext/database_tasks.rb', line 55

def each_local_turntable_cluster_configuration
  ActiveRecord::Base.configurations.keys.each do |k|
    current_turntable_cluster_configurations(k).each do |name, configuration|
      next if configuration['database'].blank?

      if local_database?(configuration)
        yield(name,configuration)
      else
        $stderr.puts "This task only modifies local databases. #{configuration['database']} is on a remote host."
      end
    end
  end
end