Use selectinload in trigger

apache · Jun 28, 2024 · 5a85601 · 5a85601
1 parent 8d5dbaa
commit 5a85601
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/airflow/models/trigger.py b/airflow/models/trigger.py
@@ -21,7 +21,7 @@
 from typing import TYPE_CHECKING, Any, Iterable
 
 from sqlalchemy import Column, Integer, String, Text, delete, func, or_, select, update
-from sqlalchemy.orm import joinedload, relationship
+from sqlalchemy.orm import joinedload, relationship, selectinload
 from sqlalchemy.sql.functions import coalesce
 
 from airflow.api_internal.internal_api_call import internal_api_call
@@ -74,7 +74,7 @@ class Trigger(Base):
         uselist=False,
     )
 
-    task_instance = relationship("TaskInstance", back_populates="trigger", lazy="joined", uselist=False)
+    task_instance = relationship("TaskInstance", back_populates="trigger", lazy="selectin", uselist=False)
 
     def __init__(
         self,
@@ -150,7 +150,7 @@ def bulk_fetch(cls, ids: Iterable[int], session: Session = NEW_SESSION) -> dict[
             select(cls)
             .where(cls.id.in_(ids))
             .options(
-                joinedload(cls.task_instance)
+                selectinload(cls.task_instance)
                 .joinedload(TaskInstance.trigger)
                 .joinedload(Trigger.triggerer_job)
             )