Kensuke-Hinata / arrow-datafusion

Apache Arrow DataFusion and Ballista query engines

DataFusion

DataFusion is an extensible query execution framework, written in Rust, that uses Apache Arrow as its in-memory format.

DataFusion supports both an SQL and a DataFrame API for building logical query plans as well as a query optimizer and execution engine capable of parallel execution against partitioned data sources (CSV and Parquet) using threads.

DataFusion also supports distributed query execution via the Ballista crate.

Use Cases

DataFusion is used to create modern, fast and efficient data pipelines, ETL processes, and database systems, which need the performance of Rust and Apache Arrow and want to provide their users the convenience of an SQL interface or a DataFrame API.

Why DataFusion?

• High Performance: Leveraging Rust and Arrow's memory model, DataFusion achieves very high performance
• Easy to Connect: Being part of the Apache Arrow ecosystem (Arrow, Parquet and Flight), DataFusion works well with the rest of the big data ecosystem
• Easy to Embed: Allowing extension at almost any point in its design, DataFusion can be tailored for your specific usecase
• High Quality: Extensively tested, both by itself and with the rest of the Arrow ecosystem, DataFusion can be used as the foundation for production systems.

Known Uses

Projects that adapt to or serve as plugins to DataFusion:

• datafusion-python
• datafusion-java
• datafusion-ruby
• datafusion-objectstore-s3
• datafusion-hdfs-native

Here are some of the projects known to use DataFusion:

• Ballista Distributed Compute Platform
• Cloudfuse Buzz
• Cube Store
• delta-rs
• InfluxDB IOx Time Series Database
• ROAPI
• Tensorbase
• Squirtle
• VegaFusion Server-side acceleration for the Vega visualization grammar

(if you know of another project, please submit a PR to add a link!)

Example Usage

Run a SQL query against data stored in a CSV:

use datafusion::prelude::*;
use datafusion::arrow::util::pretty::print_batches;
use datafusion::arrow::record_batch::RecordBatch;

#[tokio::main]
async fn main() -> datafusion::error::Result<()> {
  // register the table
  let mut ctx = ExecutionContext::new();
  ctx.register_csv("example", "tests/example.csv", CsvReadOptions::new()).await?;

  // create a plan to run a SQL query
  let df = ctx.sql("SELECT a, MIN(b) FROM example GROUP BY a LIMIT 100").await?;

  // execute and print results
  df.show().await?;
  Ok(())
}

Use the DataFrame API to process data stored in a CSV:

use datafusion::prelude::*;
use datafusion::arrow::util::pretty::print_batches;
use datafusion::arrow::record_batch::RecordBatch;

#[tokio::main]
async fn main() -> datafusion::error::Result<()> {
  // create the dataframe
  let mut ctx = ExecutionContext::new();
  let df = ctx.read_csv("tests/example.csv", CsvReadOptions::new()).await?;

  let df = df.filter(col("a").lt_eq(col("b")))?
          .aggregate(vec![col("a")], vec![min(col("b"))])?;

  // execute and print results
  df.show_limit(100).await?;
  Ok(())
}

Both of these examples will produce

+---+--------+
| a | MIN(b) |
+---+--------+
| 1 | 2      |
+---+--------+

Using DataFusion as a library

DataFusion is published on crates.io, and is well documented on docs.rs.

To get started, add the following to your Cargo.toml file:

[dependencies]
datafusion = "6.0.0"

Using DataFusion as a binary

DataFusion also includes a simple command-line interactive SQL utility. See the CLI reference for more information.

Roadmap

A quarterly roadmap will be published to give the DataFusion community visibility into the priorities of the projects contributors. This roadmap is not binding.

2022 Q1

DataFusion Core

• Publish official Arrow2 branch
• Implementation of memory manager (i.e. to enable spilling to disk as needed)

Benchmarking

• Inclusion in Db-Benchmark with all quries covered
• All TPCH queries covered

Performance Improvements

