ETL pipeline for a data lake hosted on S3. It load data from S3, process the data into analytics tables using Spark running on AWS EMR cluster
and load them back into S3.
A music streaming startup, Sparkify, has grown their user base and song database even more and want to move their data warehouse to a data lake. Their data resides in S3, in a directory of JSON logs on user activity on the app, as well as a directory with JSON metadata on the songs in their app.
As their data engineer, my task was to build an ETL pipeline that extracts their data from S3, processes them using Spark, and loads the data back into S3 as a set of dimensional tables. This will allow their analytics team to continue finding insights in what songs their users are listening to.
Following two datasets are used to ingest the data into data lake. Each dataset resides in S3 bucket:
-
Song Dataset: The first dataset is a subset of real data from the Million Song Dataset. Each file is in JSON format and contains metadata about a song and the artist of that song. The files are partitioned by the first three letters of each song's track ID. For example, here are file paths to two files in this dataset.
song_data/A/B/C/TRABCEI128F424C983.json
song_data/A/A/B/TRAABJL12903CDCF1A.json
And below is an example of what a single song file, TRAABJL12903CDCF1A.json, looks like.{"num_songs": 1, "artist_id": "ARJIE2Y1187B994AB7", "artist_latitude": null, "artist_longitude": null, "artist_location": "", "artist_name": "Line Renaud", "song_id": "SOUPIRU12A6D4FA1E1", "title": "Der Kleine Dompfaff", "duration": 152.92036, "year": 0}
-
Log Dataset: The second dataset consists of log files in JSON format generated by this event simulator based on the songs in the dataset above. These simulate activity logs from a music streaming app based on specified configurations. For example, here are file paths to two files in this dataset.
log_data/2018/11/2018-11-12-events.json
log_data/2018/11/2018-11-13-events.json
Using the song and log datasets, I created a star schema optimized for queries on song play analysis. This includes the following tables:
- The songplays fact table records in event data associated with song plays
- The users dimension table records users in the app
- The songs dimension table records songs in music database
- The artists dimension table records artists in music database
- The time dimension table stores timestamps of records in songplays broken down into specific units
- etl.py : This file contains the ETL pipeline that loads the songs and log data from S3 bucket, processes it using Spark and stores it back to S3 in the parquet format.
- dl.cfg: This file contains the AWS IAM credentials.
- Create an AWS IAM role with S3 read and write access. Then, edit the
dl.cfg
configuration file and put the newly created IAM credentials in it. - Create an S3 Bucket to store the dimensional tables in parquet format. Then, edit the
etl.py
file and replace the output data path with the newly created s3 Bucket path. - Run
etl.py
file that loads the songs and log data from S3 bucket, processes it using Spark and stores it back to S3 in the parquet format. - The stored files can be used by analytics team to continue finding insights about what songs their users are listening to.