Skip to content

TylerHillery/harlequin-adbc

Repository files navigation

harlequin-adbc

This repo provides the Harlequin adapter for ADBC (arrow database connectivity).

Warning

ADBC is a very new database connectivity method compared to ODBC, JDBC that aims at providing a more efficient way to transfer columnar data over the wire.

Due to the recency of its development, there is various level of functionality across drivers which may cause issues.

Installation

harlequin-adbc depends on adbc_driver_manager, harlequin and pyarrow, so installing this package with also install these dependencies.

I also recommend installing the extra adbc drivers for the databases you plan on connecting to otherwise you will have to download the driver yourself from another place and provide the driver_path as a cli argument. The driver path connection method hasn't been thoroughly test.

If you don't install the driver but provide a --drive-type cli argument you will get an ImportError

Using pip

To install the adapter using pip into an activated virtual environment:

pip install harlequin-adbc

with one of the driver extras:

pip install harlequin-adbc[snowflake]

Using poetry

poetry add harlequin-adbc

with one of the driver extras:

poetry add harlequin-adbc[snowflake]

Using pipx

If you do not already have Harlequin installed:

pipx install harlequin-adbc

pipx install harlequin-adbc[snowflake]

If you would like to add ADBC adapter to an existing Harlequin installation:

pipx inject harlequin harlequin-adbc

Make sure to also inject the driver package you want as well. The following options are:

  • adbc-driver-flightsql
  • adbc-driver-postgresql
  • adbc-driver-snowflake
  • adbc-driver-sqlite

Example usage:

pipx inject harlequin adbc-driver-snowflake

Usage and Configuration

You can open Harlequin with the ADBC adapter by selecting it with the -a option and passing in a connection string. The format of the connection string will depend on the driver you are using. You will also need to provide either the --driver-type or the --driver-path.

Driver Type (preferred)

When you use the --driver-type option it will try to dynamically use the driver package of the type selected adbc-driver-{driver type}. That is why it's crucial to also have that package also install in the virtual environment where Harlequin is installed.

  • --driver-type with one of the following options
    • flightsql, postgresql, snowflake, sqlite, duckdb

Driver Path

The other option is to pass the file path location of the adbc driver that you are using. Note this method is not well tested.

  • --driver-path

DB Kwargs String (Optional)

Since the drivers implement so many different options to pass through when you connect to the database this is a way to pass through these options. The format of the string is key=value separated by ;

  • --db-kwargs-str

Example: --db-kwargs-str "username=flight_username;password=flight_password;adbc.flight.sql.client_option.tls_skip_verify=true"

This will parse the string and pass through these values into the db_kwargs value in the dbapi.connect() method. Note the parser relies on the ; and = so if any of of the parameters in the string contain either of these characters it's not going to work. This is a known limitation and something being worked on.

Snowflake Driver

The Snowflake URI should be of one of the following formats:

  • user[:password]@account/database/schema[?param1=value1&paramN=valueN]
  • user[:password]@account/database[?param1=value1&paramN=valueN]
  • user[:password]@host:port/database/schema?account=user_account[&param1=value1&paramN=valueN]
  • host:port/database/schema?account=user_account[&param1=value1&paramN=valueN]

Check the Snowflake ADBC Driver Docs for more details.

Example usage:

harlequin -a adbc "$SNOWFLAKE_URI" --driver-type snowflake

FlightSQL Driver

Example usage:

harlequin -P None -a adbc "grpc+tls://localhost:31337" --driver-type flightsql --db-kwargs-str "username=flight_username;password=flight_password;adbc.flight.sql.client_option.tls_skip_verify=true"

Check the FlightSQL ADBC Driver Docs for more details.

Postgres Driver

The Postgres URI should be in the format of a Postgres DSN:

harlequin -a postgres "postgres://my-user:my-pass@localhost:5432/my-database"

DuckDB and SQLite

While DuckDB and SQLite both have ADBC drivers it's not recommend to use them. Harlequin natively supports both of the databases without having to install any other dependencies.

Know Issues

  • Snowflake adbc driver seems to be the only driver that returns the xdbc_data_type from adbc_get_objects()
  • Snowflake adbc driver has a bug with adbc_get_table_schema() that returns adbc_driver_manager.OperationalError: IO: sql: expected 12 destination arguments in Scan, not 11
  • The PostgreSQL adbc driver is overall buggy and when executing queries you might get the error IO: [libpq] Fetch header failed: no COPY in progress
  • SQLite and DuckDB don't have the same level of depth for adbc_get_objects() compared to other dbs which causes weird issues.
  • DuckDB driver uses different names for the adbc_get_objects() which causes things to break.