• Predicate evaluation
• Improve multi-column comparisons (that can't be vectorized at the moment)
• Null constant support

New Features

• Read JSON as table
• Simplify DDL with Datafusion-Cli
• Add Decimal128 data type and the attendant features such as Arrow Kernel and UDF support
• Add new experimental e-graph based optimizer

Ballista

• Begin work on design documents and plan / priorities for development

Extensions (datafusion-contrib)

• Stable S3 support
• Begin design discussions and prototyping of a stream provider

Beyond 2022 Q1

There is no clear timeline for the below, but community members have expressed interest in working on these topics.

DataFusion Core

• Custom SQL support
• Split DataFusion into multiple crates
• Push based query execution and code generation

Ballista

• Evolve architecture so that it can be deployed in a multi-tenant cloud native environment
• Ensure Ballista is scalable, elastic, and stable for production usage
• Develop distributed ML capabilities

Status

General

• SQL Parser
• SQL Query Planner
• Query Optimizer
• Constant folding
• Join Reordering
• Limit Pushdown
• Projection push down
• Predicate push down
• Type coercion
• Parallel query execution

SQL Support

• Projection
• Filter (WHERE)
• Filter post-aggregate (HAVING)
• Limit
• Aggregate
• Common math functions
• cast
• try_cast
• VALUES lists
• Postgres compatible String functions
  • ascii
  • bit_length
  • btrim
  • char_length
  • character_length
  • chr
  • concat
  • concat_ws
  • initcap
  • left
  • length
  • lpad
  • ltrim
  • octet_length
  • regexp_replace
  • repeat
  • replace
  • reverse
  • right
  • rpad
  • rtrim
  • split_part
  • starts_with
  • strpos
  • substr
  • to_hex
  • translate
  • trim
• Miscellaneous/Boolean functions
  • nullif
• Approximation functions
  • approx_distinct
• Common date/time functions
  • Basic date functions
  • Basic time functions
  • Basic timestamp functions
    • to_timestamp
    • to_timestamp_millis
    • to_timestamp_micros
    • to_timestamp_seconds
• nested functions
  • Array of columns
• Schema Queries
  • SHOW TABLES
  • SHOW COLUMNS
  • information_schema.{tables, columns}
  • information_schema other views
• Sorting
• Nested types
• Lists
• Subqueries
• Common table expressions
• Set Operations
  • UNION ALL
  • UNION
  • INTERSECT
  • INTERSECT ALL
  • EXCEPT
  • EXCEPT ALL
• Joins
  • INNER JOIN
  • LEFT JOIN
  • RIGHT JOIN
  • FULL JOIN
  • CROSS JOIN
• Window
  • Empty window
  • Common window functions
  • Window with PARTITION BY clause
  • Window with ORDER BY clause
  • Window with FILTER clause
  • Window with custom WINDOW FRAME
  • UDF and UDAF for window functions

Data Sources

• CSV
• Parquet primitive types
• Parquet nested types

Extensibility

DataFusion is designed to be extensible at all points. To that end, you can provide your own custom:

• User Defined Functions (UDFs)
• User Defined Aggregate Functions (UDAFs)
• User Defined Table Source (TableProvider) for tables
• User Defined Optimizer passes (plan rewrites)
• User Defined LogicalPlan nodes
• User Defined ExecutionPlan nodes

Rust Version Compatbility

This crate is tested with the latest stable version of Rust. We do not currently test against other, older versions of the Rust compiler.

Supported SQL

This library currently supports many SQL constructs, including

• CREATE EXTERNAL TABLE X STORED AS PARQUET LOCATION '...'; to register a table's locations
• SELECT ... FROM ... together with any expression
• ALIAS to name an expression
• CAST to change types, including e.g. Timestamp(Nanosecond, None)
• Many mathematical unary and binary expressions such as +, /, sqrt, tan, >=.
• WHERE to filter
• GROUP BY together with one of the following aggregations: MIN, MAX, COUNT, SUM, AVG, CORR, VAR, COVAR, STDDEV (sample and population)
• ORDER BY together with an expression and optional ASC or DESC and also optional NULLS FIRST or NULLS LAST

Supported Functions

DataFusion strives to implement a subset of the PostgreSQL SQL dialect where possible. We explicitly choose a single dialect to maximize interoperability with other tools and allow reuse of the PostgreSQL documents and tutorials as much as possible.

Currently, only a subset of the PostgreSQL dialect is implemented, and we will document any deviations.

Schema Metadata / Information Schema Support

DataFusion supports the showing metadata about the tables available. This information can be accessed using the views of the ISO SQL information_schema schema or the DataFusion specific SHOW TABLES and SHOW COLUMNS commands.

More information can be found in the Postgres docs).

