|
22 | 22 | import com.google.api.core.ApiFunction;
|
23 | 23 | import com.google.api.core.InternalApi;
|
24 | 24 | import com.google.api.gax.core.BackgroundResource;
|
25 |
| -import com.google.api.gax.core.GaxProperties; |
26 | 25 | import com.google.api.gax.grpc.GrpcCallContext;
|
27 | 26 | import com.google.api.gax.grpc.GrpcTransportChannel;
|
28 | 27 | import com.google.api.gax.rpc.ClientContext;
|
29 |
| -import com.google.api.gax.rpc.HeaderProvider; |
30 |
| -import com.google.api.gax.rpc.NoHeaderProvider; |
31 | 28 | import com.google.api.gax.rpc.TransportChannel;
|
32 | 29 | import com.google.api.gax.rpc.UnaryCallSettings;
|
33 | 30 | import com.google.cloud.NoCredentials;
|
34 |
| -import com.google.cloud.ServiceOptions; |
35 | 31 | import com.google.cloud.datastore.DatastoreException;
|
36 | 32 | import com.google.cloud.datastore.DatastoreOptions;
|
37 | 33 | import com.google.cloud.datastore.DatastoreUtils;
|
38 | 34 | import com.google.cloud.datastore.v1.DatastoreSettings;
|
39 | 35 | import com.google.cloud.datastore.v1.stub.DatastoreStubSettings;
|
40 | 36 | import com.google.cloud.datastore.v1.stub.GrpcDatastoreStub;
|
41 | 37 | import com.google.cloud.grpc.GrpcTransportOptions;
|
42 |
| -import com.google.common.base.Strings; |
43 | 38 | import com.google.datastore.v1.AllocateIdsRequest;
|
44 | 39 | import com.google.datastore.v1.AllocateIdsResponse;
|
45 | 40 | import com.google.datastore.v1.BeginTransactionRequest;
|
@@ -168,49 +163,12 @@ private ClientContext getClientContextForEmulator(DatastoreOptions datastoreOpti
|
168 | 163 | }
|
169 | 164 |
|
170 | 165 | 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(); |
181 | 167 | settingsBuilder.setCredentialsProvider(
|
182 | 168 | GrpcTransportOptions.setUpCredentialsProvider(datastoreOptions));
|
183 | 169 | settingsBuilder.setTransportChannelProvider(
|
184 | 170 | GrpcTransportOptions.setUpChannelProvider(
|
185 | 171 | 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()); |
215 | 173 | }
|
216 | 174 | }
|
0 commit comments