Class: Hatio::Generators::MigrationUtil
- Inherits:
-
Object
- Object
- Hatio::Generators::MigrationUtil
- Defined in:
- lib/generators/hatio/util/migration_util.rb
Class Method Summary collapse
-
.add_migration_columns(columns, pk_col_name) ⇒ Object
입력된 attributes로 부터 migration fields를 추가, columns는 Entity의 Entity Columns.
-
.add_migration_fields(attributes) ⇒ Object
입력된 attributes로 부터 migration fields를 추가.
-
.add_migration_index(table_name, attributes) ⇒ Object
add migration index.
-
.create_table_migration(class_name, table_name, attributes, options, pk_col_name) ⇒ Object
table migration 내용 생성.
-
.generated_attributes(model_attributes) ⇒ Object
입력한 attributes들로 부터 GeneratedAttribute를 생성하여 array로 리턴.
-
.id_option_str(id_type) ⇒ Object
id option str.
-
.next_migration_number ⇒ Object
generate db migration file number.
-
.remove_migration_index(table_name, attributes) ⇒ Object
remove migration index.
Class Method Details
.add_migration_columns(columns, pk_col_name) ⇒ Object
입력된 attributes로 부터 migration fields를 추가, columns는 Entity의 Entity Columns
143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 |
# File 'lib/generators/hatio/util/migration_util.rb', line 143 def self.add_migration_columns(columns, pk_col_name) output = "" userstampable, = false, false columns.each do |column| if(column.name == 'id') elsif(column.name == 'domain_id') output << "\t\t\tt.references :domain, :null => false\n" elsif(column.name == 'created_at' || column.name == 'updated_at') = true elsif(column.name == 'creator_id' || column.name == 'updater_id') userstampable = true elsif(column.ref_type == "Entity") if(column.name.ends_with?("_id")) ref_name = column.name[0 .. column.name.rindex('_') - 1] output << "\t\t\tt.references :#{ref_name}\n" else output << "\t\t\tt.#{column.col_type} :#{column.name}\n" end else output << "\t\t\tt.#{column.col_type} :#{column.name}" output << ", :null => false" if(!column.nullable.nil? && !column.nullable) output << ", :limit => #{column.col_size}" if(column.col_type == 'string' && column.col_size && column.col_size > 0) output << ", :default => #{column.def_val}" if(column.def_val && !column.def_val.empty?) output << "\n" end end output << "\t\t\tt.userstamps\n" if userstampable output << "\t\t\tt.timestamps\n" if output end |
.add_migration_fields(attributes) ⇒ Object
입력된 attributes로 부터 migration fields를 추가
39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
# File 'lib/generators/hatio/util/migration_util.rb', line 39 def self.add_migration_fields(attributes) output, userstampable, = "", false, false attributes.each do |attr| if (attr.name == 'id') elsif(attr.name == 'domain_id') output << "\t\t\tt.references :domain, :null => false\n" elsif(attr.name == 'creator_id' || attr.name == 'updater_id') userstampable = true elsif(attr.name == 'created_at' || attr.name == 'updated_at') = true elsif(attr.name == 'name') output << "\t\t\tt.references :name, :null => false, :limit => 64\n" else begin output << "\t\t\tt.#{attr.type} :#{attr.name}\n" rescue output << "\t\t\tt.#{attr.col_type} :#{attr.name}\n" end end end output << "\t\t\tt.userstamps\n" if userstampable output << "\t\t\tt.timestamps\n" if output end |
.add_migration_index(table_name, attributes) ⇒ Object
add migration index
69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
# File 'lib/generators/hatio/util/migration_util.rb', line 69 def self.add_migration_index(table_name, attributes) output, index_seq = "", 0 domain_id_attr = attributes.find { |attr| attr.name == 'domain_id' } name_attr = attributes.find { |attr| attr.name == 'name' } updated_at_attr = attributes.find { |attr| attr.name == 'updated_at' } uniq_cols = attributes.select { |col| col.uniq_rank > 0 }.sort_by(&:uniq_rank) if(!uniq_cols.empty?) uniq_col_names = uniq_cols.map { |col| ":#{col.name}" }.join(",") output << "\t\tadd_index :#{table_name}, [#{uniq_col_names}], :unique => true, :name => :ix_#{table_name}_#{index_seq}\n" index_seq += 1 end if(domain_id_attr && updated_at_attr) output << "\t\tadd_index :#{table_name}, [:domain_id, :updated_at], :name => :ix_#{table_name}_#{index_seq}\n" index_seq += 1 elsif(updated_at_attr) output << "\t\tadd_index :#{table_name}, [:updated_at], :name => :ix_#{table_name}_#{index_seq}\n" index_seq += 1 end output end |
.create_table_migration(class_name, table_name, attributes, options, pk_col_name) ⇒ Object
table migration 내용 생성
123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 |
# File 'lib/generators/hatio/util/migration_util.rb', line 123 def self.create_table_migration(class_name, table_name, attributes, , pk_col_name) id_option = self.id_option_str(.id_type) id_option = ":id => false" if(pk_col_name && pk_col_name != 'id') output = "class Create#{class_name.pluralize} < ActiveRecord::Migration\n\n" output << "\tdef change\n" output << "\t\tcreate_table #{table_name.to_sym.inspect} #{id_option} do |t|\n" output << "#{Hatio::Generators::MigrationUtil.add_migration_columns(attributes, pk_col_name)}" output << "\t\tend\n\n" output << "#{MigrationUtil.add_migration_index(table_name, attributes)}" output << "\tend\n\n" output << "end\n" output end |
.generated_attributes(model_attributes) ⇒ Object
입력한 attributes들로 부터 GeneratedAttribute를 생성하여 array로 리턴
8 9 10 11 12 13 14 |
# File 'lib/generators/hatio/util/migration_util.rb', line 8 def self.generated_attributes(model_attributes) attrs = [] model_attributes.each do |attribute| attrs << Rails::Generators::GeneratedAttribute.new(*attribute.split(":")) if attribute.include?(":") end return attrs end |
.id_option_str(id_type) ⇒ Object
id option str
26 27 28 29 30 31 32 33 34 |
# File 'lib/generators/hatio/util/migration_util.rb', line 26 def self.id_option_str(id_type) if(id_type == 'auto-increment') return '' elsif(id_type == 'none') return ",:id => false" else return ",:id => :#{id_type}" end end |
.next_migration_number ⇒ Object
generate db migration file number
19 20 21 |
# File 'lib/generators/hatio/util/migration_util.rb', line 19 def self.next_migration_number Time.now.utc.strftime("%Y%m%d%H%M%S") end |
.remove_migration_index(table_name, attributes) ⇒ Object
remove migration index
96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 |
# File 'lib/generators/hatio/util/migration_util.rb', line 96 def self.remove_migration_index(table_name, attributes) output, index_seq = "", 0 domain_id_attr = attributes.find { |attr| attr.name == 'domain_id' } name_attr = attributes.find { |attr| attr.name == 'name' } updated_at_attr = attributes.find { |attr| attr.name == 'updated_at' } uniq_cols = attributes.select { |col| col.uniq_rank > 0 }.sort_by(&:uniq_rank) if(!uniq_cols.empty?) uniq_col_names = uniq_cols.map { |col| ":#{col.name}" }.join(",") output << "\t\tremove_index :#{table_name}, [#{uniq_col_names}], :unique => true, :name => :ix_#{table_name}_#{index_seq}\n" index_seq += 1 end if(domain_id_attr && updated_at_attr) output << "\t\tremove_index :#{table_name}, [:domain_id, :updated_at], :name => :ix_#{table_name}_#{index_seq}\n" index_seq += 1 elsif(updated_at_attr) output << "\t\tremove_index :#{table_name}, [:updated_at], :name => :ix_#{table_name}_#{index_seq}\n" index_seq += 1 end output end |