activerecord-jdbc-alt-adapter 50.3.0-java

data/lib/arjdbc/postgresql.rb

@@ -0,0 +1,3 @@
+require 'arjdbc'
+require 'arjdbc/postgresql/adapter'
+require 'arjdbc/postgresql/connection_methods'

data/lib/arjdbc/postgresql/adapter.rb

@@ -0,0 +1,687 @@
+# frozen_string_literal: false
+ArJdbc.load_java_part :PostgreSQL
+
+require 'ipaddr'
+require 'active_record/connection_adapters/abstract_adapter'
+require 'active_record/connection_adapters/postgresql/column'
+require 'active_record/connection_adapters/postgresql/explain_pretty_printer'
+require 'active_record/connection_adapters/postgresql/quoting'
+require 'active_record/connection_adapters/postgresql/referential_integrity'
+require 'active_record/connection_adapters/postgresql/schema_dumper'
+require 'active_record/connection_adapters/postgresql/schema_statements'
+require 'active_record/connection_adapters/postgresql/type_metadata'
+require 'active_record/connection_adapters/postgresql/utils'
+require 'arjdbc/abstract/core'
+require 'arjdbc/abstract/connection_management'
+require 'arjdbc/abstract/database_statements'
+require 'arjdbc/abstract/statement_cache'
+require 'arjdbc/abstract/transaction_support'
+require 'arjdbc/postgresql/base/array_decoder'
+require 'arjdbc/postgresql/base/array_encoder'
+require 'arjdbc/postgresql/name'
+
+module ArJdbc
+  # Strives to provide Rails built-in PostgreSQL adapter (API) compatibility.
+  module PostgreSQL
+
+    require 'arjdbc/postgresql/column'
+    require 'arel/visitors/postgresql_jdbc'
+    # @private
+    IndexDefinition = ::ActiveRecord::ConnectionAdapters::IndexDefinition
+
+    # @private
+    ForeignKeyDefinition = ::ActiveRecord::ConnectionAdapters::ForeignKeyDefinition
+
+    # @private
+    Type = ::ActiveRecord::Type
+
+    # @see ActiveRecord::ConnectionAdapters::JdbcAdapter#jdbc_connection_class
+    def self.jdbc_connection_class
+      ::ActiveRecord::ConnectionAdapters::PostgreSQLJdbcConnection
+    end
+
+    # @see ActiveRecord::ConnectionAdapters::JdbcAdapter#jdbc_column_class
+    def jdbc_column_class; ::ActiveRecord::ConnectionAdapters::PostgreSQLColumn end
+
+    ADAPTER_NAME = 'PostgreSQL'.freeze
+
+    def adapter_name
+      ADAPTER_NAME
+    end
+
+    # TODO: Update this to pull info from the DatabaseMetaData object?
+    def postgresql_version
+      @postgresql_version ||=
+        begin
+          version = @connection.database_product
+          if match = version.match(/([\d\.]*\d).*?/)
+            version = match[1].split('.').map(&:to_i)
+            # PostgreSQL version representation does not have more than 4 digits
+            # From version 10 onwards, PG has changed its versioning policy to
+            # limit it to only 2 digits. i.e. in 10.x, 10 being the major
+            # version and x representing the patch release
+            # Refer to:
+            #   https://www.postgresql.org/support/versioning/
+            #   https://www.postgresql.org/docs/10/static/libpq-status.html -> PQserverVersion()
+            # for more info
+
+            if version.size >= 3
+              (version[0] * 100 + version[1]) * 100 + version[2]
+            elsif version.size == 2
+              if version[0] >= 10
+                version[0] * 100 * 100 + version[1]
+              else
+                (version[0] * 100 + version[1]) * 100
+              end
+            elsif version.size == 1
+              version[0] * 100 * 100
+            else
+              0
+            end
+          else
+            0
+          end
+        end
+    end
+
+    def redshift?
+      # SELECT version() :
+      #  PostgreSQL 8.0.2 on i686-pc-linux-gnu, compiled by GCC gcc (GCC) 3.4.2 20041017 (Red Hat 3.4.2-6.fc3), Redshift 1.0.647
+      if ( redshift = config[:redshift] ).nil?
+        redshift = !! (@connection.database_product || '').index('Redshift')
+      end
+      redshift
+    end
+    private :redshift?
+
+    def use_insert_returning?
+      if @use_insert_returning.nil?
+        @use_insert_returning = supports_insert_with_returning?
+      end
+      @use_insert_returning
+    end
+
+    def set_client_encoding(encoding)
+      ActiveRecord::Base.logger.warn "client_encoding is set by the driver and should not be altered, ('#{encoding}' ignored)"
+      ActiveRecord::Base.logger.debug "Set the 'allowEncodingChanges' driver property (e.g. using config[:properties]) if you need to override the client encoding when doing a copy."
+    end
+
+    # Configures the encoding, verbosity, schema search path, and time zone of the connection.
+    # This is called on `connection.connect` and should not be called manually.
+    def configure_connection
+      #if encoding = config[:encoding]
+        # The client_encoding setting is set by the driver and should not be altered.
+        # If the driver detects a change it will abort the connection.
+        # see http://jdbc.postgresql.org/documentation/91/connect.html
+        # self.set_client_encoding(encoding)
+      #end
+      self.client_min_messages = config[:min_messages] || 'warning'
+      self.schema_search_path = config[:schema_search_path] || config[:schema_order]
+
+      # Use standard-conforming strings if available so we don't have to do the E'...' dance.
+      set_standard_conforming_strings
+
+      # If using Active Record's time zone support configure the connection to return
+      # TIMESTAMP WITH ZONE types in UTC.
+      # (SET TIME ZONE does not use an equals sign like other SET variables)
+      if ActiveRecord::Base.default_timezone == :utc
+        execute("SET time zone 'UTC'", 'SCHEMA')
+      elsif tz = local_tz
+        execute("SET time zone '#{tz}'", 'SCHEMA')
+      end unless redshift?
+
+      # SET statements from :variables config hash
+      # http://www.postgresql.org/docs/8.3/static/sql-set.html
+      (config[:variables] || {}).map do |k, v|
+        if v == ':default' || v == :default
+          # Sets the value to the global or compile default
+          execute("SET SESSION #{k} TO DEFAULT", 'SCHEMA')
+        elsif ! v.nil?
+          execute("SET SESSION #{k} TO #{quote(v)}", 'SCHEMA')
+        end
+      end
+    end
+
+    # @private
+    ActiveRecordError = ::ActiveRecord::ActiveRecordError
+
+    NATIVE_DATABASE_TYPES = {
+      bigserial:    'bigserial',
+      primary_key:  'serial primary key',
+      bigint:       { name: 'bigint' },
+      binary:       { name: 'bytea' },
+      bit:          { name: 'bit' },
+      bit_varying:  { name: 'bit varying' },
+      boolean:      { name: 'boolean' },
+      box:          { name: 'box' },
+      char:         { name: 'char' },
+      cidr:         { name: 'cidr' },
+      circle:       { name: 'circle' },
+      citext:       { name: 'citext' },
+      date:         { name: 'date' },
+      daterange:    { name: 'daterange' },
+      datetime:     { name: 'timestamp' },
+      decimal:      { name: 'decimal' }, # :limit => 1000
+      float:        { name: 'float' },
+      hstore:       { name: 'hstore' },
+      inet:         { name: 'inet' },
+      int4range:    { name: 'int4range' },
+      int8range:    { name: 'int8range' },
+      integer:      { name: 'integer' },
+      interval:     { name: 'interval' }, # This doesn't get added to AR's postgres adapter until 5.1 but it fixes broken tests in 5.0 ...
+      json:         { name: 'json' },
+      jsonb:        { name: 'jsonb' },
+      line:         { name: 'line' },
+      lseg:         { name: 'lseg' },
+      ltree:        { name: 'ltree' },
+      macaddr:      { name: 'macaddr' },
+      money:        { name: 'money' },
+      numeric:      { name: 'numeric' },
+      numrange:     { name: 'numrange' },
+      path:         { name: 'path' },
+      point:        { name: 'point' },
+      polygon:      { name: 'polygon' },
+      serial:       { name: 'serial' }, # auto-inc integer, bigserial, smallserial
+      string:       { name: 'character varying' },
+      text:         { name: 'text' },
+      time:         { name: 'time' },
+      timestamp:    { name: 'timestamp' },
+      tsrange:      { name: 'tsrange' },
+      tstzrange:    { name: 'tstzrange' },
+      tsvector:     { name: 'tsvector' },
+      uuid:         { name: 'uuid' },
+      xml:          { name: 'xml' }
+    }
+
+    def native_database_types
+      NATIVE_DATABASE_TYPES
+    end
+
+    def valid_type?(type)
+      !native_database_types[type].nil?
+    end
+
+    # Enable standard-conforming strings if available.
+    def set_standard_conforming_strings
+      self.standard_conforming_strings=(true)
+    end
+
+    # Enable standard-conforming strings if available.
+    def standard_conforming_strings=(enable)
+      client_min_messages = self.client_min_messages
+      begin
+        self.client_min_messages = 'panic'
+        value = enable ? "on" : "off"
+        execute("SET standard_conforming_strings = #{value}", 'SCHEMA')
+        @standard_conforming_strings = ( value == "on" )
+      rescue
+        @standard_conforming_strings = :unsupported
+      ensure
+        self.client_min_messages = client_min_messages
+      end
+    end
+
+    def standard_conforming_strings?
+      if @standard_conforming_strings.nil?
+        client_min_messages = self.client_min_messages
+        begin
+          self.client_min_messages = 'panic'
+          value = select_one('SHOW standard_conforming_strings', 'SCHEMA')['standard_conforming_strings']
+          @standard_conforming_strings = ( value == "on" )
+        rescue
+          @standard_conforming_strings = :unsupported
+        ensure
+          self.client_min_messages = client_min_messages
+        end
+      end
+      @standard_conforming_strings == true # return false if :unsupported
+    end
+
+    def supports_ddl_transactions?; true end
+
+    def supports_advisory_locks?; true end
+
+    def supports_explain?; true end
+
+    def supports_expression_index?; true end
+
+    def supports_foreign_keys?; true end
+
+    def supports_index_sort_order?; true end
+
+    def supports_migrations?; true end
+
+    def supports_partial_index?; true end
+
+    def supports_primary_key?; true end # Supports finding primary key on non-Active Record tables
+
+    def supports_savepoints?; true end
+
+    def supports_transaction_isolation?(level = nil); true end
+
+    def supports_views?; true end
+
+    def supports_datetime_with_precision?; true end
+
+    def supports_comments?; true end
+
+    # Does PostgreSQL support standard conforming strings?
+    def supports_standard_conforming_strings?
+      standard_conforming_strings?
+      @standard_conforming_strings != :unsupported
+    end
+
+    def supports_hex_escaped_bytea?
+      postgresql_version >= 90000
+    end
+
+    def supports_materialized_views?
+      postgresql_version >= 90300
+    end
+
+    def supports_json?
+      postgresql_version >= 90200
+    end
+
+    def supports_insert_with_returning?
+      postgresql_version >= 80200
+    end
+
+    # Range data-types weren't introduced until PostgreSQL 9.2.
+    def supports_ranges?
+      postgresql_version >= 90200
+    end
+
+    def supports_extensions?
+      postgresql_version >= 90200
+    end
+
+    def enable_extension(name)
+      execute("CREATE EXTENSION IF NOT EXISTS \"#{name}\"")
+    end
+
+    def disable_extension(name)
+      execute("DROP EXTENSION IF EXISTS \"#{name}\" CASCADE")
+    end
+
+    def extension_enabled?(name)
+      if supports_extensions?
+        rows = select_rows("SELECT EXISTS(SELECT * FROM pg_available_extensions WHERE name = '#{name}' AND installed_version IS NOT NULL)", 'SCHEMA')
+        available = rows.first.first # true/false or 't'/'f'
+        available == true || available == 't'
+      end
+    end
+
+    def extensions
+      if supports_extensions?
+        rows = select_rows "SELECT extname from pg_extension", "SCHEMA"
+        rows.map { |row| row.first }
+      else
+        []
+      end
+    end
+
+    def index_algorithms
+      { :concurrently => 'CONCURRENTLY' }
+    end
+
+    # Set the authorized user for this session.
+    def session_auth=(user)
+      clear_cache!
+      execute "SET SESSION AUTHORIZATION #{user}"
+    end
+
+    # Came from postgres_adapter
+    def get_advisory_lock(lock_id) # :nodoc:
+      unless lock_id.is_a?(Integer) && lock_id.bit_length <= 63
+        raise(ArgumentError, "Postgres requires advisory lock ids to be a signed 64 bit integer")
+      end
+      select_value("SELECT pg_try_advisory_lock(#{lock_id});")
+    end
+
+    # Came from postgres_adapter
+    def release_advisory_lock(lock_id) # :nodoc:
+      unless lock_id.is_a?(Integer) && lock_id.bit_length <= 63
+        raise(ArgumentError, "Postgres requires advisory lock ids to be a signed 64 bit integer")
+      end
+      select_value("SELECT pg_advisory_unlock(#{lock_id})")
+    end
+
+    # Returns the configured supported identifier length supported by PostgreSQL
+    def max_identifier_length
+      @max_identifier_length ||= select_one('SHOW max_identifier_length', 'SCHEMA'.freeze)['max_identifier_length'].to_i
+    end
+    alias table_alias_length max_identifier_length
+    alias index_name_length max_identifier_length
+
+    def exec_insert(sql, name, binds, pk = nil, sequence_name = nil)
+      val = super
+      if !use_insert_returning? && pk
+        unless sequence_name
+          table_ref = extract_table_ref_from_insert_sql(sql)
+          sequence_name = default_sequence_name(table_ref, pk)
+          return val unless sequence_name
+        end
+        last_insert_id_result(sequence_name)
+      else
+        val
+      end
+    end
+
+    def explain(arel, binds = [])
+      sql = "EXPLAIN #{to_sql(arel, binds)}"
+      ActiveRecord::ConnectionAdapters::PostgreSQL::ExplainPrettyPrinter.new.pp(exec_query(sql, 'EXPLAIN', binds))
+    end
+
+    # @note Only for "better" AR 4.0 compatibility.
+    # @private
+    def query(sql, name = nil)
+      log(sql, name) do
+        result = []
+        @connection.execute_query_raw(sql, []) do |*values|
+          # We need to use #deep_dup here because it appears that
+          # the java method is reusing an object in some cases
+          # which makes all of the entries in the "result"
+          # array end up with the same values as the last row
+          result << values.deep_dup
+        end
+        result
+      end
+    end
+
+    # We need to make sure to deallocate all the prepared statements
+    # since apparently calling close on the statement object
+    # doesn't always free the server resources and calling
+    # 'DISCARD ALL' fails if we are inside a transaction
+    def clear_cache!
+      super
+      @connection.execute 'DEALLOCATE ALL' if supports_statement_cache? && @connection.active?
+    end
+
+    def reset!
+      clear_cache!
+      reset_transaction
+      @connection.rollback # Have to deal with rollbacks differently than the AR adapter
+      @connection.execute 'DISCARD ALL'
+      @connection.configure_connection
+    end
+
+    def default_sequence_name(table_name, pk = "id") #:nodoc:
+      serial_sequence(table_name, pk)
+    rescue ActiveRecord::StatementInvalid
+      %Q("#{table_name}_#{pk}_seq")
+    end
+
+    def last_insert_id_result(sequence_name)
+      exec_query("SELECT currval('#{sequence_name}')", 'SQL')
+    end
+
+    def all_schemas
+      select('SELECT nspname FROM pg_namespace').map { |row| row["nspname"] }
+    end
+
+    # Returns the current client message level.
+    def client_min_messages
+      return nil if redshift? # not supported on Redshift
+      # Need to use #execute so we don't try to access the type map before it is initialized
+      execute('SHOW client_min_messages', 'SCHEMA').values.first.first
+    end
+
+    # Set the client message level.
+    def client_min_messages=(level)
+      # NOTE: for now simply ignore the writer (no warn on Redshift) so that
+      # the AR copy-pasted PpstgreSQL parts stay the same as much as possible
+      return nil if redshift? # not supported on Redshift
+      execute("SET client_min_messages TO '#{level}'", 'SCHEMA')
+    end
+
+    # ORDER BY clause for the passed order option.
+    #
+    # PostgreSQL does not allow arbitrary ordering when using DISTINCT ON,
+    # so we work around this by wrapping the SQL as a sub-select and ordering
+    # in that query.
+    def add_order_by_for_association_limiting!(sql, options)
+      return sql if options[:order].blank?
+
+      order = options[:order].split(',').collect { |s| s.strip }.reject(&:blank?)
+      order.map! { |s| 'DESC' if s =~ /\bdesc$/i }
+      order = order.zip((0...order.size).to_a).map { |s,i| "id_list.alias_#{i} #{s}" }.join(', ')
+
+      sql.replace "SELECT * FROM (#{sql}) AS id_list ORDER BY #{order}"
+    end
+
+    # @note #quote_string implemented as native
+
+    def escape_bytea(string)
+      return unless string
+      if supports_hex_escaped_bytea?
+        "\\x#{string.unpack("H*")[0]}"
+      else
+        result = ''
+        string.each_byte { |c| result << sprintf('\\\\%03o', c) }
+        result
+      end
+    end
+
+    # @override
+    def quote_table_name(name)
+      schema, name_part = extract_pg_identifier_from_name(name.to_s)
+
+      unless name_part
+        quote_column_name(schema)
+      else
+        table_name, name_part = extract_pg_identifier_from_name(name_part)
+        "#{quote_column_name(schema)}.#{quote_column_name(table_name)}"
+      end
+    end
+
+    # @note #quote_column_name implemented as native
+    alias_method :quote_schema_name, :quote_column_name
+
+    # Need to clear the cache even though the AR adapter doesn't for some reason
+    def remove_column(table_name, column_name, type = nil, options = {})
+      super
+      clear_cache!
+    end
+
+    # @private
+    def column_for(table_name, column_name)
+      column_name = column_name.to_s
+      for column in columns(table_name)
+        return column if column.name == column_name
+      end
+      nil
+    end
+
+    # Returns the list of a table's column names, data types, and default values.
+    #
+    # If the table name is not prefixed with a schema, the database will
+    # take the first match from the schema search path.
+    #
+    # Query implementation notes:
+    #  - format_type includes the column size constraint, e.g. varchar(50)
+    #  - ::regclass is a function that gives the id for a table name
+    def column_definitions(table_name)
+      select_rows(<<-end_sql, 'SCHEMA')
+        SELECT a.attname, format_type(a.atttypid, a.atttypmod),
+               pg_get_expr(d.adbin, d.adrelid), a.attnotnull, a.atttypid, a.atttypmod,
+               (SELECT c.collname FROM pg_collation c, pg_type t
+                 WHERE c.oid = a.attcollation AND t.oid = a.atttypid
+                  AND a.attcollation <> t.typcollation),
+               col_description(a.attrelid, a.attnum) AS comment
+          FROM pg_attribute a
+          LEFT JOIN pg_attrdef d ON a.attrelid = d.adrelid AND a.attnum = d.adnum
+         WHERE a.attrelid = '#{quote_table_name(table_name)}'::regclass
+           AND a.attnum > 0 AND NOT a.attisdropped
+         ORDER BY a.attnum
+      end_sql
+    end
+    private :column_definitions
+
+    def truncate(table_name, name = nil)
+      execute "TRUNCATE TABLE #{quote_table_name(table_name)}", name
+    end
+
+    # @private
+    def column_name_for_operation(operation, node)
+      case operation
+      when 'maximum' then 'max'
+      when 'minimum' then 'min'
+      when 'average' then 'avg'
+      else operation.downcase
+      end
+    end
+
+    private
+
+    # Pulled from ActiveRecord's Postgres adapter and modified to use execute
+    def can_perform_case_insensitive_comparison_for?(column)
+      @case_insensitive_cache ||= {}
+      @case_insensitive_cache[column.sql_type] ||= begin
+        sql = <<-end_sql
+              SELECT exists(
+                SELECT * FROM pg_proc
+                WHERE proname = 'lower'
+                  AND proargtypes = ARRAY[#{quote column.sql_type}::regtype]::oidvector
+              ) OR exists(
+                SELECT * FROM pg_proc
+                INNER JOIN pg_cast
+                  ON ARRAY[casttarget]::oidvector = proargtypes
+                WHERE proname = 'lower'
+                  AND castsource = #{quote column.sql_type}::regtype
+              )
+        end_sql
+        select_value(sql, 'SCHEMA')
+      end
+    end
+
+    def translate_exception(exception, message)
+      case exception.message
+      when /duplicate key value violates unique constraint/
+        ::ActiveRecord::RecordNotUnique.new(message)
+      when /violates foreign key constraint/
+        ::ActiveRecord::InvalidForeignKey.new(message)
+      when /value too long/
+        ::ActiveRecord::ValueTooLong.new(message)
+      else
+        super
+      end
+    end
+
+    # @private `Utils.extract_schema_and_table` from AR
+    def extract_schema_and_table(name)
+      result = name.scan(/[^".\s]+|"[^"]*"/)[0, 2]
+      result.each { |m| m.gsub!(/(^"|"$)/, '') }
+      result.unshift(nil) if result.size == 1 # schema == nil
+      result # [schema, table]
+    end
+
+    def extract_pg_identifier_from_name(name)
+      match_data = name[0, 1] == '"' ? name.match(/\"([^\"]+)\"/) : name.match(/([^\.]+)/)
+
+      if match_data
+        rest = name[match_data[0].length..-1]
+        rest = rest[1..-1] if rest[0, 1] == "."
+        [match_data[1], (rest.length > 0 ? rest : nil)]
+      end
+    end
+
+    def extract_table_ref_from_insert_sql(sql)
+      sql[/into\s("[A-Za-z0-9_."\[\]\s]+"|[A-Za-z0-9_."\[\]]+)\s*/im]
+      $1.strip if $1
+    end
+
+    def local_tz
+      @local_tz ||= execute('SHOW TIME ZONE', 'SCHEMA').first["TimeZone"]
+    end
+
+  end
+end
+
+require 'arjdbc/util/quoted_cache'
+
+module ActiveRecord::ConnectionAdapters
+
+  # NOTE: seems needed on 4.x due loading of '.../postgresql/oid' which
+  # assumes: class PostgreSQLAdapter < AbstractAdapter
+  remove_const(:PostgreSQLAdapter) if const_defined?(:PostgreSQLAdapter)
+
+  class PostgreSQLAdapter < AbstractAdapter
+
+    # Try to use as much of the built in postgres logic as possible
+    # maybe someday we can extend the actual adapter
+    include ActiveRecord::ConnectionAdapters::PostgreSQL::ColumnDumper
+    include ActiveRecord::ConnectionAdapters::PostgreSQL::ReferentialIntegrity
+    include ActiveRecord::ConnectionAdapters::PostgreSQL::SchemaStatements
+    include ActiveRecord::ConnectionAdapters::PostgreSQL::Quoting
+
+    include Jdbc::ConnectionPoolCallbacks
+
+    include ArJdbc::Abstract::Core
+    include ArJdbc::Abstract::ConnectionManagement
+    include ArJdbc::Abstract::DatabaseStatements
+    include ArJdbc::Abstract::StatementCache
+    include ArJdbc::Abstract::TransactionSupport
+    include ArJdbc::PostgreSQL
+
+    require 'arjdbc/postgresql/oid_types'
+    include ::ArJdbc::PostgreSQL::OIDTypes
+
+    include ::ArJdbc::PostgreSQL::ColumnHelpers
+
+    include ::ArJdbc::Util::QuotedCache
+
+    # AR expects OID to be available on the adapter
+    OID = ActiveRecord::ConnectionAdapters::PostgreSQL::OID
+
+    def initialize(connection, logger = nil, connection_parameters = nil, config = {})
+      # @local_tz is initialized as nil to avoid warnings when connect tries to use it
+      @local_tz = nil
+
+      super(connection, logger, config) # configure_connection happens in super
+
+      initialize_type_map(@type_map = Type::HashLookupTypeMap.new)
+
+      @use_insert_returning = @config.key?(:insert_returning) ?
+        self.class.type_cast_config_to_boolean(@config[:insert_returning]) : nil
+    end
+
+    def arel_visitor # :nodoc:
+      Arel::Visitors::PostgreSQL.new(self)
+    end
+
+    require 'active_record/connection_adapters/postgresql/schema_definitions'
+
+    ColumnDefinition = ActiveRecord::ConnectionAdapters::PostgreSQL::ColumnDefinition
+    ColumnMethods = ActiveRecord::ConnectionAdapters::PostgreSQL::ColumnMethods
+    TableDefinition = ActiveRecord::ConnectionAdapters::PostgreSQL::TableDefinition
+    Table = ActiveRecord::ConnectionAdapters::PostgreSQL::Table
+
+    def create_table_definition(*args) # :nodoc:
+      TableDefinition.new(*args)
+    end
+
+    public :sql_for_insert
+
+    def schema_creation # :nodoc:
+      PostgreSQL::SchemaCreation.new self
+    end
+
+    def update_table_definition(table_name, base)
+      Table.new(table_name, base)
+    end
+
+    def jdbc_connection_class(spec)
+      ::ArJdbc::PostgreSQL.jdbc_connection_class
+    end
+
+    private
+
+    # Prepared statements aren't schema aware so we need to make sure we
+    # store different PreparedStatement objects for different schemas
+    def sql_key(sql)
+      "#{schema_search_path}-#{sql}"
+    end
+
+  end
+end