From e0cb426758b3d716ff143f723fc16ef2f1e4971b Mon Sep 17 00:00:00 2001 From: Xuan Date: Wed, 13 Apr 2016 10:38:17 -0700 Subject: [PATCH] YARN-4886. Add HDFS caller context for EntityGroupFSTimelineStore. Contributed by Li Lu --- .../yarn/server/timeline/EntityGroupFSTimelineStore.java | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-timeline-pluginstorage/src/main/java/org/apache/hadoop/yarn/server/timeline/EntityGroupFSTimelineStore.java b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-timeline-pluginstorage/src/main/java/org/apache/hadoop/yarn/server/timeline/EntityGroupFSTimelineStore.java index 34a20720bc5..18b8951aa0e 100644 --- a/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-timeline-pluginstorage/src/main/java/org/apache/hadoop/yarn/server/timeline/EntityGroupFSTimelineStore.java +++ b/hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-timeline-pluginstorage/src/main/java/org/apache/hadoop/yarn/server/timeline/EntityGroupFSTimelineStore.java @@ -28,6 +28,7 @@ import org.apache.hadoop.fs.RemoteIterator; import org.apache.hadoop.fs.permission.FsPermission; import org.apache.hadoop.service.CompositeService; import org.apache.hadoop.service.ServiceOperations; +import org.apache.hadoop.ipc.CallerContext; import org.apache.hadoop.util.ReflectionUtils; import org.apache.hadoop.util.Time; import org.apache.hadoop.yarn.api.records.ApplicationAttemptId; @@ -86,6 +87,8 @@ public class EntityGroupFSTimelineStore extends CompositeService static final String SUMMARY_LOG_PREFIX = "summarylog-"; static final String ENTITY_LOG_PREFIX = "entitylog-"; + static final String ATS_V15_SERVER_DFS_CALLER_CTXT = "yarn_ats_server_v1_5"; + private static final Logger LOG = LoggerFactory.getLogger( EntityGroupFSTimelineStore.class); private static final FsPermission ACTIVE_DIR_PERMISSION = @@ -187,6 +190,8 @@ public class EntityGroupFSTimelineStore extends CompositeService YarnConfiguration .TIMELINE_SERVICE_ENTITYGROUP_FS_STORE_DONE_DIR_DEFAULT)); fs = activeRootPath.getFileSystem(conf); + CallerContext.setCurrent( + new CallerContext.Builder(ATS_V15_SERVER_DFS_CALLER_CTXT).build()); super.serviceInit(conf); } @@ -304,6 +309,7 @@ public class EntityGroupFSTimelineStore extends CompositeService ServiceOperations.stopQuietly(cacheItem.getStore()); } } + CallerContext.setCurrent(null); super.serviceStop(); }