Updated current.info.

yahoo · Oct 17, 2016 · 4e146bd · 4e146bd
1 parent f2ef4ac
commit 4e146bd
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/fili-core/src/main/java/com/yahoo/bard/webservice/logging/LogBlock.java b/fili-core/src/main/java/com/yahoo/bard/webservice/logging/LogBlock.java
@@ -36,6 +36,18 @@ public void add(LogInfo phase) {
         body.put(phase.getName(), phase);
     }
 
+    /**
+     * Create a new LogBlock with the same body but updated uuid.
+     *
+     * @param uuid  New uuid.
+     * @return New LogBlock with updated uuid.
+     */
+    public LogBlock withUuid(String uuid) {
+        LogBlock newLogBlock = new LogBlock(uuid);
+        body.entrySet().forEach(entry -> newLogBlock.body.put(entry.getKey(), entry.getValue()));
+        return newLogBlock;
+    }
+
     /**
      * Add an entry of given class in this LogBlock that holds no information.
      * It is meant to be used only to define the order of {@link LogInfo} parts inside a LogBlock.

diff --git a/fili-core/src/main/java/com/yahoo/bard/webservice/logging/RequestLog.java b/fili-core/src/main/java/com/yahoo/bard/webservice/logging/RequestLog.java
@@ -408,6 +408,7 @@ public static String getId() {
     public static void addIdPrefix(String idPrefix) {
         RequestLog current = RLOG.get();
         String newId = idPrefix + getId();
+        current.info = current.info.withUuid(newId);
         current.logId = newId;
         MDC.put(ID_KEY, newId);
     }