fix: store secrets on disk when partial arn is used

aws-greengrass · Nov 27, 2024 · 5d5b169 · 5d5b169
1 parent a21b165
commit 5d5b169
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/java/com/aws/greengrass/secretmanager/LocalStoreMap.java b/src/main/java/com/aws/greengrass/secretmanager/LocalStoreMap.java
@@ -116,9 +116,9 @@ private void save(List<SecretConfiguration> secretConfiguration) {
                 secretConfiguration.forEach((secretConfig) -> {
                     secretConfig.getLabels().forEach((label) -> {
                         String arn = secretConfig.getArn();
-                        if (secrets.containsKey(arn) && secrets.get(arn).responseMap.containsKey(label)) {
-                            responses.add(secrets.get(arn).responseMap.get(label));
-                        }
+                        secrets.entrySet().stream().filter(entry -> entry.getKey().contains(arn))
+                                .filter(entry -> entry.getValue().responseMap.containsKey(label))
+                                .forEach(entry -> responses.add(entry.getValue().responseMap.get(label)));
                     });
                 });
             }