|
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