[MPH-183] Add getLocationKeys for help plugin

infosupport · Mar 15, 2024 · 9881c22 · 9881c22
1 parent b50cda2
commit 9881c22
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/src/mdo/model-v3.vm b/src/mdo/model-v3.vm
@@ -52,6 +52,7 @@
     #set ( $dummy = $imports.add( "java.util.HashMap" ) )
     #set ( $dummy = $imports.add( "java.util.List" ) )
     #set ( $dummy = $imports.add( "java.util.Map" ) )
+    #set ( $dummy = $imports.add( "java.util.Set" ) )
     #set ( $dummy = $imports.add( "java.util.Objects" ) )
     #set ( $dummy = $imports.add( "java.util.stream.Collectors" ) )
     #set ( $dummy = $imports.add( "java.util.stream.Stream" ) )
@@ -292,6 +293,10 @@ public class ${class.name}
                         .importedFrom(location.toApiLocation()).build());
     }
 
+    public Set<Object> getLocationKeys() {
+        return getDelegate().getLocationKeys();
+    }
+
     #end
     protected boolean replace(Object oldDelegate, Object newDelegate) {
         if (super.replace(oldDelegate, newDelegate)) {

diff --git a/src/mdo/model.vm b/src/mdo/model.vm
@@ -49,6 +49,7 @@
     #set ( $dummy = $imports.add( "java.util.Collections" ) )
     #set ( $dummy = $imports.add( "java.util.HashMap" ) )
     #set ( $dummy = $imports.add( "java.util.Map" ) )
+    #set ( $dummy = $imports.add( "java.util.Set" ) )
     #set ( $dummy = $imports.add( "org.apache.maven.api.annotations.Experimental" ) )
     #set ( $dummy = $imports.add( "org.apache.maven.api.annotations.Generated" ) )
     #set ( $dummy = $imports.add( "org.apache.maven.api.annotations.Immutable" ) )
@@ -257,6 +258,13 @@ public class ${class.name}
         return locations != null ? locations.get(key) : null;
     }
 
+    /**
+    * Gets the keys of the locations of the input source.
+    */
+    public Set<Object> getLocationKeys() {
+        return locations != null ? locations.keySet() : null;
+    }
+
     /**
      * Gets the input location that caused this model to be read.
      */