From fbf2f927e4f9f363845e4c7dce07f2ac8c2a760e Mon Sep 17 00:00:00 2001
From: mz-ko <hyupko@megazone.com>
Date: Mon, 22 Jul 2024 14:00:37 +0900
Subject: [PATCH] fix: Add ssm exception for collect.

Signed-off-by: mz-ko <hyupko@megazone.com>
---
 .../connector/ec2/instance_connector.py       | 64 ++++++++++---------
 1 file changed, 35 insertions(+), 29 deletions(-)

diff --git a/src/plugin/connector/ec2/instance_connector.py b/src/plugin/connector/ec2/instance_connector.py
index db991e0..97dcd22 100644
--- a/src/plugin/connector/ec2/instance_connector.py
+++ b/src/plugin/connector/ec2/instance_connector.py
@@ -2,7 +2,7 @@
 from spaceone.core.error import ERROR_REQUIRED_PARAMETER
 
 from plugin.conf.cloud_service_conf import *
-from plugin.connector.base import ResourceConnector
+from plugin.connector.base import ResourceConnector, _LOGGER
 
 
 class InstanceConnector(ResourceConnector):
@@ -224,35 +224,41 @@ def _generate_query(is_paginate=False, **query):
     def describe_instance_information(self, instances, **query) -> dict:
         instance_information = {}
 
-        for i in range(0, len(instances), 100):
-            instances_chunk = instances[i : i + 100]
-            query = self._generate_query(is_paginate=True, **query)
-            query.update(
-                {
-                    "Filters": [
-                        {
-                            "Key": "InstanceIds",
-                            "Values": self._get_instance_ids_from_instance(
-                                instances_chunk
-                            ),
-                        },
-                    ]
-                }
-            )
-            paginator = self.ssm_client.get_paginator("describe_instance_information")
-            response_iterator = paginator.paginate(**query)
-
-            for data in response_iterator:
-                for instance in data.get("InstanceInformationList", []):
-                    instance_information.update(
-                        {
-                            instance.get("InstanceId"): {
-                                "platform_type": instance.get("PlatformType"),
-                                "platform_name": instance.get("PlatformName"),
-                                "platform_version": instance.get("PlatformVersion"),
+        try:
+            for i in range(0, len(instances), 100):
+                instances_chunk = instances[i : i + 100]
+                query = self._generate_query(is_paginate=True, **query)
+                query.update(
+                    {
+                        "Filters": [
+                            {
+                                "Key": "InstanceIds",
+                                "Values": self._get_instance_ids_from_instance(
+                                    instances_chunk
+                                ),
+                            },
+                        ]
+                    }
+                )
+                paginator = self.ssm_client.get_paginator(
+                    "describe_instance_information"
+                )
+                response_iterator = paginator.paginate(**query)
+
+                for data in response_iterator:
+                    for instance in data.get("InstanceInformationList", []):
+                        instance_information.update(
+                            {
+                                instance.get("InstanceId"): {
+                                    "platform_type": instance.get("PlatformType"),
+                                    "platform_name": instance.get("PlatformName"),
+                                    "platform_version": instance.get("PlatformVersion"),
+                                }
                             }
-                        }
-                    )
+                        )
+        except Exception as e:
+            _LOGGER.error(f"[instance_information] {e}")
+            instance_information = {}
 
         return instance_information