Skip to main content

This plugin allows vdk-core to interface with and execute queries against an Impala database.

Features

  • It provides a powerful recovery mechanism handling a lot of challenges - like eventual consistency issues in Impala and more. In one production deployment of VDK it was able to improve SLA of Impala from 95% (queries directly to Impala) to 99% (queries using VDK to Impala).
  • It automatically classifies error based on who is best responsible to handle them - user (job owner) vs platform (infra owner).
  • It provides a default implementation of Kimball templates for generating SCD1, SCD2 dimension tables, Periodic Snapshot Fact Table
  • It can collect lineage data, assuming a lineage logger has been provided through the vdk-core configuration (see vdk config-help for more info)

Usage

Run

pip install vdk-impala

After this, data jobs will have access to a Impala database connection, managed by Versatile Data Kit SDK.

If it is the only database plugin installed , vdk would automatically use it. Otherwise, users need to set VDK_DB_DEFAULT_TYPE=IMPALA as an environment variable or set 'db_default_type' option in the data job config file (config.ini).

For example

    def run(job_input: IJobInput):
        job_input.execute_query("select 'Hi Impala!'")

Lineage

The package can gather lineage data for all successful Impala SQL queries that have actually read or written data. Other plugins can read and optionally send the lineage data to separate system. They need to provide ILineageLogger implementation and hook this way:

    @hookimpl
    def vdk_initialize(context: CoreContext) -> None:
        context.state.set(StoreKey[ILineageLogger]("impala-lineage-logger"), MyLogger())

Lineage is calculated based on the executed query profile. It is retrieved via the cursor by executing additional RPC request against the same Impala node that has coordinated the query right after the original query has successfully finished. See https://impala.apache.org/docs/build/html/topics/impala_logging.html for more information how profiles are stored and here https://impala.apache.org/docs/build/impala-3.1.pdf for more information about the profiles themselves.

If enabled, query plan is retrieved for every successfully executed query against Impala excluding keepalive queries like "Select 1".

Kimbal dimensional modeling templates

See the following tutorial for more details. It is based on Trino but the process is equivalent for Impala (only the database configuration requires change).

Configuration

Run vdk config-help - search for those prefixed with "IMPALA_" to see what configuration options are available.

Disclaimer

This plugin is tested against a specific impala version. The version comes from the docker-compose.yaml container's impala version. For more information on the imapala version tested against please google the docker image.

Testing

Testing this plugin locally requires installing the dependencies listed in vdk-plugins/vdk-impala/requirements.txt

Run

pip install -r requirements.txt

Metadata

Release files for vdk-impala 0.4.659212801

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for vdk-impala 0.4.659212801
File Size Uploaded
vdk-impala-0.4.659212801.tar.gz 22.2 kB Details

Release files / vdk-impala-0.4.659212801.tar.gz

Download URL vdk-impala-0.4.659212801.tar.gz
Size 22.2 kB
Tags Source
SHA-256 checksum
How to use checksums
95cdb4460bfbe65a0bdf613fc12f84dd90b0036421d8b56857b5c9f4ddb03a5e
BLAKE2b-256 checksum
How to use checksums
c6ce8331c9ab56ac8b6af29293fda4777f76016ff008527db1d00fc418295c40
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/4.0.1 CPython/3.7.13

Release history Release notifications | RSS feed

This release

0.4.659212801 This release

1 release file

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page