class Sequel::ODBC::Database

def adapter_initialize
  if (db_type = @opts[:db_type]) && (prok = Sequel::Database.load_adapter(db_type.to_sym, :map=>DATABASE_SETUP, :subdir=>'odbc'))
    prok.call(self)
  end
end

def connect(server)
  opts = server_opts(server)
  conn = if opts.include?(:drvconnect)
    ::ODBC::Database.new.drvconnect(opts[:drvconnect])
  elsif opts.include?(:driver)
    drv = ::ODBC::Driver.new
    drv.name = 'Sequel ODBC Driver130'
    opts.each do |param, value|
      if :driver == param && value !~ /\A\{.+\}\z/
        value = "{#{value}}"
      end
      drv.attrs[param.to_s.upcase] = value.to_s
    end
    ::ODBC::Database.new.drvconnect(drv)
  else
    ::ODBC::connect(opts[:database], opts[:user], opts[:password])
  end
  conn.autocommit = true
  conn
end      

def connection_execute_method
  :do
end

def database_error_classes
  [::ODBC::Error]
end

def dataset_class_default
  Dataset
end

def disconnect_connection(c)
  c.disconnect
end

def disconnect_error?(e, opts)
  super || (e.is_a?(::ODBC::Error) && /\A08S01/.match(e.message))
end

def execute(sql, opts=OPTS)
  synchronize(opts[:server]) do |conn|
    begin
      r = log_connection_yield(sql, conn){conn.run(sql)}
      yield(r) if defined?(yield)
    rescue ::ODBC::Error, ArgumentError => e
      raise_error(e)
    ensure
      r.drop if r
    end
    nil
  end
end

def execute_dui(sql, opts=OPTS)
  synchronize(opts[:server]) do |conn|
    begin
      log_connection_yield(sql, conn){conn.do(sql)}
    rescue ::ODBC::Error, ArgumentError => e
      raise_error(e)
    end
  end
end