convert Set too

kiegroup · Oct 29, 2024 · 754f013 · 754f013
1 parent b8de3e2
commit 754f013
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/...ok-integration-api/src/main/java/org/drools/ansible/rulebook/integration/api/LogUtil.java b/...ok-integration-api/src/main/java/org/drools/ansible/rulebook/integration/api/LogUtil.java
@@ -1,11 +1,8 @@
 package org.drools.ansible.rulebook.integration.api;
 
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.LinkedHashMap;
 import java.util.List;
 import java.util.Map;
-
+import java.util.Set;
 
 public class LogUtil {
 
@@ -18,12 +15,7 @@ private LogUtil() {
             Integer.class, "int",
             Boolean.class, "bool",
             String.class, "str",
-            Double.class, "float",
-            List.class, "list",
-            ArrayList.class, "list",
-            Map.class, "dict",
-            LinkedHashMap.class, "dict",
-            HashMap.class, "dict"
+            Double.class, "float"
     );
 
     public static String convertJavaClassToPythonClass(Class<?> javaClass) {
@@ -34,6 +26,8 @@ public static String convertJavaClassToPythonClass(Class<?> javaClass) {
             return "list";
         } else if (Map.class.isAssignableFrom(javaClass)) {
             return "dict";
+        } else if (Set.class.isAssignableFrom(javaClass)) {
+            return "set";
         }
         return javaClass.getSimpleName();
     }