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

Implementing `ExtendedTextMapGetter` in grpc-1.6 instrumentation (#13011) 

Signed-off-by: xiepuhuan <puhuanxie@gmail.com>
This commit is contained in:
xiepuhuan 2025-01-08 23:28:14 +08:00 committed by GitHub
parent 324fdbdd45
commit ae3f6ac481
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
instrumentation/grpc-1.6/library/src/main/java/io/opentelemetry/instrumentation/grpc/v1_6/GrpcRequestGetter.java
View File

@ -5,11 +5,14 @@
package io.opentelemetry.instrumentation.grpc.v1_6; package io.opentelemetry.instrumentation.grpc.v1_6;
import static java.util.Collections.emptyIterator;
import io.grpc.Metadata; import io.grpc.Metadata;
import io.opentelemetry.context.propagation.TextMapGetter; import io.opentelemetry.context.propagation.internal.ExtendedTextMapGetter;
import java.util.Iterator;
import javax.annotation.Nullable; import javax.annotation.Nullable;
enum GrpcRequestGetter implements TextMapGetter<GrpcRequest> { enum GrpcRequestGetter implements ExtendedTextMapGetter<GrpcRequest> {
INSTANCE; INSTANCE;
@Override @Override
@ -25,4 +28,19 @@ enum GrpcRequestGetter implements TextMapGetter<GrpcRequest> {
} }
return request.getMetadata().get(Metadata.Key.of(key, Metadata.ASCII_STRING_MARSHALLER)); return request.getMetadata().get(Metadata.Key.of(key, Metadata.ASCII_STRING_MARSHALLER));
} }
@Override
public Iterator<String> getAll(@Nullable GrpcRequest request, String key) {
if (request == null) {
return emptyIterator();
}
Iterable<String> values =
request.getMetadata().getAll(Metadata.Key.of(key, Metadata.ASCII_STRING_MARSHALLER));
if (values == null) {
return emptyIterator();
}
return values.iterator();
}
} }