# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
import grpc
import beam_job_api_pb2 as beam__job__api__pb2
[docs]class JobServiceStub(object):
"""Job Service for running RunnerAPI pipelines
"""
def __init__(self, channel):
"""Constructor.
Args:
channel: A grpc.Channel.
"""
self.Prepare = channel.unary_unary(
'/org.apache.beam.model.job_management.v1.JobService/Prepare',
request_serializer=beam__job__api__pb2.PrepareJobRequest.SerializeToString,
response_deserializer=beam__job__api__pb2.PrepareJobResponse.FromString,
)
self.Run = channel.unary_unary(
'/org.apache.beam.model.job_management.v1.JobService/Run',
request_serializer=beam__job__api__pb2.RunJobRequest.SerializeToString,
response_deserializer=beam__job__api__pb2.RunJobResponse.FromString,
)
self.GetState = channel.unary_unary(
'/org.apache.beam.model.job_management.v1.JobService/GetState',
request_serializer=beam__job__api__pb2.GetJobStateRequest.SerializeToString,
response_deserializer=beam__job__api__pb2.GetJobStateResponse.FromString,
)
self.Cancel = channel.unary_unary(
'/org.apache.beam.model.job_management.v1.JobService/Cancel',
request_serializer=beam__job__api__pb2.CancelJobRequest.SerializeToString,
response_deserializer=beam__job__api__pb2.CancelJobResponse.FromString,
)
self.GetStateStream = channel.unary_stream(
'/org.apache.beam.model.job_management.v1.JobService/GetStateStream',
request_serializer=beam__job__api__pb2.GetJobStateRequest.SerializeToString,
response_deserializer=beam__job__api__pb2.GetJobStateResponse.FromString,
)
self.GetMessageStream = channel.unary_stream(
'/org.apache.beam.model.job_management.v1.JobService/GetMessageStream',
request_serializer=beam__job__api__pb2.JobMessagesRequest.SerializeToString,
response_deserializer=beam__job__api__pb2.JobMessagesResponse.FromString,
)
[docs]class JobServiceServicer(object):
"""Job Service for running RunnerAPI pipelines
"""
[docs] def Prepare(self, request, context):
"""Prepare a job for execution. The job will not be executed until a call is made to run with the
returned preparationId.
"""
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')
[docs] def Run(self, request, context):
"""Submit the job for execution
"""
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')
[docs] def GetState(self, request, context):
"""Get the current state of the job
"""
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')
[docs] def Cancel(self, request, context):
"""Cancel the job
"""
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')
[docs] def GetStateStream(self, request, context):
"""Subscribe to a stream of state changes of the job, will immediately return the current state of the job as the first response.
"""
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')
[docs] def GetMessageStream(self, request, context):
"""Subscribe to a stream of state changes and messages from the job
"""
context.set_code(grpc.StatusCode.UNIMPLEMENTED)
context.set_details('Method not implemented!')
raise NotImplementedError('Method not implemented!')
[docs]def add_JobServiceServicer_to_server(servicer, server):
rpc_method_handlers = {
'Prepare': grpc.unary_unary_rpc_method_handler(
servicer.Prepare,
request_deserializer=beam__job__api__pb2.PrepareJobRequest.FromString,
response_serializer=beam__job__api__pb2.PrepareJobResponse.SerializeToString,
),
'Run': grpc.unary_unary_rpc_method_handler(
servicer.Run,
request_deserializer=beam__job__api__pb2.RunJobRequest.FromString,
response_serializer=beam__job__api__pb2.RunJobResponse.SerializeToString,
),
'GetState': grpc.unary_unary_rpc_method_handler(
servicer.GetState,
request_deserializer=beam__job__api__pb2.GetJobStateRequest.FromString,
response_serializer=beam__job__api__pb2.GetJobStateResponse.SerializeToString,
),
'Cancel': grpc.unary_unary_rpc_method_handler(
servicer.Cancel,
request_deserializer=beam__job__api__pb2.CancelJobRequest.FromString,
response_serializer=beam__job__api__pb2.CancelJobResponse.SerializeToString,
),
'GetStateStream': grpc.unary_stream_rpc_method_handler(
servicer.GetStateStream,
request_deserializer=beam__job__api__pb2.GetJobStateRequest.FromString,
response_serializer=beam__job__api__pb2.GetJobStateResponse.SerializeToString,
),
'GetMessageStream': grpc.unary_stream_rpc_method_handler(
servicer.GetMessageStream,
request_deserializer=beam__job__api__pb2.JobMessagesRequest.FromString,
response_serializer=beam__job__api__pb2.JobMessagesResponse.SerializeToString,
),
}
generic_handler = grpc.method_handlers_generic_handler(
'org.apache.beam.model.job_management.v1.JobService', rpc_method_handlers)
server.add_generic_rpc_handlers((generic_handler,))