From a0340f2482327f7518adbbb12eaa7f05193eca34 Mon Sep 17 00:00:00 2001 From: Laerte Pereira <5853172+Laerte@users.noreply.github.com> Date: Wed, 18 Dec 2024 15:20:16 -0300 Subject: [PATCH 1/2] fix: Get the pid of xcom command dynamically --- .../src/airflow/providers/cncf/kubernetes/utils/pod_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/providers/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py b/providers/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py index f7a602417b7fc..c4dfa6223d886 100644 --- a/providers/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py +++ b/providers/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py @@ -855,7 +855,7 @@ def extract_xcom_kill(self, pod: V1Pod): _preload_content=False, ) ) as resp: - self._exec_pod_command(resp, "kill -2 1") + self._exec_pod_command(resp, "kill -9 $(pgrep -u $(whoami) -f trap)") def _exec_pod_command(self, resp, command: str) -> str | None: res = "" From 1e3a0c5dbeef42e443f9400dae0df16629fbfc0d Mon Sep 17 00:00:00 2001 From: Laerte Pereira <5853172+Laerte@users.noreply.github.com> Date: Wed, 18 Dec 2024 15:26:34 -0300 Subject: [PATCH 2/2] fix signal number --- .../src/airflow/providers/cncf/kubernetes/utils/pod_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/providers/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py b/providers/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py index c4dfa6223d886..e65f0a63ae28b 100644 --- a/providers/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py +++ b/providers/src/airflow/providers/cncf/kubernetes/utils/pod_manager.py @@ -855,7 +855,7 @@ def extract_xcom_kill(self, pod: V1Pod): _preload_content=False, ) ) as resp: - self._exec_pod_command(resp, "kill -9 $(pgrep -u $(whoami) -f trap)") + self._exec_pod_command(resp, "kill -2 $(pgrep -u $(whoami) -f trap)") def _exec_pod_command(self, resp, command: str) -> str | None: res = ""