Skip to content

Commit a411e8a

Browse files
committed
refactor: 修复测试用例没有traceprovider的问题 --story=126421543
# Reviewed, transaction id: 53415
1 parent 8118a42 commit a411e8a

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

gcloud/core/trace.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from functools import wraps
1717

1818
from opentelemetry import trace
19+
from opentelemetry.sdk.trace import TracerProvider
1920
from opentelemetry.sdk.trace.export import SpanProcessor
2021
from opentelemetry.trace import SpanKind
2122

@@ -54,9 +55,14 @@ def propagate_attributes(attributes: dict):
5455

5556
provider = trace.get_tracer_provider()
5657

58+
if not provider:
59+
provider = TracerProvider()
60+
5761
# Add a span processor that sets attributes on every new span
5862
provider.add_span_processor(AttributeInjectionSpanProcessor(attributes))
5963

64+
trace.set_tracer_provider(provider)
65+
6066

6167
@contextmanager
6268
def start_trace(span_name: str, propagate: bool = False, **attributes):

0 commit comments

Comments
 (0)