ODP-1451 LLAP MDC from NDC string class cast exception. (#15)

* ODP-1451 LLAP MDC from NDC string class cast exception * ODP-1451 LLAP MDC from NDC string class cast exception --------- Co-authored-by: shubhamsharma <[email protected]>
acceldata-io · Jun 17, 2024 · 3c00ed1 · 3c00ed1
1 parent f8dfef9
commit 3c00ed1
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/llap-server/src/java/org/apache/hadoop/hive/llap/daemon/impl/TaskRunnerCallable.java b/llap-server/src/java/org/apache/hadoop/hive/llap/daemon/impl/TaskRunnerCallable.java
@@ -318,9 +318,18 @@ public LlapTaskUmbilicalProtocol run() throws Exception {
 
   private void setMDCFromNDC() {
     final Stack<String> clonedNDC = NDC.cloneStack();
-    final String fragId = clonedNDC.pop();
-    final String queryId = clonedNDC.pop();
-    final String dagId = clonedNDC.pop();
+    if(!clonedNDC.empty())
+      LOG.debug("Get the class name " + clonedNDC.peek().getClass());
+
+    final String fragId = clonedNDC.empty() ? "":clonedNDC.pop().toString();
+    LOG.debug("FragId " + fragId);
+
+    final String queryId = clonedNDC.empty() ? "":clonedNDC.pop().toString();
+    LOG.debug("queryId " + queryId);
+
+    final String dagId = clonedNDC.empty() ? "":clonedNDC.pop().toString();
+    LOG.debug("dagId " + dagId);
+
     MDC.put("dagId", dagId);
     MDC.put("queryId", queryId);
     MDC.put("fragmentId", fragId);