This is an automated email from the ASF dual-hosted git repository.
ayushsaxena pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/tez.git
The following commit(s) were added to refs/heads/master by this push:
new e13cab356 TEZ-4605: Enable hadoop CallerContext in TezChild and
DAGAppMaster (#394). (liang yu, Reviewed by Ayush Saxena)
e13cab356 is described below
commit e13cab3562fcebc8e45d8612fafbd84147569b08
Author: yu liang <[email protected]>
AuthorDate: Fri Feb 21 23:37:04 2025 +0800
TEZ-4605: Enable hadoop CallerContext in TezChild and DAGAppMaster (#394).
(liang yu, Reviewed by Ayush Saxena)
---
tez-dag/src/main/java/org/apache/tez/dag/app/DAGAppMaster.java | 4 +++-
.../src/main/java/org/apache/tez/runtime/task/TezChild.java | 2 ++
2 files changed, 5 insertions(+), 1 deletion(-)
diff --git a/tez-dag/src/main/java/org/apache/tez/dag/app/DAGAppMaster.java
b/tez-dag/src/main/java/org/apache/tez/dag/app/DAGAppMaster.java
index 4172a5a36..45b5266ff 100644
--- a/tez-dag/src/main/java/org/apache/tez/dag/app/DAGAppMaster.java
+++ b/tez-dag/src/main/java/org/apache/tez/dag/app/DAGAppMaster.java
@@ -2390,7 +2390,9 @@ public class DAGAppMaster extends AbstractService {
ContainerId containerId = ConverterUtils.toContainerId(containerIdStr);
ApplicationAttemptId applicationAttemptId =
containerId.getApplicationAttemptId();
-
+ org.apache.hadoop.ipc.CallerContext.setCurrent(new
org.apache.hadoop.ipc.CallerContext
+ .Builder("tez_appmaster_" + containerId.getApplicationAttemptId()
+ ).build());
long appSubmitTime = Long.parseLong(appSubmitTimeStr);
String jobUserName = System
diff --git
a/tez-runtime-internals/src/main/java/org/apache/tez/runtime/task/TezChild.java
b/tez-runtime-internals/src/main/java/org/apache/tez/runtime/task/TezChild.java
index ed14bd880..1e9ff6106 100644
---
a/tez-runtime-internals/src/main/java/org/apache/tez/runtime/task/TezChild.java
+++
b/tez-runtime-internals/src/main/java/org/apache/tez/runtime/task/TezChild.java
@@ -41,6 +41,7 @@ import javax.annotation.Nullable;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.ipc.RPC;
+import org.apache.hadoop.ipc.CallerContext;
import org.apache.hadoop.net.NetUtils;
import org.apache.hadoop.security.Credentials;
import org.apache.hadoop.security.SecurityUtil;
@@ -533,6 +534,7 @@ public class TezChild {
final int attemptNumber = Integer.parseInt(args[4]);
final String[] localDirs =
TezCommonUtils.getTrimmedStrings(System.getenv(Environment.LOCAL_DIRS
.name()));
+ CallerContext.setCurrent(new
CallerContext.Builder("tez_"+tokenIdentifier).build());
LOG.info("TezChild starting with PID=" + pid + ", containerIdentifier=" +
containerIdentifier);
if (LOG.isDebugEnabled()) {
LOG.debug("Info from cmd line: AM-host: " + host + " AM-port: " + port
