Added custom migration generator templates
This allows us to automatically include Gitlab::Database::MigrationHelpers into newly generated migrations.
This commit is contained in:
parent
e867099091
commit
d7cfe1a5bf
|
@ -0,0 +1,32 @@
|
|||
class <%= migration_class_name %> < ActiveRecord::Migration
|
||||
include Gitlab::Database::MigrationHelpers
|
||||
|
||||
# When using the methods "add_concurrent_index" or "add_column_with_default"
|
||||
# you must disable the use of transactions as these methods can not run in an
|
||||
# existing transaction. When using "add_concurrent_index" make sure that this
|
||||
# method is the _only_ method called in the migration, any other changes
|
||||
# should go in a separate migration. This ensures that upon failure _only_ the
|
||||
# index creation failes and can be re-tried or reverted easily.
|
||||
#
|
||||
# To disable transactions uncomment the following line and remove these
|
||||
# comments:
|
||||
# disable_ddl_transaction!
|
||||
|
||||
def change
|
||||
create_table :<%= table_name %> do |t|
|
||||
<% attributes.each do |attribute| -%>
|
||||
<% if attribute.password_digest? -%>
|
||||
t.string :password_digest<%= attribute.inject_options %>
|
||||
<% else -%>
|
||||
t.<%= attribute.type %> :<%= attribute.name %><%= attribute.inject_options %>
|
||||
<% end -%>
|
||||
<% end -%>
|
||||
<% if options[:timestamps] %>
|
||||
t.timestamps null: false
|
||||
<% end -%>
|
||||
end
|
||||
<% attributes_with_index.each do |attribute| -%>
|
||||
add_index :<%= table_name %>, :<%= attribute.index_name %><%= attribute.inject_index_options %>
|
||||
<% end -%>
|
||||
end
|
||||
end
|
|
@ -0,0 +1,52 @@
|
|||
class <%= migration_class_name %> < ActiveRecord::Migration
|
||||
include Gitlab::Database::MigrationHelpers
|
||||
|
||||
# When using the methods "add_concurrent_index" or "add_column_with_default"
|
||||
# you must disable the use of transactions as these methods can not run in an
|
||||
# existing transaction. When using "add_concurrent_index" make sure that this
|
||||
# method is the _only_ method called in the migration, any other changes
|
||||
# should go in a separate migration. This ensures that upon failure _only_ the
|
||||
# index creation failes and can be re-tried or reverted easily.
|
||||
#
|
||||
# To disable transactions uncomment the following line and remove these
|
||||
# comments:
|
||||
# disable_ddl_transaction!
|
||||
|
||||
<%- if migration_action == 'add' -%>
|
||||
def change
|
||||
<% attributes.each do |attribute| -%>
|
||||
<%- if attribute.reference? -%>
|
||||
add_reference :<%= table_name %>, :<%= attribute.name %><%= attribute.inject_options %>
|
||||
<%- else -%>
|
||||
add_column :<%= table_name %>, :<%= attribute.name %>, :<%= attribute.type %><%= attribute.inject_options %>
|
||||
<%- if attribute.has_index? -%>
|
||||
add_index :<%= table_name %>, :<%= attribute.index_name %><%= attribute.inject_index_options %>
|
||||
<%- end -%>
|
||||
<%- end -%>
|
||||
<%- end -%>
|
||||
end
|
||||
<%- elsif migration_action == 'join' -%>
|
||||
def change
|
||||
create_join_table :<%= join_tables.first %>, :<%= join_tables.second %> do |t|
|
||||
<%- attributes.each do |attribute| -%>
|
||||
<%= '# ' unless attribute.has_index? -%>t.index <%= attribute.index_name %><%= attribute.inject_index_options %>
|
||||
<%- end -%>
|
||||
end
|
||||
end
|
||||
<%- else -%>
|
||||
def change
|
||||
<% attributes.each do |attribute| -%>
|
||||
<%- if migration_action -%>
|
||||
<%- if attribute.reference? -%>
|
||||
remove_reference :<%= table_name %>, :<%= attribute.name %><%= attribute.inject_options %>
|
||||
<%- else -%>
|
||||
<%- if attribute.has_index? -%>
|
||||
remove_index :<%= table_name %>, :<%= attribute.index_name %><%= attribute.inject_index_options %>
|
||||
<%- end -%>
|
||||
remove_column :<%= table_name %>, :<%= attribute.name %>, :<%= attribute.type %><%= attribute.inject_options %>
|
||||
<%- end -%>
|
||||
<%- end -%>
|
||||
<%- end -%>
|
||||
end
|
||||
<%- end -%>
|
||||
end
|
Loading…
Reference in New Issue