From fc398bb0ed50451d039835f6937d1f2dd64dffdc Mon Sep 17 00:00:00 2001 From: nipunayf Date: Mon, 8 Sep 2025 10:59:38 +0530 Subject: [PATCH] Use TreeMap for artifact categorization --- .../artifactsgenerator/ArtifactsGenerator.java | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/architecture-model-generator/modules/architecture-model-generator-core/src/main/java/io/ballerina/artifactsgenerator/ArtifactsGenerator.java b/architecture-model-generator/modules/architecture-model-generator-core/src/main/java/io/ballerina/artifactsgenerator/ArtifactsGenerator.java index 421399ade6..e73822d9c0 100644 --- a/architecture-model-generator/modules/architecture-model-generator-core/src/main/java/io/ballerina/artifactsgenerator/ArtifactsGenerator.java +++ b/architecture-model-generator/modules/architecture-model-generator-core/src/main/java/io/ballerina/artifactsgenerator/ArtifactsGenerator.java @@ -29,10 +29,10 @@ import java.util.ArrayList; import java.util.HashMap; -import java.util.LinkedHashMap; import java.util.List; import java.util.Map; import java.util.Optional; +import java.util.TreeMap; import java.util.concurrent.ConcurrentHashMap; import java.util.concurrent.ConcurrentMap; @@ -120,20 +120,12 @@ public static Map> artifacts(Project project) { .forEach(artifact -> { String category = Artifact.getCategory(artifact.type()); String artifactId = artifact.id(); - artifactMap.computeIfAbsent(category, k -> new HashMap<>()).put(artifactId, artifact); + artifactMap.computeIfAbsent(category, k -> new TreeMap<>()).put(artifactId, artifact); idMap.computeIfAbsent(category, k -> new ArrayList<>()).add(artifactId); }); documentMap.put(document.name(), idMap); }); - artifactMap.forEach((category, artifacts) -> { - Map sortedArtifacts = new LinkedHashMap<>(); - artifacts.entrySet().stream() - .sorted(Map.Entry.comparingByKey()) - .forEach(entry -> sortedArtifacts.put(entry.getKey(), entry.getValue())); - artifactMap.put(category, sortedArtifacts); - }); - ArtifactsCache.getInstance().initializeProject(project.sourceRoot().toString(), documentMap); return artifactMap; }