예제 #1
0
 def __init__(self, handler):
   TProcessor.__init__(self)
   self._handler = handler
   self._processMap = {}
   self._priorityMap = {}
   self._processMap["simple_rpc"] = ContextProcessor.process_simple_rpc
   self._priorityMap["simple_rpc"] = TPriority.NORMAL
예제 #2
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._processMap["query"] = ContextProcessor.process_query
     self._processMap[
         "has_arg_docs"] = ContextProcessor.process_has_arg_docs
예제 #3
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._priorityMap = {}
     self._processMap["init"] = Processor.process_init
     self._priorityMap["init"] = TPriority.NORMAL
예제 #4
0
 def __init__(self, handler):
   TProcessor.__init__(self)
   self._handler = handler
   self._processMap = {}
   self._priorityMap = {}
   self._processMap["init"] = ContextProcessor.process_init
   self._priorityMap["init"] = TPriority.NORMAL
예제 #5
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._priorityMap = {}
     self._processMap["bar"] = ContextProcessor.process_bar
     self._priorityMap["bar"] = TPriority.NORMAL
예제 #6
0
 def __init__(self, handler, executor=None):
   TProcessor.__init__(self)
   self._handler = handler
   self._executor = executor or ThreadPoolExecutor(max_workers=32)
   self._processMap = {}
   self._processMap["sleep"] = Processor.future_process_sleep
   self._processMap["isPrime"] = Processor.future_process_isPrime
   self._processMap["getResult"] = Processor.future_process_getResult
예제 #7
0
 def __init__(self, handler, executor=None):
   TProcessor.__init__(self)
   self._handler = handler
   self._executor = executor or ThreadPoolExecutor(max_workers=32)
   self._processMap = {}
   self._processMap["sleep"] = Processor.future_process_sleep
   self._processMap["isPrime"] = Processor.future_process_isPrime
   self._processMap["getResult"] = Processor.future_process_getResult
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._priorityMap = {}
     self._processMap["global"] = ContextProcessor.process_global
     self._priorityMap["global"] = TPriority.NORMAL
     self._processMap["import"] = ContextProcessor.process_import
     self._priorityMap["import"] = TPriority.NORMAL
예제 #9
0
 def __init__(self, handler):
   TProcessor.__init__(self)
   self._handler = handler
   self._processMap = {}
   self._priorityMap = {}
   self._processMap["query"] = Processor.process_query
   self._priorityMap["query"] = TPriority.NORMAL
   self._processMap["has_arg_docs"] = Processor.process_has_arg_docs
   self._priorityMap["has_arg_docs"] = TPriority.NORMAL
예제 #10
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._priorityMap = {}
     self._processMap["getDataByKey0"] = Processor.process_getDataByKey0
     self._priorityMap["getDataByKey0"] = TPriority.NORMAL
     self._processMap["getDataByKey1"] = Processor.process_getDataByKey1
     self._priorityMap["getDataByKey1"] = TPriority.NORMAL
예제 #11
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._priorityMap = {}
     self._processMap["ping"] = ContextProcessor.process_ping
     self._priorityMap["ping"] = TPriority.NORMAL
     self._processMap["echo"] = ContextProcessor.process_echo
     self._priorityMap["echo"] = TPriority.NORMAL
예제 #12
0
 def __init__(self, handler):
   TProcessor.__init__(self)
   self._handler = handler
   self._processMap = {}
   self._priorityMap = {}
   self._processMap["ping"] = ContextProcessor.process_ping
   self._priorityMap["ping"] = TPriority.IMPORTANT
   self._processMap["pong"] = ContextProcessor.process_pong
   self._priorityMap["pong"] = TPriority.HIGH_IMPORTANT
예제 #13
0
 def __init__(self, handler):
   TProcessor.__init__(self)
   self._handler = handler
   self._processMap = {}
   self._priorityMap = {}
   self._processMap["ping"] = ContextProcessor.process_ping
   self._priorityMap["ping"] = TPriority.IMPORTANT
   self._processMap["pong"] = ContextProcessor.process_pong
   self._priorityMap["pong"] = TPriority.HIGH_IMPORTANT
예제 #14
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._priorityMap = {}
     self._processMap["bounce_map"] = ContextProcessor.process_bounce_map
     self._priorityMap["bounce_map"] = TPriority.NORMAL
     self._processMap[
         "binary_keyed_map"] = ContextProcessor.process_binary_keyed_map
     self._priorityMap["binary_keyed_map"] = TPriority.NORMAL
예제 #15
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._priorityMap = {}
     self._processMap["foo"] = Processor.process_foo
     self._priorityMap["foo"] = TPriority.NORMAL
     self._processMap["interact"] = Processor.process_interact
     self._priorityMap["interact"] = TPriority.NORMAL
     self._processMap["interactFast"] = Processor.process_interactFast
     self._priorityMap["interactFast"] = TPriority.NORMAL
예제 #16
0
 def __init__(self, handler):
   TProcessor.__init__(self)
   self._handler = handler
   self._processMap = {}
   self._priorityMap = {}
   self._processMap["authenticate"] = ContextProcessor.process_authenticate
   self._priorityMap["authenticate"] = TPriority.NORMAL
   self._processMap["signout"] = ContextProcessor.process_signout
   self._priorityMap["signout"] = TPriority.NORMAL
   self._processMap["execute"] = ContextProcessor.process_execute
   self._priorityMap["execute"] = TPriority.NORMAL
