Use Amazon Athena with Spark SQL to your open-source transactional desk codecs

AWS-powered information lakes, supported by the unrivaled availability of Amazon Easy Storage Service (Amazon S3), can deal with the dimensions, agility, and suppleness required to mix totally different information and analytics approaches. As information lakes have grown in dimension and matured in utilization, a big quantity of effort could be spent retaining the info in line with enterprise occasions. To make sure information are up to date in a transactionally constant method, a rising variety of prospects are utilizing open-source transactional desk codecs resembling Apache Iceberg, Apache Hudi, and Linux Basis Delta Lake that show you how to retailer information with excessive compression charges, natively interface together with your purposes and frameworks, and simplify incremental information processing in information lakes constructed on Amazon S3. These codecs allow ACID (atomicity, consistency, isolation, sturdiness) transactions, upserts, and deletes, and superior options resembling time journey and snapshots that have been beforehand solely out there in information warehouses. Every storage format implements this performance in barely other ways; for a comparability, seek advice from Selecting an open desk format to your transactional information lake on AWS.

In 2023, AWS introduced basic availability for Apache Iceberg, Apache Hudi, and Linux Basis Delta Lake in Amazon Athena for Apache Spark, which removes the necessity to set up a separate connector or related dependencies and handle variations, and simplifies the configuration steps required to make use of these frameworks.

On this submit, we present you methods to use Spark SQL in Amazon Athena notebooks and work with Iceberg, Hudi, and Delta Lake desk codecs. We show widespread operations resembling creating databases and tables, inserting information into the tables, querying information, and taking a look at snapshots of the tables in Amazon S3 utilizing Spark SQL in Athena.

Conditions

Full the next stipulations:

Obtain and import instance notebooks from Amazon S3

To observe alongside, obtain the notebooks mentioned on this submit from the next areas:

After you obtain the notebooks, import them into your Athena Spark surroundings by following the To import a pocket book part in Managing pocket book information.

Navigate to particular Open Desk Format part

In case you are desirous about Iceberg desk format, navigate to Working with Apache Iceberg tables part.

In case you are desirous about Hudi desk format, navigate to Working with Apache Hudi tables part.

In case you are desirous about Delta Lake desk format, navigate to Working with Linux basis Delta Lake tables part.

Working with Apache Iceberg tables

When utilizing Spark notebooks in Athena, you’ll be able to run SQL queries immediately with out having to make use of PySpark. We do that through the use of cell magics, that are particular headers in a pocket book cell that change the cell’s habits. For SQL, we are able to add the %%sql magic, which is able to interpret the complete cell contents as a SQL assertion to be run on Athena.

On this part, we present how you should utilize SQL on Apache Spark for Athena to create, analyze, and handle Apache Iceberg tables.

Arrange a pocket book session

With a purpose to use Apache Iceberg in Athena, whereas creating or enhancing a session, choose the Apache Iceberg possibility by increasing the Apache Spark properties part. It can pre-populate the properties as proven within the following screenshot.

For steps, see Modifying session particulars or Creating your personal pocket book.

The code used on this part is out there within the SparkSQL_iceberg.ipynb file to observe alongside.

Create a database and Iceberg desk

First, we create a database within the AWS Glue Knowledge Catalog. With the next SQL, we are able to create a database referred to as icebergdb:

%%sql
CREATE DATABASE icebergdb

Subsequent, within the database icebergdb, we create an Iceberg desk referred to as noaa_iceberg pointing to a location in Amazon S3 the place we are going to load the info. Run the next assertion and change the situation s3://<your-S3-bucket>/<prefix>/ together with your S3 bucket and prefix:

