module Sequel::JDBC::Postgres::DatabaseMethods

def self.extended(db)
  super
  db.send(:initialize_postgres_adapter)
end

def add_conversion_proc(oid, *)
  super
  Sequel.synchronize{@oid_convertor_map.delete(oid)}
end

def bound_variable_arg(arg, conn)
  nil
end

def copy_into(table, opts=OPTS)
  data = opts[:data]
  data = Array(data) if data.is_a?(String)
  if defined?(yield) && data
    raise Error, "Cannot provide both a :data option and a block to copy_into"
  elsif !defined?(yield) && !data
    raise Error, "Must provide either a :data option or a block to copy_into"
  end
  synchronize(opts[:server]) do |conn|
    begin
      copy_manager = Java::OrgPostgresqlCopy::CopyManager.new(conn)
      copier = copy_manager.copy_in(copy_into_sql(table, opts))
      if defined?(yield)
        while buf = yield
          java_bytes = buf.to_java_bytes
          copier.writeToCopy(java_bytes, 0, java_bytes.length)
        end
      else
        data.each do |d|
          java_bytes = d.to_java_bytes
          copier.writeToCopy(java_bytes, 0, java_bytes.length)
        end
      end
    rescue Exception => e
      copier.cancelCopy if copier
      raise
    ensure
      unless e
        begin
          copier.endCopy
        rescue NativeException => e2
          raise_error(e2)
        end
      end
    end
  end
end

def copy_table(table, opts=OPTS)
  synchronize(opts[:server]) do |conn|
    copy_manager = Java::OrgPostgresqlCopy::CopyManager.new(conn)
    copier = copy_manager.copy_out(copy_table_sql(table, opts))
    begin
      if defined?(yield)
        while buf = copier.readFromCopy
          yield(String.from_java_bytes(buf))
        end
        nil
      else
        b = String.new
        while buf = copier.readFromCopy
          b << String.from_java_bytes(buf)
        end
        b
      end
    rescue => e
      raise_error(e, :disconnect=>true)
    ensure
      if buf && !e
        raise DatabaseDisconnectError, "disconnecting as a partial COPY may leave the connection in an unusable state"
      end
    end
  end
end

def disconnect_error?(exception, opts)
  super || exception.message =~ /\A(This connection has been closed\.|FATAL: terminating connection due to administrator command|An I\/O error occurred while sending to the backend\.)\z/
end

def oid_convertor_proc(oid)
  if (conv = Sequel.synchronize{@oid_convertor_map[oid]}).nil?
    conv = if pr = conversion_procs[oid]
      lambda do |r, i|
        if v = r.getString(i)
          pr.call(v)
        end
      end
    else
      false
    end
    Sequel.synchronize{@oid_convertor_map[oid] = conv}
  end
  conv
end

def prepare_jdbc_statement(conn, sql, opts)
  ps = super
  unless opts[:name]
    ps.prepare_threshold = 0
  end
  ps
end

def set_ps_arg(cps, arg, i)
  if v = bound_variable_arg(arg, nil)
    obj = Java::OrgPostgresqlUtil::PGobject.new
    obj.setType("unknown")
    obj.setValue(v)
    cps.setObject(i, obj)
  else
    super
  end
end

def set_ps_arg_nil(cps, i)
  cps.setNull(i, JavaSQL::Types::NULL)
end

def setup_connection_with_opts(conn, opts)
  conn = super
  statement(conn) do |stmt|
    connection_configuration_sqls(opts).each{|sql| log_connection_yield(sql, conn){stmt.execute(sql)}}
  end
  conn
end

def setup_type_convertor_map
  super
  @oid_convertor_map = {}
end