k8s: get p2p messages by pod IP or service IP

Author: pinheadmz

src/backends/kubernetes/kubernetes_backend.py

@@ -10,6 +10,7 @@
 from cli.image import build_image
 from kubernetes import client, config
 from kubernetes.client.models.v1_pod import V1Pod
+from kubernetes.client.models.v1_service import V1Service
 from kubernetes.client.rest import ApiException
 from kubernetes.dynamic import DynamicClient
 from kubernetes.dynamic.exceptions import ResourceNotFoundError
@@ -121,6 +122,15 @@ def get_pod(self, pod_name: str) -> V1Pod | None:
             if e.status == 404:
                 return None
 
+    def get_service(self, service_name: str) -> V1Service | None:
+        try:
+            return cast(
+                V1Service, self.client.read_namespaced_service(name=service_name, namespace=self.namespace)
+            )
+        except ApiException as e:
+            if e.status == 404:
+                return None
+
     # We could enhance this by checking the pod status as well
     # The following pod phases are available: Pending, Running, Succeeded, Failed, Unknown
     # For example not able to pull image will be a phase of Pending, but the container status will be ErrImagePull
@@ -237,6 +247,7 @@ def get_messages(
         bitcoin_network: str = "regtest",
     ):
         b_pod = self.get_pod(self.get_pod_name(b_index, ServiceType.BITCOIN))
+        b_service = self.get_service(self.get_service_name(b_index))
         subdir = "/" if bitcoin_network == "main" else f"{bitcoin_network}/"
         base_dir = f"/root/.bitcoin/{subdir}message_capture"
         cmd = f"ls {base_dir}"
@@ -251,7 +262,7 @@ def get_messages(
         messages = []
 
         for dir_name in dirs:
-            if b_pod.status.pod_ip in dir_name:
+            if b_pod.status.pod_ip in dir_name or b_service.spec.cluster_ip in dir_name:
                 for file, outbound in [["msgs_recv.dat", False], ["msgs_sent.dat", True]]:
                     # Fetch the file contents from the container
                     file_path = f"{base_dir}/{dir_name}/{file}"