PostgreSQL exception library

gistfile1.rb

require 'active_record/connection_adapters/postgresql_adapter'

module ActiveRecord

  class NotNullViolation < ActiveRecord::WrappedDatabaseException
  end

  class CheckViolation < ActiveRecord::WrappedDatabaseException
  end

  module ConnectionAdapters
    class PostgreSQLAdapter < AbstractAdapter
      def translate_exception(exception, message)
        raise exception unless exception.respond_to?(:result)
        case exception.result.try(:error_field, PGresult::PG_DIAG_SQLSTATE)
        when UNIQUE_VIOLATION
          RecordNotUnique.new(message, exception)
        when FOREIGN_KEY_VIOLATION
          InvalidForeignKey.new(message, exception)


        # Custom error codes based on PostgreSQL documentation:
        # http://www.postgresql.org/docs/9.2/static/errcodes-appendix.html
        when '23502'
          NotNullViolation.new(message, exception)
        when '23514'
          CheckViolation.new(message, exception)
          
        else
          super
        end
      end

    end
  end
end