Skip to content

Commit bdb1b8d

Browse files
authored
Merge pull request DataDog#36 from DataDog/yang.song/fix-load-generator-staging
[OTEL-1363] Fix loadgenerator
2 parents bc5ad0f + 5d65728 commit bdb1b8d

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

src/loadgenerator/locustfile.py

+12-1
Original file line numberDiff line numberDiff line change
@@ -10,18 +10,29 @@
1010
from locust import HttpUser, task, between
1111

1212
from opentelemetry import context, baggage, trace
13+
from opentelemetry.metrics import set_meter_provider
14+
from opentelemetry.sdk.metrics import MeterProvider
15+
from opentelemetry.sdk.metrics.export import MetricExporter, PeriodicExportingMetricReader
1316
from opentelemetry.sdk.trace import TracerProvider
1417
from opentelemetry.sdk.trace.export import BatchSpanProcessor
15-
from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
18+
from opentelemetry.exporter.otlp.proto.grpc.metric_exporter import OTLPMetricExporter
19+
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
20+
from opentelemetry.instrumentation.jinja2 import Jinja2Instrumentor
1621
from opentelemetry.instrumentation.requests import RequestsInstrumentor
22+
from opentelemetry.instrumentation.system_metrics import SystemMetricsInstrumentor
1723
from opentelemetry.instrumentation.urllib3 import URLLib3Instrumentor
1824

25+
exporter = OTLPMetricExporter(insecure=True)
26+
set_meter_provider(MeterProvider([PeriodicExportingMetricReader(exporter)]))
27+
1928
tracer_provider = TracerProvider()
2029
trace.set_tracer_provider(tracer_provider)
2130
tracer_provider.add_span_processor(BatchSpanProcessor(OTLPSpanExporter()))
2231

2332
# Instrumenting manually to avoid error with locust gevent monkey
33+
Jinja2Instrumentor().instrument()
2434
RequestsInstrumentor().instrument()
35+
SystemMetricsInstrumentor().instrument()
2536
URLLib3Instrumentor().instrument()
2637

2738
categories = [

0 commit comments

Comments
 (0)