Update the API list generation logic

wso2 · Nov 11, 2024 · 3ea3869 · 3ea3869
1 parent 5dde35a
commit 3ea3869
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 17 deletions.
diff --git a/...y/src/main/java/org/wso2/micro/integrator/observability/metric/handler/MetricHandler.java b/...y/src/main/java/org/wso2/micro/integrator/observability/metric/handler/MetricHandler.java
@@ -312,8 +312,9 @@ private void incrementInboundEndpointErrorCount(String name) {
      */
     private String getApiName(String contextPath, MessageContext synCtx) {
         Collection<API> apiList = synCtx.getEnvironment().getSynapseConfiguration().getAPIs();
-        Collection<API> withVersionsApiList = getVersionsApiList(apiList);
-        Collection<API> defaultApiList = getDefaultApiList(apiList);
+        Collection<API> withVersionsApiList = new ArrayList<>();
+        Collection<API> defaultApiList = new ArrayList<>();
+        updateApiLists(apiList, withVersionsApiList, defaultApiList);
         if (!withVersionsApiList.isEmpty()) {
             String apiName = getResolvedApiName(contextPath, synCtx, withVersionsApiList);
             if (apiName != null) {
@@ -340,24 +341,15 @@ private static String getResolvedApiName(String contextPath, MessageContext synC
         return apiName;
     }
 
-    private Collection<API> getVersionsApiList(Collection<API> apiList) {
-        Collection<API> withVersionsApiList = new ArrayList<>();
+    private void updateApiLists(Collection<API> apiList, Collection<API> withVersionsApiList,
+                                           Collection<API> defaultApiList) {
         for (API api : apiList) {
             if (StringUtils.isNotBlank(api.getVersionStrategy().getVersion())) {
                 withVersionsApiList.add(api);
-            }
-        }
-        return withVersionsApiList;
-    }
-
-    private Collection<API> getDefaultApiList(Collection<API> apiList) {
-        Collection<API> defaultApiList = new ArrayList<>();
-        for (API api : apiList) {
-            if (!StringUtils.isNotBlank(api.getVersionStrategy().getVersion())) {
+            } else {
                 defaultApiList.add(api);
             }
         }
-        return defaultApiList;
     }
 
     /**

diff --git a/...nitializer/src/main/java/org/wso2/micro/integrator/initializer/ServiceBusInitializer.java b/...nitializer/src/main/java/org/wso2/micro/integrator/initializer/ServiceBusInitializer.java
@@ -172,10 +172,10 @@ protected void activate(ComponentContext ctxt) {
             }
             SynapseEnvironment synapseEnvironment = contextInfo.getSynapseEnvironment();
             List handlers = synapseEnvironment.getSynapseHandlers();
-            if (System.getProperty(ServiceBusConstants.ENABLE_PROMETHEUS_API_PROPERTY) != null
-                    && System.getProperty(ServiceBusConstants.ENABLE_PROMETHEUS_API_PROPERTY).equals("false")) {
+            String prometheusApiEnabled = System.getProperty(ServiceBusConstants.ENABLE_PROMETHEUS_API_PROPERTY);
+            if ("false".equals(prometheusApiEnabled)) {
                 handlers.remove(handlers.stream().filter(c -> c instanceof MetricHandler).findFirst().orElse(null));
-            } else {
+            } else if ("true".equals(prometheusApiEnabled)) {
                 if (!handlers.stream().anyMatch(c -> c instanceof MetricHandler)) {
                     handlers.add(new MetricHandler());
                 }