refactor the code

wso2 · Oct 22, 2023 · 954eabb · 954eabb
1 parent 38858a7
commit 954eabb
Showing 1 changed file with 1 addition and 11 deletions.
diff --git a/...carbon/identity/api/server/extension/management/v1/function/ExtensionListItemBuilder.java b/...carbon/identity/api/server/extension/management/v1/function/ExtensionListItemBuilder.java
@@ -20,13 +20,10 @@
 
 import org.apache.commons.logging.Log;
 import org.apache.commons.logging.LogFactory;
-import org.wso2.carbon.identity.api.server.extension.management.common.utils.ExtensionMgtConstants;
 import org.wso2.carbon.identity.api.server.extension.management.common.utils.ExtensionMgtUtils;
 import org.wso2.carbon.identity.api.server.extension.management.v1.model.ExtensionListItem;
 import org.wso2.carbon.identity.extension.mgt.model.ExtensionInfo;
 
-import java.util.HashMap;
-import java.util.Map;
 import java.util.function.Function;
 
 /**
@@ -51,14 +48,7 @@ public ExtensionListItem apply(ExtensionInfo extensionInfo) {
         extensionListItem.setSelf(ExtensionMgtUtils.getExtensionInfoLocation(extensionInfo.getType(),
                 extensionInfo.getId()));
         if (extensionInfo.getCustomAttributes() != null) {
-            for (Map<String, Object> customAttributeMap : extensionInfo.getCustomAttributes()) {
-                Map<String, Object> customAttribute = new HashMap<>();
-                customAttribute.put(ExtensionMgtConstants.KEY,
-                        customAttributeMap.get(ExtensionMgtConstants.KEY));
-                customAttribute.put(ExtensionMgtConstants.VALUE,
-                        customAttributeMap.get(ExtensionMgtConstants.VALUE).toString());
-                extensionListItem.addCustomAttributesItem(customAttribute);
-            }
+            extensionListItem.setCustomAttributes(extensionInfo.getCustomAttributes());
         }
         return extensionListItem;
     }