Module: ForAlterStatements

Included in:
ActiveRecord::ConnectionAdapters::DepartureAdapter
Defined in:
lib/active_record/connection_adapters/for_alter.rb

Class Method Summary collapse

Instance Method Summary collapse

Class Method Details

.included(_) ⇒ Object 



5
6
7
 
# File 'lib/active_record/connection_adapters/for_alter.rb', line 5

def included(_)
  STDERR.puts 'Including for_alter statements'
end

Instance Method Details

#add_column_for_alter(table_name, column_name, type, options = {}) ⇒ Object 



78
79
80
81
82
 
# File 'lib/active_record/connection_adapters/for_alter.rb', line 78

def add_column_for_alter(table_name, column_name, type, options = {})
  td = create_table_definition(table_name)
  cd = td.new_column_definition(column_name, type, options)
  schema_creation.accept(ActiveRecord::ConnectionAdapters::AddColumnDefinition.new(cd))
end

#add_index_for_alter(table_name, column_name, options = {}) ⇒ Object 



54
55
56
57
58
59
60
 
# File 'lib/active_record/connection_adapters/for_alter.rb', line 54

def add_index_for_alter(table_name, column_name, options = {})
  index_name, index_type, index_columns, _,
    index_algorithm, index_using = add_index_options(table_name, column_name, options)

  index_algorithm[0, 0] = ', ' if index_algorithm.present?
  "ADD #{index_type} INDEX #{quote_column_name(index_name)} #{index_using} (#{index_columns})#{index_algorithm}"
end

#add_timestamps_for_alter(table_name, options = {}) ⇒ Object 



67
68
69
70
71
72
 
# File 'lib/active_record/connection_adapters/for_alter.rb', line 67

def add_timestamps_for_alter(table_name, options = {})
  [
    add_column_for_alter(table_name, :created_at, :datetime, options),
    add_column_for_alter(table_name, :updated_at, :datetime, options)
  ]
end

#bulk_change_table(table_name, operations) ⇒ Object

:nodoc:



10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/active_record/connection_adapters/for_alter.rb', line 10

def bulk_change_table(table_name, operations) #:nodoc:
  sqls = operations.flat_map do |command, args|
    table = args.shift
    arguments = args

    method = :"#{command}_for_alter"

    raise "Unknown method called : #{method}(#{arguments.inspect})" unless respond_to?(method, true)
    public_send(method, table, *arguments)
  end.join(', ')

  execute("ALTER TABLE #{quote_table_name(table_name)} #{sqls}")
end

#change_column_for_alter(table_name, column_name, type, options = {}) ⇒ Object 



24
25
26
27
28
29
30
31
32
33
34
35
36
37
 
# File 'lib/active_record/connection_adapters/for_alter.rb', line 24

def change_column_for_alter(table_name, column_name, type, options = {})
  column = column_for(table_name, column_name)
  type ||= column.sql_type

  options = {
    default: column.default,
    null: column.null,
    comment: column.comment
  }.merge(options)

  td = create_table_definition(table_name)
  cd = td.new_column_definition(column.name, type, options)
  schema_creation.accept(ActiveRecord::ConnectionAdapters::ChangeColumnDefinition.new(cd, column.name))
end

#remove_column_for_alter(_table_name, column_name, _type = nil, _options = {}) ⇒ Object 



84
85
86
 
# File 'lib/active_record/connection_adapters/for_alter.rb', line 84

def remove_column_for_alter(_table_name, column_name, _type = nil, _options = {})
  "DROP COLUMN #{quote_column_name(column_name)}"
end

#remove_columns_for_alter(table_name, *column_names) ⇒ Object 



88
89
90
 
# File 'lib/active_record/connection_adapters/for_alter.rb', line 88

def remove_columns_for_alter(table_name, *column_names)
  column_names.map { |column_name| remove_column_for_alter(table_name, column_name) }
end

#remove_index_for_alter(table_name, options = {}) ⇒ Object 



62
63
64
65
 
# File 'lib/active_record/connection_adapters/for_alter.rb', line 62

def remove_index_for_alter(table_name, options = {})
  index_name = index_name_for_remove(table_name, options)
  "DROP INDEX #{quote_column_name(index_name)}"
end

#remove_timestamps_for_alter(table_name, _options = {}) ⇒ Object 



74
75
76
 
# File 'lib/active_record/connection_adapters/for_alter.rb', line 74

def remove_timestamps_for_alter(table_name, _options = {})
  [remove_column_for_alter(table_name, :updated_at), remove_column_for_alter(table_name, :created_at)]
end

#rename_column_for_alter(table_name, column_name, new_column_name) ⇒ Object 



39
40
41
42
43
44
45
46
47
48
49
50
51
52
 
# File 'lib/active_record/connection_adapters/for_alter.rb', line 39

def rename_column_for_alter(table_name, column_name, new_column_name)
  column  = column_for(table_name, column_name)
  options = {
    default: column.default,
    null: column.null,
    auto_increment: column.auto_increment?
  }

  columns_sql = "SHOW COLUMNS FROM #{quote_table_name(table_name)} LIKE #{quote(column_name)}"
  current_type = exec_query(columns_sql, 'SCHEMA').first['Type']
  td = create_table_definition(table_name)
  cd = td.new_column_definition(new_column_name, current_type, options)
  schema_creation.accept(ActiveRecord::ConnectionAdapters::ChangeColumnDefinition.new(cd, column.name))
end