From 7e935f177c2c5195fc3055d649fa962d84b275e9 Mon Sep 17 00:00:00 2001
From: Henning Baldersheim <balder@yahoo-inc.com>
Date: Wed, 12 Feb 2020 13:42:08 +0000
Subject: - Add trace as a field in the josn access log. - Add the trace top
 the accesslog entry if debugging is enabled in SearchHandler.

---
 .../com/yahoo/container/logging/TraceRenderer.java | 173 +++++++++++++++++++++
 1 file changed, 173 insertions(+)
 create mode 100644 container-accesslogging/src/main/java/com/yahoo/container/logging/TraceRenderer.java

(limited to 'container-accesslogging/src/main/java/com/yahoo/container/logging/TraceRenderer.java')

diff --git a/container-accesslogging/src/main/java/com/yahoo/container/logging/TraceRenderer.java b/container-accesslogging/src/main/java/com/yahoo/container/logging/TraceRenderer.java
new file mode 100644
index 00000000000..b974111b007
--- /dev/null
+++ b/container-accesslogging/src/main/java/com/yahoo/container/logging/TraceRenderer.java
@@ -0,0 +1,173 @@
+package com.yahoo.container.logging;
+
+import com.yahoo.yolean.trace.TraceNode;
+import com.yahoo.yolean.trace.TraceVisitor;
+import com.fasterxml.jackson.core.JsonGenerator;
+
+import java.io.IOException;
+
+public class TraceRenderer extends TraceVisitor {
+    private static final String TRACE_CHILDREN = "children";
+    private static final String TRACE_MESSAGE = "message";
+    private static final String TRACE_TIMESTAMP = "timestamp";
+    private static final String TRACE = "trace";
+
+    private final long basetime;
+    private final JsonGenerator generator;
+    private final FieldConsumer fieldConsumer;
+    private boolean hasFieldName = false;
+    int emittedChildNesting = 0;
+    int currentChildNesting = 0;
+    private boolean insideOpenObject = false;
+
+    public interface FieldConsumer {
+        void consume(Object object) throws IOException;
+    }
+
+    private static class Consumer implements FieldConsumer {
+        private final JsonGenerator generator;
+
+        Consumer(JsonGenerator generator) {
+            this.generator = generator;
+        }
+
+        @Override
+        public void consume(Object object) throws IOException {
+            generator.writeObject(object);
+        }
+    }
+
+    TraceRenderer(JsonGenerator generator, long basetime) {
+        this(generator, new Consumer(generator), basetime);
+    }
+    public TraceRenderer(JsonGenerator generator, FieldConsumer consumer, long basetime) {
+        this.generator = generator;
+        this.fieldConsumer = consumer;
+        this.basetime = basetime;
+    }
+
+    @Override
+    public void entering(TraceNode node) {
+        ++currentChildNesting;
+    }
+
+    @Override
+    public void leaving(TraceNode node) {
+        conditionalEndObject();
+        if (currentChildNesting == emittedChildNesting) {
+            try {
+                generator.writeEndArray();
+                generator.writeEndObject();
+            } catch (IOException e) {
+                throw new TraceRenderWrapper(e);
+            }
+            --emittedChildNesting;
+        }
+        --currentChildNesting;
+    }
+
+    @Override
+    public void visit(TraceNode node) {
+        try {
+            doVisit(node.timestamp(), node.payload(), node.children().iterator().hasNext());
+        } catch (IOException e) {
+            throw new TraceRenderWrapper(e);
+        }
+    }
+
+    private void doVisit(long timestamp, Object payload, boolean hasChildren) throws IOException {
+        boolean dirty = false;
+        if (timestamp != 0L) {
+            header();
+            generator.writeStartObject();
+            generator.writeNumberField(TRACE_TIMESTAMP, timestamp - basetime);
+            dirty = true;
+        }
+        if (payload != null) {
+            if (!dirty) {
+                header();
+                generator.writeStartObject();
+            }
+            generator.writeFieldName(TRACE_MESSAGE);
+            fieldConsumer.consume(payload);
+            dirty = true;
+        }
+        if (dirty) {
+            if (!hasChildren) {
+                generator.writeEndObject();
+            } else {
+                setInsideOpenObject(true);
+            }
+        }
+    }
+    private void header() {
+        fieldName();
+        for (int i = 0; i < (currentChildNesting - emittedChildNesting); ++i) {
+            startChildArray();
+        }
+        emittedChildNesting = currentChildNesting;
+    }
+
+    private void startChildArray() {
+        try {
+            conditionalStartObject();
+            generator.writeArrayFieldStart(TRACE_CHILDREN);
+        } catch (IOException e) {
+            throw new TraceRenderWrapper(e);
+        }
+    }
+
+    private void conditionalStartObject() throws IOException {
+        if (!isInsideOpenObject()) {
+            generator.writeStartObject();
+        } else {
+            setInsideOpenObject(false);
+        }
+    }
+
+    private void conditionalEndObject() {
+        if (isInsideOpenObject()) {
+            // This triggers if we were inside a data node with payload and
+            // subnodes, but none of the subnodes contained data
+            try {
+                generator.writeEndObject();
+                setInsideOpenObject(false);
+            } catch (IOException e) {
+                throw new TraceRenderWrapper(e);
+            }
+        }
+    }
+
+    private void fieldName() {
+        if (hasFieldName) {
+            return;
+        }
+
+        try {
+            generator.writeFieldName(TRACE);
+        } catch (IOException e) {
+            throw new TraceRenderWrapper(e);
+        }
+        hasFieldName = true;
+    }
+
+    boolean isInsideOpenObject() {
+        return insideOpenObject;
+    }
+
+    void setInsideOpenObject(boolean insideOpenObject) {
+        this.insideOpenObject = insideOpenObject;
+    }
+    public static final class TraceRenderWrapper extends RuntimeException {
+
+        /**
+         * Should never be serialized, but this is still needed.
+         */
+        private static final long serialVersionUID = 2L;
+
+        TraceRenderWrapper(IOException wrapped) {
+            super(wrapped);
+        }
+
+    }
+}
-- 
cgit v1.2.3