feat: get_batch_dag bug2

This commit is contained in:
caoqianming 2025-03-25 09:54:57 +08:00
parent 11d7e1a772
commit 0e8cce0669
1 changed files with 3 additions and 1 deletions

View File

@ -915,7 +915,9 @@ def get_batch_dag(batch_number: str):
while len(nodes_set) > prev_size:
prev_size = len(nodes_set)
# 查询所有与当前批次相关的记录作为source或target
logs = BatchLog.objects.filter(Q(source__id__in=nodes_set) | Q(target__ide__in=nodes_set)).order_by("create_time")
logs = BatchLog.objects.filter(Q(source__id__in=nodes_set) | Q(target__id__in=nodes_set)).select_related(
"source", "target"
).order_by("create_time")
# 处理每条记录,扩展节点和边
for log in logs:
source = log.source.id