View difference between Paste ID: YGdfwrZS and

-
+#
 # File:  RAILS_APP_ROOT/lib/extensions/postgres_timestamps.rb
 #
 # What:  Hack up Rails/AR 3 postgresql adapter to do the following:
 #
 #          1. Always qualify timestamp types WITH TIME ZONE
 #          2. Assign DEFAULT CURRENT_TIMESTAMP to created_at, updated_at
 #          3. Assign a special updated_at trigger to tables with that column
 #
 # Note:  XXX This logic assumes that a pg FUNCTION "ar_trg_updated_at()"
 #        exists in the target database !!!
 #
 #        See also http://stackoverflow.com/questions/5098339/activerecord-pg-support-automatic-timestamps-in-ddl
 #
 #  How:  This file is require'd in "config/environment.rb" if and only
 #        if our connection class' ADAPTER_NAME == 'PostgreSQL'
 #
 
 module ActiveRecord
   module ConnectionAdapters
 
     class PostgreSQLAdapter
       [:datetime, :timestamp].each do |type|
         NATIVE_DATABASE_TYPES[type] = 'TIMESTAMP WITH TIME ZONE'
       end
     end
 
     class ColumnDefinition
       alias :original_to_sql :to_sql
       def to_sql
         if ['created_at', 'updated_at'].include?(name.to_s)
           "#{base.quote_column_name(name)} #{sql_type} DEFAULT CURRENT_TIMESTAMP"
         else
           original_to_sql
         end
       end
     end
 
     module SchemaStatements
       alias :original_create_table :create_table
       def create_table(table_name, options = {})
         block = Proc.new rescue Proc.new {|*args| }
         ret = original_create_table(table_name, options, &block)
         if column_exists?(table_name, :updated_at)
           execute <<-eotrigger
             CREATE TRIGGER ar_trg_updated_at
                     BEFORE UPDATE
                         ON #{quote_table_name(table_name)}
               FOR EACH ROW
          EXECUTE PROCEDURE ar_trg_updated_at()
           eotrigger
         end
         ret
       end
     end
 
   end # -- ConnectionAdapters
 end # -- ActiveRecord