Back to Airflow

Priority Weights

airflow-core/docs/administration-and-deployment/priority-weight.rst

3.2.14.0 KB
Original Source

.. Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements. See the NOTICE file distributed with this work for additional information regarding copyright ownership. The ASF licenses this file to you under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at

.. http://www.apache.org/licenses/LICENSE-2.0

.. Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.

.. _concepts:priority-weight:

Priority Weights

priority_weight defines priorities in the executor queue. The default priority_weight is 1, and can be bumped to any integer; larger numbers mean higher priority. Moreover, each task has a true priority_weight that is calculated based on its weight_rule which defines the weighting method used for the effective total priority weight of the task.

Below are the weighting methods. By default, Airflow's weighting method is downstream.

.. grid:: 3

.. grid-item-card:: downstream :shadow: none

The effective weight of the task is the aggregate sum of all
downstream descendants. As a result, upstream tasks will have
higher weight and will be scheduled more aggressively when
using positive weight values. This is useful when you have
multiple Dag run instances and desire to have all upstream
tasks to complete for all runs before each Dag can continue
processing downstream tasks.

.. grid-item-card:: upstream :shadow: none

The effective weight is the aggregate sum of all upstream ancestors.
This is the opposite where downstream tasks have higher weight
and will be scheduled more aggressively when using positive weight
values. This is useful when you have multiple Dag run instances
and prefer to have each Dag complete before starting upstream
tasks of other Dag runs.

.. grid-item-card:: absolute :shadow: none

The effective weight is the exact ``priority_weight`` specified
without additional weighting. You may want to do this when you
know exactly what priority weight each task should have.
Additionally, when set to ``absolute``, there is bonus effect of
significantly speeding up the task creation process as for very
large Dags.

The priority_weight parameter can be used in conjunction with :ref:concepts:pool.

.. note::

As most database engines are using 32-bit for integers, the maximum value for any calculated or
defined ``priority_weight`` is 2,147,483,647 and the minimum value is -2,147,483,648.

Custom Weight Rule

.. versionadded:: 2.9.0

You can implement your own custom weighting method by extending the PriorityWeightStrategy class and registering it in a plugin.

.. exampleinclude:: /../src/airflow/example_dags/plugins/decreasing_priority_weight_strategy.py :language: python :dedent: 0 :start-after: [START custom_priority_weight_strategy] :end-before: [END custom_priority_weight_strategy]

To check if the custom priority weight strategy is already available in Airflow, you can run the bash command airflow plugins. Then to use it, you can create an instance of the custom class and provide it in the weight_rule parameter of the task or provide the path of the custom class:

.. exampleinclude:: /../src/airflow/example_dags/example_custom_weight.py :language: python :dedent: 0 :start-after: [START example_custom_weight_dag] :end-before: [END example_custom_weight_dag]

After the Dag is running, you can check the priority_weight parameter on the task to verify that it is using the custom priority strategy rule.

|experimental|