diff --git a/lib/generators/ratyrate/templates/migration.rb b/lib/generators/ratyrate/templates/migration.rb index fe47af2..7f84660 100644 --- a/lib/generators/ratyrate/templates/migration.rb +++ b/lib/generators/ratyrate/templates/migration.rb @@ -1,20 +1,24 @@ class CreateRates < ActiveRecord::Migration<%= migration_version %> + def self.up - create_table :rates do |t| - t.belongs_to :rater - t.belongs_to :rateable, :polymorphic => true - t.float :stars, :null => false - t.string :dimension - t.timestamps - end - - add_index :rates, :rater_id - add_index :rates, [:rateable_id, :rateable_type] + create_table :rates do |t| + t.belongs_to :rater + t.belongs_to :rateable, :polymorphic => true + t.float :stars, :null => false + t.string :dimension + t.timestamps end - def self.down - drop_table :rates - end + add_index :rates, :rater_id, name: "rate_rater_id" + add_index :rates, [:rateable_id, :rateable_type], name: "rate_rateable_id" + end + + def self.down + remove_index :rates, [:rateable_id, :rateable_type], name: "rate_rateable_id" + remove_index :rates, :rater_id, name: "rate_rater_id" + + drop_table :rates + end end