Skip to content

Commit

Permalink
Chore: Made collector configurable with string bean in test-webapp
Browse files Browse the repository at this point in the history
  • Loading branch information
ricklue committed Aug 23, 2024
1 parent d537491 commit 3699a91
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 13 deletions.
27 changes: 15 additions & 12 deletions src/main/java/nl/nn/testtool/Config.java
Original file line number Diff line number Diff line change
Expand Up @@ -25,12 +25,9 @@
import io.opentelemetry.api.trace.propagation.W3CTraceContextPropagator;
import io.opentelemetry.context.propagation.ContextPropagators;
import io.opentelemetry.context.propagation.TextMapPropagator;
import io.opentelemetry.exporter.logging.SystemOutLogRecordExporter;
import io.opentelemetry.exporter.zipkin.ZipkinSpanExporter;
import io.opentelemetry.exporter.otlp.trace.OtlpGrpcSpanExporter;
import io.opentelemetry.sdk.OpenTelemetrySdk;
import io.opentelemetry.sdk.logs.SdkLoggerProvider;
import io.opentelemetry.sdk.logs.export.BatchLogRecordProcessor;
import io.opentelemetry.sdk.resources.Resource;
import io.opentelemetry.sdk.trace.SdkTracerProvider;
import io.opentelemetry.sdk.trace.export.BatchSpanProcessor;
Expand Down Expand Up @@ -117,16 +114,22 @@
@Configuration
public class Config {
@Bean
OpenTelemetry openTelemetry() {
OpenTelemetry openTelemetry(@Qualifier("openTelemetryCollector") String openTelemetryCollector) {
Resource resource = Resource.getDefault().toBuilder().put(ServiceAttributes.SERVICE_NAME, "ladybug").put(ServiceAttributes.SERVICE_VERSION, "1.0.0").build();
String endpointZipkin = "http://localhost:9411/api/v2/spans";
ZipkinSpanExporter zipkinExporter = ZipkinSpanExporter.builder().setEndpoint(endpointZipkin).build();
OtlpGrpcSpanExporter jaegerExporter = OtlpGrpcSpanExporter.builder().build();

SdkTracerProvider sdkTracerProvider = SdkTracerProvider.builder()
.addSpanProcessor(BatchSpanProcessor.builder(zipkinExporter).build())
.setResource(resource)
.build();
SdkTracerProvider sdkTracerProvider;
if (openTelemetryCollector.contains("9411")) {
sdkTracerProvider = SdkTracerProvider.builder()
.addSpanProcessor(BatchSpanProcessor.builder(ZipkinSpanExporter.builder().setEndpoint(openTelemetryCollector).build()).build())
.setResource(resource)
.build();
} else if (openTelemetryCollector.equals("jaeger")) {
sdkTracerProvider = SdkTracerProvider.builder()
.addSpanProcessor(BatchSpanProcessor.builder(OtlpGrpcSpanExporter.builder().build()).build())
.setResource(resource)
.build();
} else {
return OpenTelemetry.noop();
}

OpenTelemetry openTelemetry = OpenTelemetrySdk.builder()
.setTracerProvider(sdkTracerProvider)
Expand Down
2 changes: 1 addition & 1 deletion src/main/java/nl/nn/testtool/Report.java
Original file line number Diff line number Diff line change
Expand Up @@ -517,7 +517,7 @@ private <T> T addCheckpoint(String threadName, String sourceClassName, String n
}
}
Span checkpointSpan = checkpointSpanBuilder.startSpan();
checkpointSpan.setAttribute("checkpointType", checkpointType);
checkpointSpan.setAttribute("checkpointType", checkpoint.getType());
checkpointSpan.setAttribute("checkpointTypeAsString", checkpoint.getTypeAsString());
checkpointSpan.setAttribute("checkpointLevel", checkpoint.getLevel());
checkpoint.setSpan(checkpointSpan);
Expand Down

0 comments on commit 3699a91

Please sign in to comment.