|
19 | 19 | import static org.mockito.Mockito.when; |
20 | 20 |
|
21 | 21 | import com.linecorp.armeria.server.ServerBuilder; |
22 | | -import com.linecorp.armeria.server.grpc.GrpcService; |
| 22 | +import com.linecorp.armeria.server.ServiceRequestContext; |
| 23 | +import com.linecorp.armeria.server.grpc.protocol.AbstractUnaryGrpcService; |
23 | 24 | import com.linecorp.armeria.server.logging.LoggingService; |
24 | 25 | import com.linecorp.armeria.testing.junit5.server.ServerExtension; |
25 | | -import io.grpc.stub.StreamObserver; |
26 | 26 | import io.opentelemetry.api.GlobalOpenTelemetry; |
27 | 27 | import io.opentelemetry.api.OpenTelemetry; |
28 | 28 | import io.opentelemetry.api.metrics.LongCounter; |
|
31 | 31 | import io.opentelemetry.api.metrics.MeterProvider; |
32 | 32 | import io.opentelemetry.api.trace.TracerProvider; |
33 | 33 | import io.opentelemetry.context.propagation.ContextPropagators; |
34 | | -import io.opentelemetry.proto.collector.metrics.v1.ExportMetricsServiceRequest; |
35 | 34 | import io.opentelemetry.proto.collector.metrics.v1.ExportMetricsServiceResponse; |
36 | | -import io.opentelemetry.proto.collector.metrics.v1.MetricsServiceGrpc; |
37 | 35 | import io.opentelemetry.sdk.common.InstrumentationScopeInfo; |
38 | 36 | import io.opentelemetry.sdk.common.export.MemoryMode; |
39 | 37 | import io.opentelemetry.sdk.metrics.SdkMeterProvider; |
|
45 | 43 | import io.opentelemetry.sdk.metrics.internal.data.ImmutableGaugeData; |
46 | 44 | import io.opentelemetry.sdk.resources.Resource; |
47 | 45 | import java.util.Collection; |
| 46 | +import java.util.concurrent.CompletableFuture; |
| 47 | +import java.util.concurrent.CompletionStage; |
48 | 48 | import java.util.concurrent.TimeUnit; |
49 | 49 | import java.util.concurrent.atomic.AtomicReference; |
50 | 50 | import java.util.function.Supplier; |
@@ -72,23 +72,16 @@ class OtlpGrpcMetricExporterBuilderTest { |
72 | 72 | @Override |
73 | 73 | protected void configure(ServerBuilder sb) { |
74 | 74 | sb.service( |
75 | | - GrpcService.builder() |
76 | | - .addService( |
77 | | - new MetricsServiceGrpc.MetricsServiceImplBase() { |
78 | | - @Override |
79 | | - public void export( |
80 | | - ExportMetricsServiceRequest request, |
81 | | - StreamObserver< |
82 | | - io.opentelemetry.proto.collector.metrics.v1 |
83 | | - .ExportMetricsServiceResponse> |
84 | | - responseObserver) { |
85 | | - responseObserver.onNext( |
86 | | - ExportMetricsServiceResponse.getDefaultInstance()); |
87 | | - responseObserver.onCompleted(); |
88 | | - responseObserver.onCompleted(); |
89 | | - } |
90 | | - }) |
91 | | - .build()); |
| 75 | + "/opentelemetry.proto.collector.metrics.v1.MetricsService/Export", |
| 76 | + new AbstractUnaryGrpcService() { |
| 77 | + @Override |
| 78 | + protected CompletionStage<byte[]> handleMessage( |
| 79 | + ServiceRequestContext ctx, byte[] message) { |
| 80 | + return CompletableFuture.completedFuture( |
| 81 | + ExportMetricsServiceResponse.getDefaultInstance().toByteArray()); |
| 82 | + } |
| 83 | + }); |
| 84 | + sb.http(0); |
92 | 85 | sb.decorator(LoggingService.newDecorator()); |
93 | 86 | } |
94 | 87 | }; |
|
0 commit comments