open-telemetry
/
opentelemetry-java-instrumentation
mirror of https://github.com/open-telemetry/opentelemetry-java-instrumentation.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

support logs in the fake test backend (#3374) 

* support logs in the fake test backend

* fix copypasta
This commit is contained in:
jason plumb 2021-06-22 00:27:03 -07:00 committed by GitHub
parent 9a8df0b199
commit bf3830a8b9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 53 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
smoke-tests/fake-backend/src/main/java/io/opentelemetry/smoketest/fakebackend/FakeBackendMain.java
View File

@ -30,6 +30,7 @@ import com.linecorp.armeria.server.Server;
import com.linecorp.armeria.server.grpc.GrpcService;
import com.linecorp.armeria.server.healthcheck.HealthCheckService;
import io.netty.buffer.ByteBufOutputStream;
import io.opentelemetry.proto.collector.logs.v1.ExportLogsServiceRequest;
import io.opentelemetry.proto.collector.metrics.v1.ExportMetricsServiceRequest;
import io.opentelemetry.proto.collector.trace.v1.ExportTraceServiceRequest;
import java.io.IOException;
@ -68,6 +69,15 @@ public class FakeBackendMain {
marshaller.writeValue(value, gen);
}
});
serializers.addSerializer(
new StdSerializer<>(ExportLogsServiceRequest.class) {
@Override
public void serialize(
ExportLogsServiceRequest value, JsonGenerator gen, SerializerProvider provider)
throws IOException {
marshaller.writeValue(value, gen);
}
});
module.setSerializers(serializers);
mapper.addModule(module);
OBJECT_MAPPER = mapper.build();
@ -76,18 +86,21 @@ public class FakeBackendMain {
public static void main(String[] args) {
var traceCollector = new FakeTraceCollectorService();
var metricsCollector = new FakeMetricsCollectorService();
var logsCollector = new FakeLogsCollectorService();
var server =
Server.builder()
.http(8080)
.service(GrpcService.builder()
.addService(traceCollector)
.addService(metricsCollector)
.addService(logsCollector)
.build())
.service(
"/clear",
(ctx, req) -> {
traceCollector.clearRequests();
metricsCollector.clearRequests();
logsCollector.clearRequests();
return HttpResponse.of(HttpStatus.OK);
})
.service(
@ -108,6 +121,15 @@ public class FakeBackendMain {
return HttpResponse.of(
HttpStatus.OK, MediaType.JSON, HttpData.wrap(buf.buffer()));
})
.service(
"/get-logs",
(ctx, req) -> {
var requests = logsCollector.getRequests();
var buf = new ByteBufOutputStream(ctx.alloc().buffer());
OBJECT_MAPPER.writeValue((OutputStream) buf, requests);
return HttpResponse.of(
HttpStatus.OK, MediaType.JSON, HttpData.wrap(buf.buffer()));
})
.service("/health", HealthCheckService.of())
.build();

31
smoke-tests/fake-backend/src/main/java/io/opentelemetry/smoketest/fakebackend/FakeLogsCollectorService.java Normal file
View File

@ -0,0 +1,31 @@
package io.opentelemetry.smoketest.fakebackend;
import com.google.common.collect.ImmutableList;
import io.grpc.stub.StreamObserver;
import io.opentelemetry.proto.collector.logs.v1.ExportLogsServiceRequest;
import io.opentelemetry.proto.collector.logs.v1.ExportLogsServiceResponse;
import io.opentelemetry.proto.collector.logs.v1.LogsServiceGrpc;
import java.util.List;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingDeque;
public class FakeLogsCollectorService extends LogsServiceGrpc.LogsServiceImplBase{
private BlockingQueue<ExportLogsServiceRequest> exportRequests = new LinkedBlockingDeque<>();
List<ExportLogsServiceRequest> getRequests() {
return ImmutableList.copyOf(exportRequests);
}
void clearRequests() {
exportRequests.clear();
}
@Override
public void export(ExportLogsServiceRequest request,
StreamObserver<ExportLogsServiceResponse> responseObserver) {
exportRequests.add(request);
responseObserver.onNext(ExportLogsServiceResponse.getDefaultInstance());
responseObserver.onCompleted();
}
}