内容简介:Machine learning is the hot topic of the industry. It won't be so cool if not for the data processing involvedAirflow is an ETL(Extract, Transform, Load) workflow orchestration tool, used in data transformation pipelines.Imagine you have an ML model that d
An Introduction to Apache Airflow
What is Airflow?
Airflow is a platform created by the community to programmatically author, schedule, and monitor workflows.
Machine learning is the hot topic of the industry. It won't be so cool if not for the data processing involved
Airflow is an ETL(Extract, Transform, Load) workflow orchestration tool, used in data transformation pipelines.
Uses of Airflow
Imagine you have an ML model that does twitter sentiment analysis. Now you want to run that model for your favorite people on twitter for their tweets every day. Such a workflow would look something like this.
As you can see, the data flows from one end of the pipeline to the other end. There can be branches, but no cycles.
What problems does AirflowAirflow solve?
Crons are an age-old way of scheduling tasks.
- With cron creating and maintaining a relationship between tasks is a nightmare, whereas, in AirflowAirflow, it is as simple as writing Python code.
- Cron needs external support to log, track, and manage tasks. Airflow UI to track and monitor the workflow execution
- Cron jobs are not reproducible unless externally configured. The AirflowAirflow keeps an audit trail of all tasks executed.
- Scalable
How to define a workflow in AirflowAirflow?
Workflows are defined using Python files.
DAG
Airflow provides DAG
Python class to create a Directed Acyclic Graph, a representation of the workflow.
from Airflow.models import DAG from airflow.utils.dates import days_ago args = { 'start_date': days_ago(0), } dag = DAG( dag_id='example_bash_operator', default_args=args, schedule_interval='* * * * *', )
start_date
enables you to run a task on a particular date.
Schedule_interval
is the interval in which each workflow is supposed to run. '* * * * *'
means the tasks need to run every minute. Don't scratch your brain over this syntax. You can play around with these using https://crontab.guru/ .
Operator
Operators
define the nodes of the DAG. Each operator is an independent task.
In the following example, we use two Operators
from airflow.operators.bash_operator import BashOperator from airflow.operators.python_operator import PythonOperator
-
PythonOperator
which calls a python function
def print_function(): print ("Hey I am a task") run_this_last = PythonOperator( task_id='run_this_last', dag=dag, python_callable=print_function )
-
BashOperator
which runs a bash command
run_this = BashOperator( task_id='run_after_loop', bash_command='echo 1', dag=dag, )
- The tasks are linked together using
>>
python operator.
run_this >> run_this_last
A sample DAG with branches would look something like this.
Airflow Architecture
Airflow has 4 major components.
Webserver
The webserver is the component that is responsible for handling all the UI and REST APIs.
Scheduler
Scheduler goes through the DAGs every n
seconds and schedules the task to be executed.
The scheduler also has an internal component called Executor . The executor is responsible for spinning up workers and executing the task to completion.
Worker
Workers run the task that is being handed over by the executor.
Types of Executor
SequentialExecutor
SequentialExecutor runs only one task at a time. The workers run the same machine as the scheduler is.
Pros
- Simple and easy to setup
- Good for testing DAGs during development
Cons
- Not scalable
- It cannot run multiple tasks at the same time.
- Not suitable for production
LocalExecutor
LocalExecutor is the same as the Sequential Executor, except it can run multiple tasks at a time.
Pros
- Can run multiple tasks
- Good for running DAGs during development
Cons
- Not scalable
- Single point of failure
- Not suitable for production
CeleryExecutor
Celery is used for running distributed asynchronous python tasks.
Hence, CeleryExecutor has been a part of AirflowAirflow for a long time, even before Kubernetes.
CeleryExecutors has a fixed number of workers running to pick-up the tasks as they get scheduled.
Pros
- It provides scalability.
- Celery manages the workers. In case of a failure, Celery spins up a new one.
Cons
- Celery needs RabbitMQ/Redis to for queuing the task, which is reinventing the wheel of what AirflowAirflow already supports.
- The above dependency also makes the setup complex.
KubernetesExecutor
KubernetesExecutor runs each task in an individual Kubernetes pod. Unlike CeleryCelery, it spins up worker pods on demand , hence enabling maximum usage of resources.
Pros
- It Combines the pros of scalability and simplicity of CeleryExecutor and LocalExecutor.
- Fine-grained control over resources allocated to tasks. One can define the amount of CPU/memory required at a task level.
Cons
- Kubernetes is new to AirflowAirflow, and the documentation is not straightforward.
Now that we have understood Airflow's basics let's learn how to write our workflow in the next post.
以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网
猜你喜欢:本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们。