github/KubeDiagrams
1
0
Fork 0
mirror of https://github.com/philippemerle/KubeDiagrams.git synced 2026-05-18 09:46:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a label to service to workload edges. The label is .

Browse Source
This commit is contained in:
Philippe Merle
2025-03-05 11:47:27 +01:00
parent 3e5272b6e9
commit 1af0865366
1 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
bin/kube-diagrams
View File

@@ -589,7 +589,22 @@ class EdgesContext(list):
Add edges from a service resource to workload resources,
endpoint slices and endpoints.
"""
self.add_all_workload_resources("spec.selector")
labels = []
for service_port in query_path(self.resource, "spec.ports", []):
port = service_port.get('port')
targetPort = service_port.get('targetPort')
protocol = service_port.get('protocol', 'TCP')
if targetPort is None or targetPort == port:
label = f"{port}/{protocol}"
else:
label = f"{port}->{targetPort}/{protocol}"
labels.append(label)
edge = {
**get_edge_config("SELECTOR"),
"label": f" {'\n'.join(labels)}",
"fontsize": "10"
}
self.add_all_workload_resources("spec.selector", edge_kind=edge)
self.add_all_resources_matching_labels(
"EndpointSlice",
{
@@ -809,9 +824,9 @@ def process_edges():
else:
edge_configuration = get_edge_config(edge_name)
if edge_configuration.get("direction") == "up":
_ = diagram_nodes[edge_from] \
_ = diagram_nodes[edge_to] \
>> Edge(**get_edge_config("INVISIBLE")) \
>> diagram_nodes[edge_to]
>> diagram_nodes[edge_from]
_ = diagram_nodes[edge_from] \
>> Edge(**edge_configuration) \
>> diagram_nodes[edge_to]