%%sql
CREATE TABLE icebergdb.noaa_iceberg(
station string,
date string,
latitude string,
longitude string,
elevation string,
title string,
temp string,
temp_attributes string,
dewp string,
dewp_attributes string,
slp string,
slp_attributes string,
stp string,
stp_attributes string,
visib string,
visib_attributes string,
wdsp string,
wdsp_attributes string,
mxspd string,
gust string,
max string,
max_attributes string,
min string,
min_attributes string,
prcp string,
prcp_attributes string,
sndp string,
frshtt string)
USING iceberg
PARTITIONED BY (yr string)
LOCATION 's3://<your-S3-bucket>/<prefix>/noaaiceberg/'

Insert information into the desk

To populate the noaa_iceberg Iceberg desk, we insert information from the Parquet desk sparkblogdb.noaa_pq that was created as a part of the stipulations. You are able to do this utilizing an INSERT INTO assertion in Spark:

%%sql
INSERT INTO icebergdb.noaa_iceberg choose * from sparkblogdb.noaa_pq

Alternatively, you should utilize CREATE TABLE AS SELECT with the USING iceberg clause to create an Iceberg desk and insert information from a supply desk in a single step:

%%sql
CREATE TABLE icebergdb.noaa_iceberg
USING iceberg
PARTITIONED BY (yr)
AS SELECT * FROM sparkblogdb.noaa_pq

Question the Iceberg desk

Now that the info is inserted within the Iceberg desk, we are able to begin analyzing it. Let’s run a Spark SQL to search out the minimal recorded temperature by yr for the 'SEATTLE TACOMA AIRPORT, WA US' location:

%%sql
choose title, yr, min(MIN) as minimum_temperature
from icebergdb.noaa_iceberg
the place title="SEATTLE TACOMA AIRPORT, WA US"
group by 1,2

We get following output.

Replace information within the Iceberg desk

Let’s take a look at methods to replace information in our desk. We need to replace the station title 'SEATTLE TACOMA AIRPORT, WA US' to 'Sea-Tac'. Utilizing Spark SQL, we are able to run an UPDATE assertion towards the Iceberg desk:

%%sql
UPDATE icebergdb.noaa_iceberg
SET title="Sea-Tac"
WHERE title="SEATTLE TACOMA AIRPORT, WA US"

We are able to then run the earlier SELECT question to search out the minimal recorded temperature for the 'Sea-Tac' location:

%%sql
choose title, yr, min(MIN) as minimum_temperature
from icebergdb.noaa_iceberg
the place title="Sea-Tac"
group by 1,2

We get the next output.

Compact information information

Open desk codecs like Iceberg work by creating delta modifications in file storage, and monitoring the variations of rows by manifest information. Extra information information results in extra metadata saved in manifest information, and small information information usually trigger an pointless quantity of metadata, leading to much less environment friendly queries and better Amazon S3 entry prices. Working Iceberg’s rewrite_data_files process in Spark for Athena will compact information information, combining many small delta change information right into a smaller set of read-optimized Parquet information. Compacting information quickens the learn operation when queried. To run compaction on our desk, run the next Spark SQL:

%%sql
CALL spark_catalog.system.rewrite_data_files
(desk => 'icebergdb.noaa_iceberg', technique=>'type', sort_order => 'zorder(title)')

rewrite_data_files presents choices to specify your type technique, which will help reorganize and compact information.

Record desk snapshots

Every write, replace, delete, upsert, and compaction operation on an Iceberg desk creates a brand new snapshot of a desk whereas retaining the outdated information and metadata round for snapshot isolation and time journey. To record the snapshots of an Iceberg desk, run the next Spark SQL assertion:

%%sql
SELECT *
FROM spark_catalog.icebergdb.noaa_iceberg.snapshots

Expire outdated snapshots

Frequently expiring snapshots is really useful to delete information information which can be not wanted, and to maintain the scale of desk metadata small. It can by no means take away information which can be nonetheless required by a non-expired snapshot. In Spark for Athena, run the next SQL to run out snapshots for the desk icebergdb.noaa_iceberg which can be older than a selected timestamp:

