# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
import grpc

from google.cloud.trace_v2.proto import trace_pb2 as google_dot_devtools_dot_cloudtrace__v2_dot_proto_dot_trace__pb2
from google.cloud.trace_v2.proto import tracing_pb2 as google_dot_devtools_dot_cloudtrace__v2_dot_proto_dot_tracing__pb2
from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2


class TraceServiceStub(object):
  """This file describes an API for collecting and viewing traces and spans
  within a trace.  A Trace is a collection of spans corresponding to a single
  operation or set of operations for an application. A span is an individual
  timed event which forms a node of the trace tree. A single trace may
  contain span(s) from multiple services.
  """

  def __init__(self, channel):
    """Constructor.

    Args:
      channel: A grpc.Channel.
    """
    self.BatchWriteSpans = channel.unary_unary(
        '/google.devtools.cloudtrace.v2.TraceService/BatchWriteSpans',
        request_serializer=google_dot_devtools_dot_cloudtrace__v2_dot_proto_dot_tracing__pb2.BatchWriteSpansRequest.SerializeToString,
        response_deserializer=google_dot_protobuf_dot_empty__pb2.Empty.FromString,
        )
    self.CreateSpan = channel.unary_unary(
        '/google.devtools.cloudtrace.v2.TraceService/CreateSpan',
        request_serializer=google_dot_devtools_dot_cloudtrace__v2_dot_proto_dot_trace__pb2.Span.SerializeToString,
        response_deserializer=google_dot_devtools_dot_cloudtrace__v2_dot_proto_dot_trace__pb2.Span.FromString,
        )


class TraceServiceServicer(object):
  """This file describes an API for collecting and viewing traces and spans
  within a trace.  A Trace is a collection of spans corresponding to a single
  operation or set of operations for an application. A span is an individual
  timed event which forms a node of the trace tree. A single trace may
  contain span(s) from multiple services.
  """

  def BatchWriteSpans(self, request, context):
    """Sends new spans to new or existing traces. You cannot update
    existing spans.
    """
    context.set_code(grpc.StatusCode.UNIMPLEMENTED)
    context.set_details('Method not implemented!')
    raise NotImplementedError('Method not implemented!')

  def CreateSpan(self, request, context):
    """Creates a new span.
    """
    context.set_code(grpc.StatusCode.UNIMPLEMENTED)
    context.set_details('Method not implemented!')
    raise NotImplementedError('Method not implemented!')


def add_TraceServiceServicer_to_server(servicer, server):
  rpc_method_handlers = {
      'BatchWriteSpans': grpc.unary_unary_rpc_method_handler(
          servicer.BatchWriteSpans,
          request_deserializer=google_dot_devtools_dot_cloudtrace__v2_dot_proto_dot_tracing__pb2.BatchWriteSpansRequest.FromString,
          response_serializer=google_dot_protobuf_dot_empty__pb2.Empty.SerializeToString,
      ),
      'CreateSpan': grpc.unary_unary_rpc_method_handler(
          servicer.CreateSpan,
          request_deserializer=google_dot_devtools_dot_cloudtrace__v2_dot_proto_dot_trace__pb2.Span.FromString,
          response_serializer=google_dot_devtools_dot_cloudtrace__v2_dot_proto_dot_trace__pb2.Span.SerializeToString,
      ),
  }
  generic_handler = grpc.method_handlers_generic_handler(
      'google.devtools.cloudtrace.v2.TraceService', rpc_method_handlers)
  server.add_generic_rpc_handlers((generic_handler,))
