Skip to content

Commit

Permalink
Change to OTLP endpoint
Browse files Browse the repository at this point in the history
  • Loading branch information
stefanhellander committed Apr 29, 2024
1 parent d088ee2 commit 1fb343c
Show file tree
Hide file tree
Showing 3 changed files with 18 additions and 4 deletions.
18 changes: 14 additions & 4 deletions fedn/fedn/common/telemetry.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@
import psutil
from opentelemetry import trace
from opentelemetry.exporter.jaeger.thrift import JaegerExporter
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
from opentelemetry.sdk.resources import SERVICE_NAME, Resource
from opentelemetry.sdk.trace import TracerProvider
from opentelemetry.sdk.trace.export import BatchSpanProcessor
Expand Down Expand Up @@ -62,15 +63,24 @@ def initialize_tracer():
)
)

print(telemetry_server)
print(telemetry_port)
# Set up the Jaeger exporter
jaeger_exporter = JaegerExporter(
agent_host_name=telemetry_server,
agent_port=telemetry_port,
# jaeger_exporter = JaegerExporter(
# # agent_host_name=telemetry_server,
# collector_endpoint=f"http://{telemetry_server}:{telemetry_port}/api/traces",
# # agent_port=telemetry_port,
# )

otlp_exporter = OTLPSpanExporter(
endpoint="http://localhost:4317", # Default OTLP port for Jaeger, adjust as needed
insecure=True # Use this only if you are not setting up TLS/SSL
)


# Attach the exporter to the tracer provider
trace.get_tracer_provider().add_span_processor(
BatchSpanProcessor(jaeger_exporter)
BatchSpanProcessor(otlp_exporter)
)

return trace.get_tracer(__name__)
Expand Down
3 changes: 3 additions & 0 deletions fedn/fedn/network/combiner/combiner.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,6 +108,9 @@ def __init__(self, config):
logger.info(response)
logger.info("Status.UnMatchedConfig")
sys.exit("Exiting: Missing config")
else:
logger.info("Status.Unassigned")
time.sleep(5)

cert = announce_config['certificate']
key = announce_config['key']
Expand Down
1 change: 1 addition & 0 deletions fedn/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,6 +32,7 @@
"opentelemetry-api",
"opentelemetry-sdk",
"opentelemetry-exporter-jaeger",
"opentelemetry-exporter-otlp",
],
extras_require={
'flower': ["flwr==1.8.0"]
Expand Down

0 comments on commit 1fb343c

Please sign in to comment.