To show tables available for use in DataFusion, use the SHOW TABLES command or the information_schema.tables view:

> show tables;
+---------------+--------------------+------------+------------+
| table_catalog | table_schema       | table_name | table_type |
+---------------+--------------------+------------+------------+
| datafusion    | public             | t          | BASE TABLE |
| datafusion    | information_schema | tables     | VIEW       |
+---------------+--------------------+------------+------------+

> select * from information_schema.tables;

+---------------+--------------------+------------+--------------+
| table_catalog | table_schema       | table_name | table_type   |
+---------------+--------------------+------------+--------------+
| datafusion    | public             | t          | BASE TABLE   |
| datafusion    | information_schema | TABLES     | SYSTEM TABLE |
+---------------+--------------------+------------+--------------+

To show the schema of a table in DataFusion, use the SHOW COLUMNS command or the or information_schema.columns view:

> show columns from t;
+---------------+--------------+------------+-------------+-----------+-------------+
| table_catalog | table_schema | table_name | column_name | data_type | is_nullable |
+---------------+--------------+------------+-------------+-----------+-------------+
| datafusion    | public       | t          | a           | Int32     | NO          |
| datafusion    | public       | t          | b           | Utf8      | NO          |
| datafusion    | public       | t          | c           | Float32   | NO          |
+---------------+--------------+------------+-------------+-----------+-------------+

>   select table_name, column_name, ordinal_position, is_nullable, data_type from information_schema.columns;
+------------+-------------+------------------+-------------+-----------+
| table_name | column_name | ordinal_position | is_nullable | data_type |
+------------+-------------+------------------+-------------+-----------+
| t          | a           | 0                | NO          | Int32     |
| t          | b           | 1                | NO          | Utf8      |
| t          | c           | 2                | NO          | Float32   |
+------------+-------------+------------------+-------------+-----------+

Supported Data Types

DataFusion uses Arrow, and thus the Arrow type system, for query execution. The SQL types from sqlparser-rs are mapped to Arrow types according to the following table

SQL Data Type	Arrow DataType
CHAR	Utf8
VARCHAR	Utf8
UUID	Not yet supported
CLOB	Not yet supported
BINARY	Not yet supported
VARBINARY	Not yet supported
DECIMAL	Float64
FLOAT	Float32
SMALLINT	Int16
INT	Int32
BIGINT	Int64
REAL	Float32
DOUBLE	Float64
BOOLEAN	Boolean
DATE	Date32
TIME	Time64(TimeUnit::Millisecond)
TIMESTAMP	Timestamp(TimeUnit::Nanosecond)
INTERVAL	Not yet supported
REGCLASS	Not yet supported
TEXT	Not yet supported
BYTEA	Not yet supported
CUSTOM	Not yet supported
ARRAY	Not yet supported

Roadmap

Please see Roadmap for information of where the project is headed.

Architecture Overview

There is no formal document describing DataFusion's architecture yet, but the following presentations offer a good overview of its different components and how they interact together.

• (March 2021): The DataFusion architecture is described in Query Engine Design and the Rust-Based DataFusion in Apache Arrow: recording (DataFusion content starts ~ 15 minutes in) and slides
• (February 2021): How DataFusion is used within the Ballista Project is described in *Ballista: Distributed Compute with Rust and Apache Arrow: recording

Developer's guide

Please see Developers Guide for information about developing DataFusion.