%%sql
CALL spark_catalog.system.expire_snapshots
('icebergdb.noaa_iceberg', TIMESTAMP '2023-11-30 00:00:00.000')

Be aware that the timestamp worth is specified as a string in format yyyy-MM-dd HH:mm:ss.fff. The output will give a rely of the variety of information and metadata information deleted.

Drop the desk and database

You may run the next Spark SQL to scrub up the Iceberg tables and related information in Amazon S3 from this train:

%%sql
DROP TABLE icebergdb.noaa_iceberg PURGE

Run the next Spark SQL to take away the database icebergdb:

%%sql
DROP DATABASE icebergdb

To be taught extra about all of the operations you’ll be able to carry out on Iceberg tables utilizing Spark for Athena, seek advice from Spark Queries and Spark Procedures within the Iceberg documentation.

Working with Apache Hudi tables

Subsequent, we present how you should utilize SQL on Spark for Athena to create, analyze, and handle Apache Hudi tables.

Arrange a pocket book session

With a purpose to use Apache Hudi in Athena, whereas creating or enhancing a session, choose the Apache Hudi possibility by increasing the Apache Spark properties part.

For steps, see Modifying session particulars or Creating your personal pocket book.

The code used on this part needs to be out there within the SparkSQL_hudi.ipynb file to observe alongside.

Create a database and Hudi desk

First, we create a database referred to as hudidb that shall be saved within the AWS Glue Knowledge Catalog adopted by Hudi desk creation:

%%sql
CREATE DATABASE hudidb

We create a Hudi desk pointing to a location in Amazon S3 the place we are going to load the info. Be aware that the desk is of copy-on-write sort. It’s outlined by sort="cow" within the desk DDL. We’ve got outlined station and date because the a number of main keys and preCombinedField as yr. Additionally, the desk is partitioned on yr. Run the next assertion and change the situation s3://<your-S3-bucket>/<prefix>/ together with your S3 bucket and prefix:

%%sql
CREATE TABLE hudidb.noaa_hudi(
station string,
date string,
latitude string,
longitude string,
elevation string,
title string,
temp string,
temp_attributes string,
dewp string,
dewp_attributes string,
slp string,
slp_attributes string,
stp string,
stp_attributes string,
visib string,
visib_attributes string,
wdsp string,
wdsp_attributes string,
mxspd string,
gust string,
max string,
max_attributes string,
min string,
min_attributes string,
prcp string,
prcp_attributes string,
sndp string,
frshtt string,
yr string)
USING HUDI
PARTITIONED BY (yr)
TBLPROPERTIES(
primaryKey = 'station, date',
preCombineField = 'yr',
sort="cow"
)
LOCATION 's3://<your-S3-bucket>/<prefix>/noaahudi/'

Insert information into the desk

Like with Iceberg, we use the INSERT INTO assertion to populate the desk by studying information from the sparkblogdb.noaa_pq desk created within the earlier submit:

%%sql
INSERT INTO hudidb.noaa_hudi choose * from sparkblogdb.noaa_pq

Question the Hudi desk

Now that the desk is created, let’s run a question to search out the utmost recorded temperature for the 'SEATTLE TACOMA AIRPORT, WA US' location:

%%sql
choose title, yr, max(MAX) as maximum_temperature
from hudidb.noaa_hudi
the place title="SEATTLE TACOMA AIRPORT, WA US"
group by 1,2

Replace information within the Hudi desk

Let’s change the station title 'SEATTLE TACOMA AIRPORT, WA US' to 'Sea–Tac'. We are able to run an UPDATE assertion on Spark for Athena to replace the data of the noaa_hudi desk:

%%sql
UPDATE hudidb.noaa_hudi
SET title="Sea-Tac"
WHERE title="SEATTLE TACOMA AIRPORT, WA US"

We run the earlier SELECT question to search out the utmost recorded temperature for the 'Sea-Tac' location:

%%sql
choose title, yr, max(MAX) as maximum_temperature
from hudidb.noaa_hudi
the place title="Sea-Tac"
group by 1,2

