YARN-4886. Add HDFS caller context for EntityGroupFSTimelineStore. Contributed by Li Lu
(cherry picked from commit e0cb426758
)
This commit is contained in:
parent
4a7e52d302
commit
5bc64dafc3
|
@ -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();
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue