Welcome to itachi’s documentation!¶
itachi¶
itachi brings useful functions from modern database management systems to Apache Spark :)
For example, you can import the Postgres extensions and write Spark code that looks just like Postgres.
The functions are implemented as native Spark functions, so they’re performant.
In general, only those functions that difficult for the Apache Spark Community to maintain in the master branch will be added to this library.
Installation¶
Fetch the JAR file from Maven.
libraryDependencies += “com.github.yaooqinn” %% “itachi” % “0.1.0”
Here’s the Maven link where the JAR files are stored.
itachi requires Spark 3+.
Simple function registration¶
Access the Postgres / Teradata functions with these commands::
org.apache.itachi.registerPostgresFunctions
org.apache.itachi.registerTeradataFunctions
Simple example¶
Suppose you have the following data table and would like to join the two arrays, with the familiar array_cat function from Postgres.:
+------+------+
| arr1| arr2|
+------+------+
|[1, 2]| []|
|[1, 2]|[1, 3]|
+------+------+
Concatenate the two arrays::
spark
.sql("select array_cat(arr1, arr2) as both_arrays from some_data")
.show()
+------------+
| both_arrays|
+------------+
| [1, 2]|
|[1, 2, 1, 3]|
+------------+
itachi lets you write Spark SQL code that looks just like Postgres SQL!
Spark SQL extensions installation¶
Config your spark applications with spark.sql.extensions, e.g. spark.sql.extensions=org.apache.spark.sql.extra.PostgreSQLExtensions
org.apache.spark.sql.extra.PostgreSQLExtensions
org.apache.spark.sql.extra.TeradataExtensions
Databricks Installation¶
Create an init script in DBFS::
dbutils.fs.mkdirs("dbfs:/databricks/scripts/")
dbutils.fs.put("/databricks/scripts/itachi-install.sh","""
#!/bin/bash
wget --quiet -O /mnt/driver-daemon/jars/itachi_2.12-0.1.0.jar https://repo1.maven.org/maven2/com/github/yaooqinn/itachi_2.12/0.1.0/itachi_2.12-0.1.0.jar""", true)
Before starting the cluster, set the Spark Config::
spark.sql.extensions org.apache.spark.sql.extra.PostgreSQLExtensions
Also set the DBFS file path before starting the cluster::
dbfs:/databricks/scripts/itachi-install.sh
You can now attach a notebook to the cluster using Postgres SQL syntax.
Spark SQL Compliance¶
This is a Spark SQL extension supplying add-on or aliased functions to the Apache Spark SQL builtin standard functions.
The functions in this library take precedence over the native Spark functions in the even of a name conflict.
Contributing¶
More popular modern dbms system function can be added with your help