Run time journey queries

We are able to use time journey queries in SQL on Athena to research previous information snapshots. For instance:

%%sql
choose title, yr, max(MAX) as maximum_temperature
from hudidb.noaa_hudi timestamp as of '2023-12-01 23:53:43.100'
the place title="SEATTLE TACOMA AIRPORT, WA US"
group by 1,2

This question checks the Seattle Airport temperature information as of a selected time previously. The timestamp clause lets us journey again with out altering present information. Be aware that the timestamp worth is specified as a string in format yyyy-MM-dd HH:mm:ss.fff.

Optimize question velocity with clustering

To enhance question efficiency, you’ll be able to carry out clustering on Hudi tables utilizing SQL in Spark for Athena:

%%sql
CALL run_clustering(desk => 'hudidb.noaa_hudi', order => 'title')

Compact tables

Compaction is a desk service employed by Hudi particularly in Merge On Learn (MOR) tables to merge updates from row-based log information to the corresponding columnar-based base file periodically to supply a brand new model of the bottom file. Compaction will not be relevant to Copy On Write (COW) tables and solely applies to MOR tables. You may run the next question in Spark for Athena to carry out compaction on MOR tables:

%%sql
CALL run_compaction(op => 'run', desk => 'hudi_table_mor');

Drop the desk and database

Run the next Spark SQL to take away the Hudi desk you created and related information from the Amazon S3 location:

%%sql
DROP TABLE hudidb.noaa_hudi PURGE

Run the next Spark SQL to take away the database hudidb:

%%sql
DROP DATABASE hudidb

To find out about all of the operations you’ll be able to carry out on Hudi tables utilizing Spark for Athena, seek advice from SQL DDL and Procedures within the Hudi documentation.

Working with Linux basis Delta Lake tables

Subsequent, we present how you should utilize SQL on Spark for Athena to create, analyze, and handle Delta Lake tables.

Arrange a pocket book session

With a purpose to use Delta Lake in Spark for Athena, whereas creating or enhancing a session, choose Linux Basis Delta Lake by increasing the Apache Spark properties part.

For steps, see Modifying session particulars or Creating your personal pocket book.

The code used on this part needs to be out there within the SparkSQL_delta.ipynb file to observe alongside.

Create a database and Delta Lake desk

On this part, we create a database within the AWS Glue Knowledge Catalog. Utilizing following SQL, we are able to create a database referred to as deltalakedb:

%%sql
CREATE DATABASE deltalakedb

Subsequent, within the database deltalakedb, we create a Delta Lake desk referred to as noaa_delta pointing to a location in Amazon S3 the place we are going to load the info. Run the next assertion and change the situation s3://<your-S3-bucket>/<prefix>/ together with your S3 bucket and prefix:

%%sql
CREATE TABLE deltalakedb.noaa_delta(
station string,
date string,
latitude string,
longitude string,
elevation string,
title string,
temp string,
temp_attributes string,
dewp string,
dewp_attributes string,
slp string,
slp_attributes string,
stp string,
stp_attributes string,
visib string,
visib_attributes string,
wdsp string,
wdsp_attributes string,
mxspd string,
gust string,
max string,
max_attributes string,
min string,
min_attributes string,
prcp string,
prcp_attributes string,
sndp string,
frshtt string)
USING delta
PARTITIONED BY (yr string)
LOCATION 's3://<your-S3-bucket>/<prefix>/noaadelta/'

Insert information into the desk

We use an INSERT INTO assertion to populate the desk by studying information from the sparkblogdb.noaa_pq desk created within the earlier submit:

%%sql
INSERT INTO deltalakedb.noaa_delta choose * from sparkblogdb.noaa_pq

You can too use CREATE TABLE AS SELECT to create a Delta Lake desk and insert information from a supply desk in a single question.

Question the Delta Lake desk

