@Override public EmptyMessageParameters getUnresolvedMessageParameters() { return EmptyMessageParameters.getInstance(); }
/** * Requests the job details. * * @param jobId The job id * @return Job details */ public CompletableFuture<JobDetailsInfo> getJobDetails(JobID jobId) { final JobDetailsHeaders detailsHeaders = JobDetailsHeaders.getInstance(); final JobMessageParameters params = new JobMessageParameters(); params.jobPathParameter.resolve(jobId); return sendRequest( detailsHeaders, params); }
@Override public void cancel(JobID jobID) throws Exception { JobTerminationMessageParameters params = new JobTerminationMessageParameters(); params.jobPathParameter.resolve(jobID); params.terminationModeQueryParameter.resolve(Collections.singletonList(TerminationModeQueryParameter.TerminationMode.CANCEL)); CompletableFuture<EmptyResponseBody> responseFuture = sendRequest( JobTerminationHeaders.getInstance(), params); responseFuture.get(timeout.toMillis(), TimeUnit.MILLISECONDS); }
@VisibleForTesting <M extends MessageHeaders<EmptyRequestBody, P, EmptyMessageParameters>, P extends ResponseBody> CompletableFuture<P> sendRequest(M messageHeaders) { return sendRequest(messageHeaders, EmptyMessageParameters.getInstance(), EmptyRequestBody.getInstance()); }
private CompletableFuture<SavepointInfo> pollSavepointAsync( final JobID jobId, final TriggerId triggerID) { return pollResourceAsync(() -> { final SavepointStatusHeaders savepointStatusHeaders = SavepointStatusHeaders.getInstance(); final SavepointStatusMessageParameters savepointStatusMessageParameters = savepointStatusHeaders.getUnresolvedMessageParameters(); savepointStatusMessageParameters.jobIdPathParameter.resolve(jobId); savepointStatusMessageParameters.triggerIdPathParameter.resolve(triggerID); return sendRequest( savepointStatusHeaders, savepointStatusMessageParameters); }); }
private TestJobTerminationHandler() { super(JobTerminationHeaders.getInstance()); }
public TestAccumulatorHandler() { super(JobAccumulatorsHeaders.getInstance()); }
private TestListJobsHandler() { super(JobsOverviewHeaders.getInstance()); }
public JarPlanHandler( final GatewayRetriever<? extends RestfulGateway> leaderRetriever, final Time timeout, final Map<String, String> responseHeaders, final MessageHeaders<JarPlanRequestBody, JobPlanInfo, JarPlanMessageParameters> messageHeaders, final Path jarDir, final Configuration configuration, final Executor executor) { this( leaderRetriever, timeout, responseHeaders, messageHeaders, jarDir, configuration, executor, jobGraph -> new JobPlanInfo(JsonPlanGenerator.generatePlan(jobGraph))); }
private static Map<String, String> getFlinkConfig(final String host, final int port) throws Exception { final ClusterConfigurationInfo clusterConfigurationInfoEntries = restClient.sendRequest( host, port, ClusterConfigurationInfoHeaders.getInstance()).get(); return clusterConfigurationInfoEntries.stream().collect(Collectors.toMap( ClusterConfigurationInfoEntry::getKey, ClusterConfigurationInfoEntry::getValue)); }
private <M extends MessageHeaders<EmptyRequestBody, P, U>, U extends MessageParameters, P extends ResponseBody> CompletableFuture<P> sendRequest(M messageHeaders, U messageParameters) { return sendRequest(messageHeaders, messageParameters, EmptyRequestBody.getInstance()); }
@Override JarPlanMessageParameters getUnresolvedJarMessageParameters() { return handler.getMessageHeaders().getUnresolvedMessageParameters(); }
private static HandlerRequest<EmptyRequestBody, EmptyMessageParameters> createRequest( final Path uploadedFile) throws HandlerRequestException, IOException { return new HandlerRequest<>( EmptyRequestBody.getInstance(), EmptyMessageParameters.getInstance(), Collections.emptyMap(), Collections.emptyMap(), Collections.singleton(uploadedFile.toFile())); } }
@Override public void stop(JobID jobID) throws Exception { JobTerminationMessageParameters params = new JobTerminationMessageParameters(); params.jobPathParameter.resolve(jobID); params.terminationModeQueryParameter.resolve(Collections.singletonList(TerminationModeQueryParameter.TerminationMode.STOP)); CompletableFuture<EmptyResponseBody> responseFuture = sendRequest( JobTerminationHeaders.getInstance(), params); responseFuture.get(timeout.toMillis(), TimeUnit.MILLISECONDS); }
@Override public EmptyMessageParameters getUnresolvedMessageParameters() { return EmptyMessageParameters.getInstance(); }
@Override JarRunMessageParameters getUnresolvedJarMessageParameters() { return handler.getMessageHeaders().getUnresolvedMessageParameters(); }
private static JarListInfo listJars(JarListHandler handler, RestfulGateway restfulGateway) throws Exception { HandlerRequest<EmptyRequestBody, EmptyMessageParameters> listRequest = new HandlerRequest<>( EmptyRequestBody.getInstance(), EmptyMessageParameters.getInstance()); return handler.handleRequest(listRequest, restfulGateway) .get(); }
@Override public EmptyMessageParameters getUnresolvedMessageParameters() { return EmptyMessageParameters.getInstance(); }
private static String uploadJar(JarUploadHandler handler, Path jar, RestfulGateway restfulGateway) throws Exception { HandlerRequest<EmptyRequestBody, EmptyMessageParameters> uploadRequest = new HandlerRequest<>( EmptyRequestBody.getInstance(), EmptyMessageParameters.getInstance(), Collections.emptyMap(), Collections.emptyMap(), Collections.singletonList(jar.toFile())); final JarUploadResponseBody uploadResponse = handler.handleRequest(uploadRequest, restfulGateway) .get(); return uploadResponse.getFilename(); }
private <M extends MessageHeaders<R, P, EmptyMessageParameters>, R extends RequestBody, P extends ResponseBody> CompletableFuture<P> sendRequest(M messageHeaders, R request) { return sendRequest(messageHeaders, EmptyMessageParameters.getInstance(), request); }