diff --git a/sdk/src/main/java/io/dapr/client/DaprClientGrpc.java b/sdk/src/main/java/io/dapr/client/DaprClientGrpc.java index 527117708..e5e639a85 100644 --- a/sdk/src/main/java/io/dapr/client/DaprClientGrpc.java +++ b/sdk/src/main/java/io/dapr/client/DaprClientGrpc.java @@ -21,6 +21,8 @@ import io.dapr.client.domain.SaveStateRequest; import io.dapr.client.domain.State; import io.dapr.client.domain.StateOptions; import io.dapr.serializer.DaprObjectSerializer; +import io.dapr.utils.Constants; +import io.dapr.utils.Properties; import io.dapr.utils.TypeRef; import io.dapr.v1.CommonProtos; import io.dapr.v1.DaprGrpc; @@ -33,6 +35,7 @@ import io.grpc.Context; import io.grpc.ForwardingClientCall; import io.grpc.Metadata; import io.grpc.MethodDescriptor; +import io.grpc.Metadata.Key; import io.opencensus.implcore.trace.propagation.PropagationComponentImpl; import io.opencensus.implcore.trace.propagation.TraceContextFormat; import io.opencensus.trace.SpanContext; @@ -483,8 +486,15 @@ public class DaprClientGrpc extends AbstractDaprClient { SpanContext opencensusSpanContext = extractOpenCensusSpanContext(context); if (opencensusSpanContext != null) { byte[] grpcTraceBin = OPENCENSUS_BINARY_FORMAT.toByteArray(opencensusSpanContext); - headers.put(Metadata.Key.of("grpc-trace-bin", Metadata.BINARY_BYTE_MARSHALLER), grpcTraceBin); + headers.put(Key.of("grpc-trace-bin", Metadata.BINARY_BYTE_MARSHALLER), grpcTraceBin); } + + String daprApiToken = Properties.getStringOrDefault(Constants.DAPR_API_TOKEN, + Constants.DAPR_API_TOKEN, null); + if (daprApiToken != null) { + headers.put(Key.of(Constants.DAPR_API_TOKEN_HEADER, Metadata.ASCII_STRING_MARSHALLER), daprApiToken); + } + super.start(responseListener, headers); } }; diff --git a/sdk/src/main/java/io/dapr/client/DaprClientGrpcInterceptor.java b/sdk/src/main/java/io/dapr/client/DaprClientGrpcInterceptor.java deleted file mode 100644 index 794857dc7..000000000 --- a/sdk/src/main/java/io/dapr/client/DaprClientGrpcInterceptor.java +++ /dev/null @@ -1,44 +0,0 @@ -/* - * Copyright (c) Microsoft Corporation. - * Licensed under the MIT License. - */ - -package io.dapr; - -import io.dapr.utils.Constants; -import io.dapr.utils.Properties; -import io.grpc.CallOptions; -import io.grpc.Channel; -import io.grpc.ClientCall; -import io.grpc.ClientInterceptor; -import io.grpc.ForwardingClientCall; -import io.grpc.Metadata; -import io.grpc.Metadata.Key; -import io.grpc.MethodDescriptor; - -/** - * A dapr grpc client call interceptor. - */ -public class DaprClientGrpcInterceptor implements ClientInterceptor { - - /** - * {@inheritDoc} - */ - @Override - public ClientCall interceptCall(MethodDescriptor methodDescriptor, - CallOptions callOptions, Channel channel) { - return new ForwardingClientCall.SimpleForwardingClientCall(channel.newCall(methodDescriptor, - callOptions)) { - @Override - public void start(final Listener responseListener, final Metadata headers) { - String daprApiToken = Properties.getStringOrDefault(Constants.DAPR_API_TOKEN, - Constants.DAPR_API_TOKEN, null); - if (daprApiToken != null) { - headers.put(Key.of(Constants.DAPR_API_TOKEN_HEADER, Metadata.ASCII_STRING_MARSHALLER), daprApiToken); - } - super.start(responseListener, headers); - } - }; - } -} \ No newline at end of file