Skip to content

Commit 9b43798

Browse files
committed
adding no extra headers
1 parent ef5f002 commit 9b43798

File tree

1 file changed

+2
-44
lines changed

1 file changed

+2
-44
lines changed

google-cloud-datastore/src/main/java/com/google/cloud/datastore/spi/v1/GrpcDatastoreRpc.java

+2-44
Original file line numberDiff line numberDiff line change
@@ -22,24 +22,19 @@
2222
import com.google.api.core.ApiFunction;
2323
import com.google.api.core.InternalApi;
2424
import com.google.api.gax.core.BackgroundResource;
25-
import com.google.api.gax.core.GaxProperties;
2625
import com.google.api.gax.grpc.GrpcCallContext;
2726
import com.google.api.gax.grpc.GrpcTransportChannel;
2827
import com.google.api.gax.rpc.ClientContext;
29-
import com.google.api.gax.rpc.HeaderProvider;
30-
import com.google.api.gax.rpc.NoHeaderProvider;
3128
import com.google.api.gax.rpc.TransportChannel;
3229
import com.google.api.gax.rpc.UnaryCallSettings;
3330
import com.google.cloud.NoCredentials;
34-
import com.google.cloud.ServiceOptions;
3531
import com.google.cloud.datastore.DatastoreException;
3632
import com.google.cloud.datastore.DatastoreOptions;
3733
import com.google.cloud.datastore.DatastoreUtils;
3834
import com.google.cloud.datastore.v1.DatastoreSettings;
3935
import com.google.cloud.datastore.v1.stub.DatastoreStubSettings;
4036
import com.google.cloud.datastore.v1.stub.GrpcDatastoreStub;
4137
import com.google.cloud.grpc.GrpcTransportOptions;
42-
import com.google.common.base.Strings;
4338
import com.google.datastore.v1.AllocateIdsRequest;
4439
import com.google.datastore.v1.AllocateIdsResponse;
4540
import com.google.datastore.v1.BeginTransactionRequest;
@@ -168,49 +163,12 @@ private ClientContext getClientContextForEmulator(DatastoreOptions datastoreOpti
168163
}
169164

170165
private ClientContext getClientContext(DatastoreOptions datastoreOptions) throws IOException {
171-
HeaderProvider internalHeaderProvider =
172-
DatastoreSettings.defaultApiClientHeaderProviderBuilder()
173-
.setClientLibToken(
174-
ServiceOptions.getGoogApiClientLibName(),
175-
GaxProperties.getLibraryVersion(datastoreOptions.getClass()))
176-
.setResourceToken(getResourceToken(datastoreOptions))
177-
.build();
178-
179-
DatastoreSettingsBuilder settingsBuilder =
180-
new DatastoreSettingsBuilder(DatastoreSettings.newBuilder().build());
166+
DatastoreSettings.Builder settingsBuilder = DatastoreSettings.newBuilder();
181167
settingsBuilder.setCredentialsProvider(
182168
GrpcTransportOptions.setUpCredentialsProvider(datastoreOptions));
183169
settingsBuilder.setTransportChannelProvider(
184170
GrpcTransportOptions.setUpChannelProvider(
185171
DatastoreSettings.defaultGrpcTransportProviderBuilder(), datastoreOptions));
186-
settingsBuilder.setInternalHeaderProvider(internalHeaderProvider);
187-
settingsBuilder.setHeaderProvider(
188-
datastoreOptions.getMergedHeaderProvider(new NoHeaderProvider()));
189-
ClientContext clientContext = ClientContext.create(settingsBuilder.build());
190-
return clientContext;
191-
}
192-
193-
private String getResourceToken(DatastoreOptions datastoreOptions) {
194-
StringBuilder builder = new StringBuilder("project_id=");
195-
builder.append(datastoreOptions.getProjectId());
196-
if (!Strings.isNullOrEmpty(datastoreOptions.getDatabaseId())) {
197-
builder.append("&database_id=");
198-
builder.append(datastoreOptions.getDatabaseId());
199-
}
200-
return builder.toString();
201-
}
202-
203-
// This class is needed solely to get access to protected method setInternalHeaderProvider()
204-
private static class DatastoreSettingsBuilder extends DatastoreSettings.Builder {
205-
206-
private DatastoreSettingsBuilder(DatastoreSettings settings) {
207-
super(settings);
208-
}
209-
210-
@Override
211-
protected DatastoreSettings.Builder setInternalHeaderProvider(
212-
HeaderProvider internalHeaderProvider) {
213-
return super.setInternalHeaderProvider(internalHeaderProvider);
214-
}
172+
return ClientContext.create(settingsBuilder.build());
215173
}
216174
}

0 commit comments

Comments
 (0)