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.fs.permission.FsPermission;
|
||||||
import org.apache.hadoop.service.CompositeService;
|
import org.apache.hadoop.service.CompositeService;
|
||||||
import org.apache.hadoop.service.ServiceOperations;
|
import org.apache.hadoop.service.ServiceOperations;
|
||||||
|
import org.apache.hadoop.ipc.CallerContext;
|
||||||
import org.apache.hadoop.util.ReflectionUtils;
|
import org.apache.hadoop.util.ReflectionUtils;
|
||||||
import org.apache.hadoop.util.Time;
|
import org.apache.hadoop.util.Time;
|
||||||
import org.apache.hadoop.yarn.api.records.ApplicationAttemptId;
|
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 SUMMARY_LOG_PREFIX = "summarylog-";
|
||||||
static final String ENTITY_LOG_PREFIX = "entitylog-";
|
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(
|
private static final Logger LOG = LoggerFactory.getLogger(
|
||||||
EntityGroupFSTimelineStore.class);
|
EntityGroupFSTimelineStore.class);
|
||||||
private static final FsPermission ACTIVE_DIR_PERMISSION =
|
private static final FsPermission ACTIVE_DIR_PERMISSION =
|
||||||
|
@ -187,6 +190,8 @@ public class EntityGroupFSTimelineStore extends CompositeService
|
||||||
YarnConfiguration
|
YarnConfiguration
|
||||||
.TIMELINE_SERVICE_ENTITYGROUP_FS_STORE_DONE_DIR_DEFAULT));
|
.TIMELINE_SERVICE_ENTITYGROUP_FS_STORE_DONE_DIR_DEFAULT));
|
||||||
fs = activeRootPath.getFileSystem(conf);
|
fs = activeRootPath.getFileSystem(conf);
|
||||||
|
CallerContext.setCurrent(
|
||||||
|
new CallerContext.Builder(ATS_V15_SERVER_DFS_CALLER_CTXT).build());
|
||||||
super.serviceInit(conf);
|
super.serviceInit(conf);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -304,6 +309,7 @@ public class EntityGroupFSTimelineStore extends CompositeService
|
||||||
ServiceOperations.stopQuietly(cacheItem.getStore());
|
ServiceOperations.stopQuietly(cacheItem.getStore());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
CallerContext.setCurrent(null);
|
||||||
super.serviceStop();
|
super.serviceStop();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue