Bug 1672992 Modify main ping payloads when writing to BQ

See https://bugzilla.mozilla.org/show_bug.cgi?id=#c2
mozilla · Dec 9, 2020 · 3c3c03c · 3c3c03c
1 parent 789bf75
commit 3c3c03c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/...c/main/java/com/mozilla/telemetry/ingestion/core/transform/PubsubMessageToObjectNode.java b/...c/main/java/com/mozilla/telemetry/ingestion/core/transform/PubsubMessageToObjectNode.java
@@ -345,6 +345,15 @@ public ObjectNode apply(TableId tableId, Map<String, String> attributes, byte[]
       if (additionalProperties != null) {
         contents.put(FieldName.ADDITIONAL_PROPERTIES, Json.asString(additionalProperties));
       }
+
+      // TODO: REMOVE! This exists only for testing as described in Bug 1672992
+      if ("telemetry".equals(attributes.get(Attribute.DOCUMENT_NAMESPACE))
+          && "main".equals(attributes.get(Attribute.DOCUMENT_TYPE))) {
+        ObjectNode payload = (ObjectNode) contents.path("payload");
+        JsonNode histograms = payload.path("histograms");
+        payload.set("histograms2", histograms);
+      }
+
       return contents;
     }