org.apache.beam.runners.reference.testing

Class TestJobService

java.lang.Object

org.apache.beam.model.jobmanagement.v1.JobServiceGrpc.JobServiceImplBase

org.apache.beam.runners.reference.testing.TestJobService

All Implemented Interfaces:

org.apache.beam.vendor.grpc.v1_13_1.io.grpc.BindableService

public class TestJobService
extends org.apache.beam.model.jobmanagement.v1.JobServiceGrpc.JobServiceImplBase

A JobService for tests.

A TestJobService always returns a fixed staging endpoint, job preparation id, job id, and job state. As soon as a job is run, it is put into the given job state.

Constructor Summary

Constructors

Constructor and Description
TestJobService(org.apache.beam.model.pipeline.v1.Endpoints.ApiServiceDescriptor stagingEndpoint, java.lang.String preparationId, java.lang.String jobId, org.apache.beam.model.jobmanagement.v1.JobApi.JobState.Enum jobState)

Method Summary

Modifier and Type	Method and Description
void	getState(org.apache.beam.model.jobmanagement.v1.JobApi.GetJobStateRequest request, org.apache.beam.vendor.grpc.v1_13_1.io.grpc.stub.StreamObserver<org.apache.beam.model.jobmanagement.v1.JobApi.GetJobStateResponse> responseObserver)
void	prepare(org.apache.beam.model.jobmanagement.v1.JobApi.PrepareJobRequest request, org.apache.beam.vendor.grpc.v1_13_1.io.grpc.stub.StreamObserver<org.apache.beam.model.jobmanagement.v1.JobApi.PrepareJobResponse> responseObserver)
void	run(org.apache.beam.model.jobmanagement.v1.JobApi.RunJobRequest request, org.apache.beam.vendor.grpc.v1_13_1.io.grpc.stub.StreamObserver<org.apache.beam.model.jobmanagement.v1.JobApi.RunJobResponse> responseObserver)

Methods inherited from class org.apache.beam.model.jobmanagement.v1.JobServiceGrpc.JobServiceImplBase

bindService, cancel, getMessageStream, getStateStream

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

TestJobService

public TestJobService(org.apache.beam.model.pipeline.v1.Endpoints.ApiServiceDescriptor stagingEndpoint,
                      java.lang.String preparationId,
                      java.lang.String jobId,
                      org.apache.beam.model.jobmanagement.v1.JobApi.JobState.Enum jobState)

Method Detail

prepare

public void prepare(org.apache.beam.model.jobmanagement.v1.JobApi.PrepareJobRequest request,
                    org.apache.beam.vendor.grpc.v1_13_1.io.grpc.stub.StreamObserver<org.apache.beam.model.jobmanagement.v1.JobApi.PrepareJobResponse> responseObserver)

Overrides:

prepare in class org.apache.beam.model.jobmanagement.v1.JobServiceGrpc.JobServiceImplBase

run

public void run(org.apache.beam.model.jobmanagement.v1.JobApi.RunJobRequest request,
                org.apache.beam.vendor.grpc.v1_13_1.io.grpc.stub.StreamObserver<org.apache.beam.model.jobmanagement.v1.JobApi.RunJobResponse> responseObserver)

Overrides:

run in class org.apache.beam.model.jobmanagement.v1.JobServiceGrpc.JobServiceImplBase

getState

public void getState(org.apache.beam.model.jobmanagement.v1.JobApi.GetJobStateRequest request,
                     org.apache.beam.vendor.grpc.v1_13_1.io.grpc.stub.StreamObserver<org.apache.beam.model.jobmanagement.v1.JobApi.GetJobStateResponse> responseObserver)

Overrides:

getState in class org.apache.beam.model.jobmanagement.v1.JobServiceGrpc.JobServiceImplBase