aboutsummaryrefslogtreecommitdiffstats
path: root/activerecord/lib/active_record/schema_migration.rb
blob: fd226d5eba5e5382462aebd91904e7509927ee18 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
require 'active_record/scoping/default'
require 'active_record/scoping/named'
require 'active_record/base'

module ActiveRecord
  class SchemaMigration < ActiveRecord::Base

    def self.table_name
      "#{Base.table_name_prefix}schema_migrations#{Base.table_name_suffix}"
    end

    def self.index_name
      "#{Base.table_name_prefix}unique_schema_migrations#{Base.table_name_suffix}"
    end

    def self.create_table
      if connection.table_exists?(table_name)
        cols = connection.columns(table_name).collect { |col| col.name }
        unless cols.include?("migrated_at")
          connection.add_column(table_name, "migrated_at", :datetime)
          q_table_name = connection.quote_table_name(table_name)
          q_timestamp = connection.quoted_date(Time.now)
          connection.update("UPDATE #{q_table_name} SET migrated_at = '#{q_timestamp}' WHERE migrated_at IS NULL")
          connection.change_column(table_name, "migrated_at", :datetime, :null => false)
        end
        unless cols.include?("fingerprint")
          connection.add_column(table_name, "fingerprint", :string, :limit => 32)
        end
        unless cols.include?("name")
          connection.add_column(table_name, "name", :string)
        end
      else
        connection.create_table(table_name, :id => false) do |t|
          t.column "version", :string, :null => false
          t.column "migrated_at", :datetime, :null => false
          t.column "fingerprint", :string, :limit => 32
          t.column "name", :string
        end
        connection.add_index(table_name, "version", :unique => true, :name => index_name)
      end
      reset_column_information
    end

    def self.drop_table
      if connection.index_exists?(table_name, "version", :unique => true, :name => index_name)
        connection.remove_index(table_name, :name => index_name)
      end
      if connection.table_exists?(table_name)
        connection.drop_table(table_name)
      end
    end

    def version
      super.to_i
    end
  end
end