if (header.hasCallerContext()) { callerContext = new CallerContext.Builder(header.getCallerContext().getContext()) .setSignature(header.getCallerContext().getSignature() .toByteArray()) .build();
new CallerContext.Builder("mr_" + firstTaskid.toString()).build());
new CallerContext.Builder("mr_" + firstTaskid.toString()).build());
if (header.hasCallerContext()) { callerContext = new CallerContext.Builder(header.getCallerContext().getContext()) .setSignature(header.getCallerContext().getSignature() .toByteArray()) .build();
fs = activeRootPath.getFileSystem(conf); CallerContext.setCurrent( new CallerContext.Builder(ATS_V15_SERVER_DFS_CALLER_CTXT).build()); super.serviceInit(conf);
containerId.getApplicationAttemptId(); if (applicationAttemptId != null) { CallerContext.setCurrent(new CallerContext.Builder( "mr_appmaster_" + applicationAttemptId.toString()).build());
when(app.getApplicationPriority()).thenReturn(Priority.newInstance(10)); when(app.getCallerContext()) .thenReturn(new CallerContext.Builder("context").build()); when(app.getState()).thenReturn(RMAppState.SUBMITTED); return app;
containerId.getApplicationAttemptId(); if (applicationAttemptId != null) { CallerContext.setCurrent(new CallerContext.Builder( "mr_appmaster_" + applicationAttemptId.toString()).build());
new CallerContext.Builder(ATS_V15_SERVER_DFS_CALLER_CTXT).build()); super.serviceInit(conf);
/** * Test the AuditLog format for successful events with the various * parameters. */ private void testSuccessLogFormat(boolean checkIP) { testSuccessLogFormatHelper(checkIP, null, null, null); testSuccessLogFormatHelper(checkIP, APPID, null, null); testSuccessLogFormatHelper(checkIP, null, null, CONTAINERID); testSuccessLogFormatHelper(checkIP, null, ATTEMPTID, null); testSuccessLogFormatHelper(checkIP, APPID, ATTEMPTID, null); testSuccessLogFormatHelper(checkIP, APPID, null, CONTAINERID); testSuccessLogFormatHelper(checkIP, null, ATTEMPTID, CONTAINERID); testSuccessLogFormatHelper(checkIP, APPID, ATTEMPTID, CONTAINERID); testSuccessLogFormatHelper(checkIP, APPID, ATTEMPTID, CONTAINERID, null, null); testSuccessLogFormatHelper(checkIP, APPID, ATTEMPTID, CONTAINERID, new CallerContext.Builder(null).setSignature(null).build(), RESOURCE); testSuccessLogFormatHelper(checkIP, APPID, ATTEMPTID, CONTAINERID, new CallerContext.Builder(CALLER_CONTEXT).setSignature(null).build(), RESOURCE); testSuccessLogFormatHelper(checkIP, APPID, ATTEMPTID, CONTAINERID, new CallerContext.Builder(null).setSignature(CALLER_SIGNATURE).build(), RESOURCE); testSuccessLogFormatHelper(checkIP, APPID, ATTEMPTID, CONTAINERID, new CallerContext.Builder(CALLER_CONTEXT).setSignature(CALLER_SIGNATURE) .build(), RESOURCE); RMAuditLogger.ArgsBuilder args = new RMAuditLogger.ArgsBuilder() .append(Keys.QUEUENAME, QUEUE).append(Keys.RECURSIVE, "true"); testSuccessLogFormatHelper(checkIP, null, null, null, null, null, Server.getRemoteIp(), args); testSuccessLogFormatHelperWithIP(checkIP, APPID, ATTEMPTID, CONTAINERID); testSuccessLogNulls(checkIP); }
/** * Test the AuditLog format for failure events with the various * parameters. */ private void testFailureLogFormat(boolean checkIP) { testFailureLogFormatHelper(checkIP, null, null, null); testFailureLogFormatHelper(checkIP, APPID, null, null); testFailureLogFormatHelper(checkIP, null, null, CONTAINERID); testFailureLogFormatHelper(checkIP, null, ATTEMPTID, null); testFailureLogFormatHelper(checkIP, APPID, ATTEMPTID, null); testFailureLogFormatHelper(checkIP, APPID, null, CONTAINERID); testFailureLogFormatHelper(checkIP, null, ATTEMPTID, CONTAINERID); testFailureLogFormatHelper(checkIP, APPID, ATTEMPTID, CONTAINERID); testFailureLogFormatHelper(checkIP, APPID, ATTEMPTID, CONTAINERID, new CallerContext.Builder(null).setSignature(null).build(), RESOURCE); testFailureLogFormatHelper(checkIP, APPID, ATTEMPTID, CONTAINERID, new CallerContext.Builder(CALLER_CONTEXT).setSignature(null).build(), RESOURCE); testFailureLogFormatHelper(checkIP, APPID, ATTEMPTID, CONTAINERID, new CallerContext.Builder(null).setSignature(CALLER_SIGNATURE).build(), RESOURCE); testFailureLogFormatHelper(checkIP, APPID, ATTEMPTID, CONTAINERID, new CallerContext.Builder(CALLER_CONTEXT).setSignature(CALLER_SIGNATURE) .build(), RESOURCE); RMAuditLogger.ArgsBuilder args = new RMAuditLogger.ArgsBuilder() .append(Keys.QUEUENAME, QUEUE).append(Keys.RECURSIVE, "true"); testFailureLogFormatHelper(checkIP, null, null, null, null, null, args); }
throws Exception{ if (CallerContext.getCurrent() == null) { CallerContext ctx = new CallerContext.Builder("CLI").build(); CallerContext.setCurrent(ctx);
protected RMApp storeApp(RMStateStore store, ApplicationId appId, long submitTime, long startTime) throws Exception { ApplicationSubmissionContext context = new ApplicationSubmissionContextPBImpl(); context.setApplicationId(appId); context.setAMContainerSpec(new ContainerLaunchContextPBImpl()); RMApp mockApp = mock(RMApp.class); when(mockApp.getApplicationId()).thenReturn(appId); when(mockApp.getSubmitTime()).thenReturn(submitTime); when(mockApp.getStartTime()).thenReturn(startTime); when(mockApp.getApplicationSubmissionContext()).thenReturn(context); when(mockApp.getUser()).thenReturn("test"); when(mockApp.getCallerContext()) .thenReturn(new CallerContext.Builder("context").build()); store.storeNewApplication(mockApp); return mockApp; }
@Override public CallerContext getCallerContext() { ApplicationStateDataProtoOrBuilder p = viaProto ? proto : builder; RpcHeaderProtos.RPCCallerContextProto pbContext = p.getCallerContext(); if (pbContext != null) { CallerContext context = new CallerContext.Builder(pbContext.getContext()) .setSignature(pbContext.getSignature().toByteArray()).build(); return context; } return null; }
throws Exception{ if (CallerContext.getCurrent() == null) { CallerContext ctx = new CallerContext.Builder("CLI").build(); CallerContext.setCurrent(ctx);
@Override public void setHadoopCallerContext(String callerContext) { CallerContext.setCurrent(new CallerContext.Builder(callerContext).build()); }