diff --git a/metadata-io/src/main/java/com/linkedin/metadata/timeseries/elastic/query/ESAggregatedStatsDAO.java b/metadata-io/src/main/java/com/linkedin/metadata/timeseries/elastic/query/ESAggregatedStatsDAO.java index 1254d621b3d..f6aad3bd635 100644 --- a/metadata-io/src/main/java/com/linkedin/metadata/timeseries/elastic/query/ESAggregatedStatsDAO.java +++ b/metadata-io/src/main/java/com/linkedin/metadata/timeseries/elastic/query/ESAggregatedStatsDAO.java @@ -56,6 +56,7 @@ public class ESAggregatedStatsDAO { private static final String ES_TERMS_AGGREGATION_PREFIX = "terms_"; private static final String ES_MAX_AGGREGATION_PREFIX = "max_"; private static final String ES_FIELD_TIMESTAMP = "timestampMillis"; + private static final String ES_FIELD_URN = "urn"; private static final String ES_AGG_TIMESTAMP = ES_AGGREGATION_PREFIX + ES_FIELD_TIMESTAMP; private static final String ES_AGG_MAX_TIMESTAMP = ES_AGGREGATION_PREFIX + ES_MAX_AGGREGATION_PREFIX + ES_FIELD_TIMESTAMP; @@ -172,11 +173,13 @@ public class ESAggregatedStatsDAO { if (fieldPath.equals(ES_FIELD_TIMESTAMP)) { return DataSchema.Type.LONG; } - /* TODO: Remove if not needed after merge. + if (fieldPath.equals(ES_FIELD_URN)) { + return DataSchema.Type.STRING; + } if (fieldPath.equals(MappingsBuilder.EVENT_GRANULARITY)) { return DataSchema.Type.RECORD; } - */ + String[] memberParts = fieldPath.split("\\."); if (memberParts.length == 1) { // Search in the timeseriesFieldSpecs. @@ -439,18 +442,6 @@ public class ESAggregatedStatsDAO { return lastAggregationBuilder; } - private boolean isIntegralType(DataSchema.Type fieldType) { - switch (fieldType) { - case INT: - case FLOAT: - case DOUBLE: - case LONG: - return true; - default: - return false; - } - } - private GenericTable generateResponseFromElastic(SearchResponse searchResponse, GroupingBucket[] groupingBuckets, AggregationSpec[] aggregationSpecs, AspectSpec aspectSpec) { GenericTable resultTable = new GenericTable(); diff --git a/metadata-service/restli-servlet-impl/src/main/java/com/linkedin/metadata/resources/analytics/Analytics.java b/metadata-service/restli-servlet-impl/src/main/java/com/linkedin/metadata/resources/analytics/Analytics.java index a0afcda7333..1f19094a746 100644 --- a/metadata-service/restli-servlet-impl/src/main/java/com/linkedin/metadata/resources/analytics/Analytics.java +++ b/metadata-service/restli-servlet-impl/src/main/java/com/linkedin/metadata/resources/analytics/Analytics.java @@ -53,8 +53,12 @@ public class Analytics extends SimpleResourceTemplate