lib/migrate.rb

   1 module ActiveRecord
   2   module ConnectionAdapters
   3     module SchemaStatements
   4       def quote_column_names(column_name)
   5         Array(column_name).map { |e| quote_column_name(e) }.join(", ")
   6       end
   7
   8       def add_primary_key(table_name, column_name, options = {})
   9         column_names = Array(column_name)
  10         quoted_column_names = column_names.map { |e| quote_column_name(e) }.join(", ")
  11         execute "ALTER TABLE #{table_name} ADD PRIMARY KEY (#{quoted_column_names})"
  12       end
  13
  14       def remove_primary_key(table_name)
  15         execute "ALTER TABLE #{table_name} DROP PRIMARY KEY"
  16       end
  17
  18       def add_foreign_key(table_name, column_name, reftbl, refcol = nil)
  19         execute "ALTER TABLE #{table_name} ADD " +
  20           "FOREIGN KEY (#{quote_column_names(column_name)}) " +
  21           "REFERENCES #{reftbl} (#{quote_column_names(refcol || column_name)})"
  22       end
  23
  24       def remove_foreign_key(table_name, column_name, reftbl, refcol = nil)
  25         execute "ALTER TABLE #{table_name} DROP " +
  26           "CONSTRAINT #{table_name}_#{column_name[0]}_fkey"
  27       end
  28
  29       alias_method :old_options_include_default?, :options_include_default?
  30
  31       def options_include_default?(options)
  32         return false if options[:options] =~ /AUTO_INCREMENT/i
  33         return old_options_include_default?(options)
  34       end
  35
  36       alias_method :old_add_column_options!, :add_column_options!
  37
  38       def add_column_options!(sql, options)
  39         sql << " UNSIGNED" if options[:unsigned]
  40         old_add_column_options!(sql, options)
  41         sql << " #{options[:options]}"
  42       end
  43     end
  44
  45     class PostgreSQLAdapter
  46       alias_method :old_native_database_types, :native_database_types
  47
  48       def native_database_types
  49         types = old_native_database_types
  50         types[:double] = { :name => "double precision" }
  51         types[:integer_pk] = { :name => "serial PRIMARY KEY" }
  52         types[:bigint_pk] = { :name => "bigserial PRIMARY KEY" }
  53         types[:bigint_pk_64] = { :name => "bigserial PRIMARY KEY" }
  54         types[:bigint_auto_64] = { :name => "bigint" } #fixme: need autoincrement?
  55         types[:bigint_auto_11] = { :name => "bigint" } #fixme: need autoincrement?
  56         types[:bigint_auto_20] = { :name => "bigint" } #fixme: need autoincrement?
  57         types[:four_byte_unsigned] = { :name => "bigint" } # meh
  58         types[:inet] = { :name=> "inet" }
  59
  60         enumerations.each_key do |e|
  61           types[e.to_sym]= { :name => e }
  62         end
  63
  64         types
  65       end
  66
  67       def myisam_table
  68         return { :id => false, :force => true, :options => ""}
  69       end
  70
  71       def innodb_table
  72         return { :id => false, :force => true, :options => ""}
  73       end
  74
  75       def innodb_option
  76         return ""
  77       end
  78
  79       def change_engine (table_name, engine)
  80       end
  81
  82       def add_fulltext_index (table_name, column)
  83         execute "CREATE INDEX #{table_name}_#{column}_idx on #{table_name} (#{column})"
  84       end
  85
  86       def enumerations
  87         @enumerations ||= Hash.new
  88       end
  89
  90       def create_enumeration (enumeration_name, values)
  91         enumerations[enumeration_name] = values
  92         execute "create type #{enumeration_name} as enum ('#{values.join '\',\''}')"
  93       end
  94
  95       def drop_enumeration (enumeration_name)
  96         execute "drop type #{enumeration_name}"
  97         enumerations.delete(enumeration_name)
  98       end
  99
 100       def alter_primary_key(table_name, new_columns)
 101         execute "alter table #{table_name} drop constraint #{table_name}_pkey; alter table #{table_name} add primary key (#{new_columns.join(',')})"
 102       end
 103
 104       def interval_constant(interval)
 105         "'#{interval}'::interval"
 106       end
 107
 108       def add_index(table_name, column_name, options = {})
 109         column_names = Array(column_name)
 110         index_name   = index_name(table_name, :column => column_names)
 111
 112         if Hash === options # legacy support, since this param was a string
 113           index_type = options[:unique] ? "UNIQUE" : ""
 114           index_name = options[:name] || index_name
 115           index_method = options[:method] || "BTREE"
 116         else
 117           index_type = options
 118         end
 119
 120         quoted_column_names = column_names.map { |e| quote_column_name(e) }
 121         if Hash === options and options[:lowercase]
 122           quoted_column_names = quoted_column_names.map { |e| "LOWER(#{e})" }
 123         end
 124         quoted_column_names = quoted_column_names.join(", ")
 125
 126         execute "CREATE #{index_type} INDEX #{quote_column_name(index_name)} ON #{quote_table_name(table_name)} USING #{index_method} (#{quoted_column_names})"
 127       end
 128     end
 129   end
 130 end