예제 #17
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._priorityMap = {}
     self._processMap["get"] = ContextProcessor.process_get
     self._priorityMap["get"] = TPriority.NORMAL
     self._processMap["put"] = ContextProcessor.process_put
     self._priorityMap["put"] = TPriority.NORMAL
     self._processMap["remove"] = ContextProcessor.process_remove
     self._priorityMap["remove"] = TPriority.NORMAL
예제 #18
0
 def __init__(self, handler, loop=None):
     TProcessor.__init__(self)
     self._handler = handler
     self._loop = loop or asyncio.get_event_loop()
     self._processMap = {}
     self._priorityMap = {}
     self._processMap["authenticate"] = Processor.process_authenticate
     self._priorityMap["authenticate"] = TPriority.NORMAL
     self._processMap["signout"] = Processor.process_signout
     self._priorityMap["signout"] = TPriority.NORMAL
     self._processMap["execute"] = Processor.process_execute
     self._priorityMap["execute"] = TPriority.NORMAL
예제 #19
0
파일: Raiser.py 프로젝트: facebook/fbthrift
 def __init__(self, handler):
   TProcessor.__init__(self)
   self._handler = handler
   self._processMap = {}
   self._priorityMap = {}
   self._processMap["doBland"] = Processor.process_doBland
   self._priorityMap["doBland"] = TPriority.NORMAL
   self._processMap["doRaise"] = Processor.process_doRaise
   self._priorityMap["doRaise"] = TPriority.NORMAL
   self._processMap["get200"] = Processor.process_get200
   self._priorityMap["get200"] = TPriority.NORMAL
   self._processMap["get500"] = Processor.process_get500
   self._priorityMap["get500"] = TPriority.NORMAL
예제 #20
0
파일: Raiser.py 프로젝트: ktprime/fbthrift
 def __init__(self, handler):
   TProcessor.__init__(self)
   self._handler = handler
   self._processMap = {}
   self._priorityMap = {}
   self._processMap["doBland"] = ContextProcessor.process_doBland
   self._priorityMap["doBland"] = TPriority.NORMAL
   self._processMap["doRaise"] = ContextProcessor.process_doRaise
   self._priorityMap["doRaise"] = TPriority.NORMAL
   self._processMap["get200"] = ContextProcessor.process_get200
   self._priorityMap["get200"] = TPriority.NORMAL
   self._processMap["get500"] = ContextProcessor.process_get500
   self._priorityMap["get500"] = TPriority.NORMAL
예제 #21
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._priorityMap = {}
예제 #22
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._processMap["init"] = ContextProcessor.process_init
예제 #23
0
 def __init__(self, handler):
   TProcessor.__init__(self)
   self._handler = handler
   self._processMap = {}
   self._priorityMap = {}
예제 #24
0
def baseplateify_processor(
    processor: TProcessor,
    logger: Logger,
    baseplate: Baseplate,
    edge_context_factory: Optional[EdgeContextFactory] = None,
) -> TProcessor:
    """Wrap a Thrift Processor with Baseplate's span lifecycle.

    :param processor: The service's processor to wrap.
    :param logger: The logger to use for error and debug logging.
    :param baseplate: The baseplate instance for your application.
    :param edge_context_factory: A configured factory for handling edge request
        context.

    """
    def make_processor_fn(
            fn_name: str, processor_fn: Callable[...,
                                                 Any]) -> Callable[..., Any]:
        def call_processor_with_span_context(self: Any, seqid: int,
                                             iprot: TProtocolBase,
                                             oprot: TProtocolBase) -> Any:
            context = baseplate.make_context_object()

            # Allow case-insensitivity for THeader headers
            headers: Mapping[bytes,
                             bytes] = CaseInsensitiveDict(  # type: ignore
                                 data=iprot.get_headers())

            trace_info: Optional[TraceInfo]
            try:
                sampled = bool(headers.get(b"Sampled") == b"1")
                flags = headers.get(b"Flags", None)
                trace_info = TraceInfo.from_upstream(
                    int(headers[b"Trace"]),
                    int(headers[b"Parent"]),
                    int(headers[b"Span"]),
                    sampled,
                    int(flags) if flags is not None else None,
                )
            except (KeyError, ValueError):
                trace_info = None

            edge_payload = headers.get(b"Edge-Request", None)
            context.raw_edge_context = edge_payload
            if edge_context_factory:
                context.edge_context = edge_context_factory.from_upstream(
                    edge_payload)

            span = baseplate.make_server_span(context,
                                              name=fn_name,
                                              trace_info=trace_info)

            try:
                service_name = headers[b"User-Agent"].decode()
            except (KeyError, UnicodeDecodeError):
                pass
            else:
                span.set_tag("peer.service", service_name)

            context.headers = headers

            handler = processor._handler
            context_aware_handler = _ContextAwareHandler(
                handler, context, logger)
            context_aware_processor = processor.__class__(
                context_aware_handler)
            return processor_fn(context_aware_processor, seqid, iprot, oprot)

        return call_processor_with_span_context

    instrumented_process_map = {}
    for fn_name, processor_fn in processor._processMap.items():
        context_aware_processor_fn = make_processor_fn(fn_name, processor_fn)
        instrumented_process_map[fn_name] = context_aware_processor_fn
    processor._processMap = instrumented_process_map
    processor.baseplate = baseplate
    return processor
예제 #25
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._processMap["add"] = Processor.process_add
예제 #26
0
 def __init__(self, handler):
     TProcessor.__init__(self)
     self._handler = handler
     self._processMap = {}
     self._processMap["init"] = Processor.process_init