Improve data quality using Apache Airflow and check operator
The Data Team at Qubole collects usage and telemetry data from a million machines a month. We run many complex ETL workflows to process this data and provide reports, insights and recommendations to customers, analysts and data scientists. We use open source distribution of Apache Airflow to orchestrate our ETLs and process more than 1 terabyte of data daily.
These ETLs differ in terms of frequencies, types of data, transformation logic and their SLA’s. Due to the volume of data and differences amongst ETLs, it becomes difficult to monitor the quality of data. Errors are introduced at all stages - extraction, transformation or load and usually happen due to infrastructural or logical issues.
In order to catch these errors, we came up with the idea of using assert queries, just like we have assert statements in a unit test framework. These queries would run after an extraction/transformation/load step has finished and run some predefined diagnostic queries on the data to match the output against some expected value.
In this talk, I will
Discuss the complexities involved in detecting discrepancies in the output of any data transformation process and protecting any downstream process in case of any issue.
Introduce the approach we have adopted for running these assert queries based on the Check operator in Apache Airflow to quantify data quality and alert on it.
Discuss the enhancements we have made in the Qubole’s fork of Apache Airflow’s check operator in order to use it at a bigger scale and with more variety of data. We plan to contribute these enhancements back to Apache Airflow soon.
Talk about the lessons learnt and best practices in maintaining data sanity for data in motion.
We have integrated most of our ETLs with these data quality verification techniques, and the results look promising. We have been able to make this work across ETLs having nothing in common but the fact that they run on Apache Airflow.
- Data quality issues we faced with data ingestion/transformation.
- Approach we have adopted using Apache airflow check operators.
- Enhancements we had to make to Check operators.
- Integration of Apache Airflow Check operators with our ETLs.
- Challenges faced in developing the alerting framework.
- Lesson learnt and best practices in using Apache Airflow for data quality checks.
- Limitations and Future work.
Sakshi is a graduate from BITS Pilani and has been working with Qubole for the last 2 years. She has worked with the data team at Qubole and was involved in building a data streaming platform and data warehouse for the company.