[incubator-kie-issues#1373] Define removeApplicationProperty (apache#…

…6008) Co-authored-by: Gabriele-Cardosi <[email protected]>
rgdoliveira · Aug 6, 2024 · f010daa · f010daa
1 parent 14907da
commit f010daa
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/...ommon/src/main/java/org/drools/codegen/common/DroolsModelApplicationPropertyProvider.java b/...ommon/src/main/java/org/drools/codegen/common/DroolsModelApplicationPropertyProvider.java
@@ -40,6 +40,11 @@ public Collection<String> getApplicationProperties() {
             public void setApplicationProperty(String key, String value) {
                 properties.put(key, value);
             }
+
+            @Override
+            public void removeApplicationProperty(String key) {
+                properties.remove(key);
+            }
         };
     }
 
@@ -48,4 +53,6 @@ public void setApplicationProperty(String key, String value) {
     Collection<String> getApplicationProperties();
 
     void setApplicationProperty(String key, String value);
+
+    void removeApplicationProperty(String key);
 }
diff --git a/...rools-codegen-common/src/main/java/org/drools/codegen/common/DroolsModelBuildContext.java b/...rools-codegen-common/src/main/java/org/drools/codegen/common/DroolsModelBuildContext.java
@@ -40,6 +40,8 @@ public interface DroolsModelBuildContext {
 
     void setApplicationProperty(String key, String value);
 
+    void removeApplicationProperty(String key);
+
     String getPackageName();
 
     ClassLoader getClassLoader();

diff --git a/...mmon/src/main/java/org/drools/codegen/common/context/AbstractDroolsModelBuildContext.java b/...mmon/src/main/java/org/drools/codegen/common/context/AbstractDroolsModelBuildContext.java
@@ -105,6 +105,11 @@ public void setApplicationProperty(String key, String value) {
         applicationProperties.setApplicationProperty(key, value);
     }
 
+    @Override
+    public void removeApplicationProperty(String key) {
+        applicationProperties.removeApplicationProperty(key);
+    }
+
     @Override
     public String getPackageName() {
         return packageName;