fix: log and skip if entity can't be joined (#95)

Author: aaron-steinfeld

hypertrace-graphql-entity-schema/src/main/java/org/hypertrace/graphql/entity/joiner/DefaultEntityJoinerBuilder.java

@@ -32,6 +32,7 @@
 import lombok.AllArgsConstructor;
 import lombok.Value;
 import lombok.experimental.Accessors;
+import lombok.extern.slf4j.Slf4j;
 import org.hypertrace.core.graphql.common.request.AttributeAssociation;
 import org.hypertrace.core.graphql.common.request.AttributeRequest;
 import org.hypertrace.core.graphql.common.request.FilterRequestBuilder;
@@ -58,6 +59,7 @@
 import org.hypertrace.graphql.metric.request.MetricRequest;
 import org.hypertrace.graphql.metric.schema.argument.AggregatableOrderArgument;
 
+@Slf4j
 class DefaultEntityJoinerBuilder implements EntityJoinerBuilder {
 
   private static final int ZERO_OFFSET = 0;
@@ -164,6 +166,18 @@ private <T> Collection<Cell<T, String, Entity>> buildResultCellsForSource(
         Map<String, String> requestedIdsByType,
         Table<String, String, Entity> entityResultTable) {
       return requestedIdsByType.entrySet().stream()
+          .filter(
+              typeIdPair -> {
+                if (!entityResultTable.contains(typeIdPair.getKey(), typeIdPair.getValue())) {
+                  log.error(
+                      "Requested Entity not present in response. Type: {}, ID: {}, Source {}",
+                      typeIdPair.getKey(),
+                      typeIdPair.getValue(),
+                      joinSource);
+                  return false;
+                }
+                return true;
+              })
           .map(
               typeIdPair ->
                   immutableCell(