Now that the info is inserted within the Delta Lake desk, we are able to begin analyzing it. Let’s run a Spark SQL to search out the minimal recorded temperature for the 'SEATTLE TACOMA AIRPORT, WA US' location:

%%sql
choose title, yr, max(MAX) as minimum_temperature
from deltalakedb.noaa_delta
the place title="SEATTLE TACOMA AIRPORT, WA US"
group by 1,2

Replace information within the Delta lake desk

Let’s change the station title 'SEATTLE TACOMA AIRPORT, WA US' to 'Sea–Tac'. We are able to run an UPDATE assertion on Spark for Athena to replace the data of the noaa_delta desk:

%%sql
UPDATE deltalakedb.noaa_delta
SET title="Sea-Tac"
WHERE title="SEATTLE TACOMA AIRPORT, WA US"

We are able to run the earlier SELECT question to search out the minimal recorded temperature for the 'Sea-Tac' location, and the end result needs to be the identical as earlier:

%%sql
choose title, yr, max(MAX) as minimum_temperature
from deltalakedb.noaa_delta
the place title="Sea-Tac"
group by 1,2

Compact information information

In Spark for Athena, you’ll be able to run OPTIMIZE on the Delta Lake desk, which is able to compact the small information into bigger information, so the queries aren’t burdened by the small file overhead. To carry out the compaction operation, run the next question:

%%sql
OPTIMIZE deltalakedb.noaa_delta

Consult with Optimizations within the Delta Lake documentation for various choices out there whereas working OPTIMIZE.

Take away information not referenced by a Delta Lake desk

You may take away information saved in Amazon S3 which can be not referenced by a Delta Lake desk and are older than the retention threshold by working the VACCUM command on the desk utilizing Spark for Athena:

%%sql
VACUUM deltalakedb.noaa_delta

Consult with Take away information not referenced by a Delta desk within the Delta Lake documentation for choices out there with VACUUM.

Drop the desk and database

Run the next Spark SQL to take away the Delta Lake desk you created:

%%sql
DROP TABLE deltalakedb.noaa_delta

Run the next Spark SQL to take away the database deltalakedb:

%%sql
DROP DATABASE deltalakedb

Working DROP TABLE DDL on the Delta Lake desk and database deletes the metadata for these objects, however doesn’t robotically delete the info information in Amazon S3. You may run the next Python code within the pocket book’s cell to delete the info from the S3 location:

import boto3

s3 = boto3.useful resource('s3')
bucket = s3.Bucket('<your-S3-bucket>')
bucket.objects.filter(Prefix="<prefix>/noaadelta/").delete()

To be taught extra concerning the SQL statements that you could run on a Delta Lake desk utilizing Spark for Athena, seek advice from the quickstart within the Delta Lake documentation.

Conclusion

This submit demonstrated methods to use Spark SQL in Athena notebooks to create databases and tables, insert and question information, and carry out widespread operations like updates, compactions, and time journey on Hudi, Delta Lake, and Iceberg tables. Open desk codecs add ACID transactions, upserts, and deletes to information lakes, overcoming limitations of uncooked object storage. By eradicating the necessity to set up separate connectors, Spark on Athena’s built-in integration reduces configuration steps and administration overhead when utilizing these well-liked frameworks for constructing dependable information lakes on Amazon S3. To be taught extra about deciding on an open desk format to your information lake workloads, seek advice from Selecting an open desk format to your transactional information lake on AWS.

In regards to the Authors

Pathik Shah is a Sr. Analytics Architect on Amazon Athena. He joined AWS in 2015 and has been focusing within the massive information analytics house since then, serving to prospects construct scalable and strong options utilizing AWS analytics providers.

Raj Devnath is a Product Supervisor at AWS on Amazon Athena. He’s captivated with constructing merchandise prospects love and serving to prospects extract worth from their information. His background is in delivering options for a number of finish markets, resembling finance, retail, good buildings, dwelling automation, and information communication methods.