YARN-4886. Add HDFS caller context for EntityGroupFSTimelineStore. Contributed by Li Lu

(cherry picked from commit e0cb426758)
(cherry picked from commit 5bc64dafc3)
This commit is contained in:
Xuan 2016-04-13 10:38:17 -07:00
parent ccaf746eea
commit 3127b86492
1 changed files with 6 additions and 0 deletions

View File

@ -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();
}