@VisibleForTesting Token<?> getDelegationTokenFromHS(MRClientProtocol hsProxy) throws IOException, InterruptedException { GetDelegationTokenRequest request = recordFactory .newRecordInstance(GetDelegationTokenRequest.class); request.setRenewer(Master.getMasterPrincipal(conf)); org.apache.hadoop.yarn.api.records.Token mrDelegationToken; mrDelegationToken = hsProxy.getDelegationToken(request) .getDelegationToken(); return ConverterUtils.convertFromYarn(mrDelegationToken, hsProxy.getConnectAddress()); }
public KillTaskAttemptResponse forceKillTaskAttempt( KillTaskAttemptRequest request) throws YarnException, IOException { return protocolHandler.killTaskAttempt(request); }
@Override public GetCountersResponseProto getCounters(RpcController controller, GetCountersRequestProto proto) throws ServiceException { GetCountersRequest request = new GetCountersRequestPBImpl(proto); try { GetCountersResponse response = real.getCounters(request); return ((GetCountersResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
gcRequest.setJobId(job.getID()); Assert.assertNotNull("Counters is null", proxy.getCounters(gcRequest).getCounters()); JobReport jr = proxy.getJobReport(gjrRequest).getJobReport(); verifyJobReport(jr); gtaceRequest.setMaxEvents(10); Assert.assertNotNull("TaskCompletionEvents is null", proxy.getTaskAttemptCompletionEvents(gtaceRequest).getCompletionEventList()); gdRequest.setTaskAttemptId(attempt.getID()); Assert.assertNotNull("Diagnostics is null", proxy.getDiagnostics(gdRequest).getDiagnosticsList()); gtarRequest.setTaskAttemptId(attempt.getID()); TaskAttemptReport tar = proxy.getTaskAttemptReport(gtarRequest).getTaskAttemptReport(); verifyTaskAttemptReport(tar); gtrRequest.setTaskId(task.getID()); Assert.assertNotNull("TaskReport is null", proxy.getTaskReport(gtrRequest).getTaskReport()); gtreportsRequest.setTaskType(TaskType.MAP); Assert.assertNotNull("TaskReports for map is null", proxy.getTaskReports(gtreportsRequest).getTaskReportList()); gtreportsRequest.setTaskType(TaskType.REDUCE);
killJobRequest.setJobId(app.getJobId()); try { client.killJob(killJobRequest); fail("viewonlyuser killed job"); } catch (AccessControlException e) { killTaskRequest.setTaskId(task.getID()); try { client.killTask(killTaskRequest); fail("viewonlyuser killed task"); } catch (AccessControlException e) { killTaskAttemptRequest.setTaskAttemptId(attempt.getID()); try { client.killTaskAttempt(killTaskAttemptRequest); fail("viewonlyuser killed task attempt"); } catch (AccessControlException e) { failTaskAttemptRequest.setTaskAttemptId(attempt.getID()); try { client.failTaskAttempt(failTaskAttemptRequest); fail("viewonlyuser killed task attempt"); } catch (AccessControlException e) {
@Override public GetDelegationTokenResponseProto getDelegationToken( RpcController controller, GetDelegationTokenRequestProto proto) throws ServiceException { GetDelegationTokenRequest request = new GetDelegationTokenRequestPBImpl(proto); try { GetDelegationTokenResponse response = real.getDelegationToken(request); return ((GetDelegationTokenResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public GetJobReportResponseProto getJobReport(RpcController controller, GetJobReportRequestProto proto) throws ServiceException { GetJobReportRequestPBImpl request = new GetJobReportRequestPBImpl(proto); try { GetJobReportResponse response = real.getJobReport(request); return ((GetJobReportResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public CancelDelegationTokenResponseProto cancelDelegationToken( RpcController controller, CancelDelegationTokenRequestProto proto) throws ServiceException { CancelDelegationTokenRequestPBImpl request = new CancelDelegationTokenRequestPBImpl(proto); try { CancelDelegationTokenResponse response = real.cancelDelegationToken(request); return ((CancelDelegationTokenResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public GetDiagnosticsResponseProto getDiagnostics(RpcController controller, GetDiagnosticsRequestProto proto) throws ServiceException { GetDiagnosticsRequest request = new GetDiagnosticsRequestPBImpl(proto); try { GetDiagnosticsResponse response = real.getDiagnostics(request); return ((GetDiagnosticsResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public FailTaskAttemptResponseProto failTaskAttempt(RpcController controller, FailTaskAttemptRequestProto proto) throws ServiceException { FailTaskAttemptRequest request = new FailTaskAttemptRequestPBImpl(proto); try { FailTaskAttemptResponse response = real.failTaskAttempt(request); return ((FailTaskAttemptResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public GetTaskReportResponseProto getTaskReport(RpcController controller, GetTaskReportRequestProto proto) throws ServiceException { GetTaskReportRequest request = new GetTaskReportRequestPBImpl(proto); try { GetTaskReportResponse response = real.getTaskReport(request); return ((GetTaskReportResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public GetTaskAttemptReportResponseProto getTaskAttemptReport( RpcController controller, GetTaskAttemptReportRequestProto proto) throws ServiceException { GetTaskAttemptReportRequest request = new GetTaskAttemptReportRequestPBImpl(proto); try { GetTaskAttemptReportResponse response = real.getTaskAttemptReport(request); return ((GetTaskAttemptReportResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public GetTaskAttemptCompletionEventsResponseProto getTaskAttemptCompletionEvents( RpcController controller, GetTaskAttemptCompletionEventsRequestProto proto) throws ServiceException { GetTaskAttemptCompletionEventsRequest request = new GetTaskAttemptCompletionEventsRequestPBImpl(proto); try { GetTaskAttemptCompletionEventsResponse response = real.getTaskAttemptCompletionEvents(request); return ((GetTaskAttemptCompletionEventsResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public GetDelegationTokenResponseProto getDelegationToken( RpcController controller, GetDelegationTokenRequestProto proto) throws ServiceException { GetDelegationTokenRequest request = new GetDelegationTokenRequestPBImpl(proto); try { GetDelegationTokenResponse response = real.getDelegationToken(request); return ((GetDelegationTokenResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public GetJobReportResponseProto getJobReport(RpcController controller, GetJobReportRequestProto proto) throws ServiceException { GetJobReportRequestPBImpl request = new GetJobReportRequestPBImpl(proto); try { GetJobReportResponse response = real.getJobReport(request); return ((GetJobReportResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public CancelDelegationTokenResponseProto cancelDelegationToken( RpcController controller, CancelDelegationTokenRequestProto proto) throws ServiceException { CancelDelegationTokenRequestPBImpl request = new CancelDelegationTokenRequestPBImpl(proto); try { CancelDelegationTokenResponse response = real.cancelDelegationToken(request); return ((CancelDelegationTokenResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public GetDiagnosticsResponseProto getDiagnostics(RpcController controller, GetDiagnosticsRequestProto proto) throws ServiceException { GetDiagnosticsRequest request = new GetDiagnosticsRequestPBImpl(proto); try { GetDiagnosticsResponse response = real.getDiagnostics(request); return ((GetDiagnosticsResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public FailTaskAttemptResponseProto failTaskAttempt(RpcController controller, FailTaskAttemptRequestProto proto) throws ServiceException { FailTaskAttemptRequest request = new FailTaskAttemptRequestPBImpl(proto); try { FailTaskAttemptResponse response = real.failTaskAttempt(request); return ((FailTaskAttemptResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public GetTaskReportResponseProto getTaskReport(RpcController controller, GetTaskReportRequestProto proto) throws ServiceException { GetTaskReportRequest request = new GetTaskReportRequestPBImpl(proto); try { GetTaskReportResponse response = real.getTaskReport(request); return ((GetTaskReportResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }
@Override public GetTaskAttemptReportResponseProto getTaskAttemptReport( RpcController controller, GetTaskAttemptReportRequestProto proto) throws ServiceException { GetTaskAttemptReportRequest request = new GetTaskAttemptReportRequestPBImpl(proto); try { GetTaskAttemptReportResponse response = real.getTaskAttemptReport(request); return ((GetTaskAttemptReportResponsePBImpl)response).getProto(); } catch (IOException e) { throw new ServiceException(e); } }