Skip to content

Commit 398ba01

Browse files
authored
[azure][fix]: Reimplemented fleet and clusters connection (#2137)
1 parent 869d5d6 commit 398ba01

File tree

1 file changed

+8
-7
lines changed

1 file changed

+8
-7
lines changed

plugins/azure/fix_plugin_azure/resource/containerservice.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,9 @@ class AzureFleet(MicrosoftResource):
112112
hub_profile: Optional[AzureFleetHubProfile] = field(default=None, metadata={'description': 'The FleetHubProfile configures the fleet hub.'}) # fmt: skip
113113
azure_fleet_identity: Optional[AzureManagedServiceIdentity] = field(default=None, metadata={'description': 'Managed service identity (system assigned and/or user assigned identities)'}) # fmt: skip
114114
resource_group: Optional[str] = field(default=None, metadata={"description": "Resource group name"})
115-
cluster_resource_id: Optional[str] = field(default=None, metadata={"description": "Reference to the cluster ID"})
115+
_cluster_resource_ids: Optional[List[str]] = field(
116+
default=None, metadata={"description": "Reference to the cluster IDs"}
117+
)
116118

117119
def post_process(self, graph_builder: GraphBuilder, source: Json) -> None:
118120
def collect_fleets() -> None:
@@ -128,18 +130,17 @@ def collect_fleets() -> None:
128130
items: List[Json] = graph_builder.client.list(api_spec)
129131
if not items:
130132
return
131-
item: Json = next(iter(items), {})
132-
133133
try:
134-
self.cluster_resource_id = item["properties"]["clusterResourceId"]
134+
self._cluster_resource_ids = [item["properties"]["clusterResourceId"] for item in items]
135135
except KeyError as e:
136-
log.warning(f"An error occured while setting cluster_resource_id: {e}")
136+
log.warning(f"An error occured while taking cluster id: {e}")
137137

138138
graph_builder.submit_work(service, collect_fleets)
139139

140140
def connect_in_graph(self, builder: GraphBuilder, source: Json) -> None:
141-
if cluster_id := self.cluster_resource_id:
142-
builder.add_edge(self, edge_type=EdgeType.default, clazz=AzureManagedCluster, id=cluster_id)
141+
if cluster_ids := self._cluster_resource_ids:
142+
for cluster_id in cluster_ids:
143+
builder.add_edge(self, edge_type=EdgeType.default, clazz=AzureManagedCluster, id=cluster_id)
143144

144145

145146
@define(eq=False, slots=False)

0 commit comments

Comments
 (0)