correctly get string label value for each jsonNode types

hypertrace · Aug 10, 2024 · 910fc4e · 910fc4e
1 parent 39ea48e
commit 910fc4e
Showing 1 changed file with 16 additions and 1 deletion.
diff --git a/...rc/main/java/org/hypertrace/core/documentstore/metric/BaseDocStoreMetricProviderImpl.java b/...rc/main/java/org/hypertrace/core/documentstore/metric/BaseDocStoreMetricProviderImpl.java
@@ -101,7 +101,9 @@ public List<DocStoreMetric> getCustomMetrics(final CustomMetricConfig customMetr
                         jsonNodeMap.entrySet().iterator(), Spliterator.ORDERED),
                     false)
                 .filter(entry -> !VALUE_KEY.equals(entry.getKey()))
-                .collect(toUnmodifiableMap(Entry::getKey, entry -> entry.getValue().textValue()));
+                .collect(
+                    toUnmodifiableMap(
+                        Entry::getKey, entry -> getStringLabelValue(entry.getValue())));
 
         final DocStoreMetric metric =
             DocStoreMetric.builder()
@@ -119,4 +121,17 @@ public List<DocStoreMetric> getCustomMetrics(final CustomMetricConfig customMetr
       return emptyList();
     }
   }
+
+  private String getStringLabelValue(JsonNode jsonNode) {
+    switch (jsonNode.getNodeType()) {
+      case STRING:
+        return jsonNode.textValue();
+      case NUMBER:
+      case BOOLEAN:
+        return String.valueOf(jsonNode.numberValue());
+      default:
+        throw new IllegalArgumentException(
+            String.format("Unsupported JSON node type: %s", jsonNode.getNodeType()));
+    }
+  }
 }