Python set_current_spanの例

コード例 #1

def app(event_loop, request):
    os.environ.pop('F_DATADIR', None)
    os.environ.pop('FAUST_DATADIR', None)
    os.environ.pop('F_WORKDIR', None)
    os.environ.pop('FAUST_WORKDIR', None)
    marks = request.node.get_closest_marker('app')
    options = AppMarks(
        **{
            **{
                'name': 'funtest',
                'store': 'memory://',
                'cache': 'memory://'
            },
            **((marks.kwargs or {}) if marks else {}),
        })
    app = faust.App(
        options.name,
        store=options.store,
        cache=options.cache,
    )
    app.finalize()
    set_current_span(None)
    try:
        yield app
    finally:
        assert app.tracer is None

コード例 #2

ファイル: aiokafka.py プロジェクト: swist/faust

 def _start_span(self, name: str) -> opentracing.Span:
     tracer = self.app.tracer
     if tracer is not None:
         span = tracer.get_tracer('_aiokafka').start_span(
             operation_name=name, )
         span.set_tag(tags.SAMPLING_PRIORITY, 1)
         set_current_span(span)
         return span
     else:
         return noop_span()

コード例 #3

ファイル: aiokafka.py プロジェクト: jbooth-mastery/faust

 def _start_span(self, name: str, *,
                 lazy: bool = False) -> opentracing.Span:
     tracer = self.app.tracer
     if tracer is not None:
         span = tracer.get_tracer(self.trace_category).start_span(
             operation_name=name,
         )
         span.set_tag(tags.SAMPLING_PRIORITY, 1)
         self.app._span_add_default_tags(span)
         set_current_span(span)
         if lazy:
             self._transform_span_lazy(span)
         return span
     else:
         return noop_span()

コード例 #4

ファイル: conftest.py プロジェクト: voidp34r/faust

def app(event_loop):
    os.environ.pop('F_DATADIR', None)
    os.environ.pop('FAUST_DATADIR', None)
    os.environ.pop('F_WORKDIR', None)
    os.environ.pop('FAUST_WORKDIR', None)
    instance = faust.App('testid')
    instance.producer = Mock(
        name='producer',
        autospec=Producer,
        maybe_start=AsyncMock(),
        start=AsyncMock(),
        send=AsyncMock(),
        send_and_wait=AsyncMock(),
    )
    instance.finalize()
    set_current_span(None)
    return instance

コード例 #5

ファイル: conftest.py プロジェクト: wayfair-contribs/faust

def app(event_loop, request):
    os.environ.pop("F_DATADIR", None)
    os.environ.pop("FAUST_DATADIR", None)
    os.environ.pop("F_WORKDIR", None)
    os.environ.pop("FAUST_WORKDIR", None)
    marks = request.node.get_closest_marker("app")
    options, rest = create_appmarks(**((marks.kwargs or {}) if marks else {}))
    app = faust.App(
        options.name,
        store=options.store,
        cache=options.cache,
        **rest,
    )
    app.finalize()
    set_current_span(None)
    try:
        yield app
    finally:
        assert app.tracer is None

コード例 #6

ファイル: conftest.py プロジェクト: marcosschroh/faust

def app(event_loop, request):
    settings = request.node.get_closest_marker('conf')
    kwargs = settings.kwargs or {} if settings else {}

    os.environ.pop('F_DATADIR', None)
    os.environ.pop('FAUST_DATADIR', None)
    os.environ.pop('F_WORKDIR', None)
    os.environ.pop('FAUST_WORKDIR', None)
    instance = faust.App('testid', **kwargs)
    instance.producer = Mock(
        name='producer',
        autospec=Producer,
        maybe_start=AsyncMock(),
        start=AsyncMock(),
        send=AsyncMock(),
        send_and_wait=AsyncMock(),
    )
    instance.finalize()
    set_current_span(None)
    return instance