|
24 | 24 | import com.google.api.gax.grpc.GrpcCallSettings; |
25 | 25 | import com.google.api.gax.grpc.GrpcStubCallableFactory; |
26 | 26 | import com.google.api.gax.rpc.ClientContext; |
27 | | -import com.google.api.gax.rpc.RequestParamsExtractor; |
28 | 27 | import com.google.api.gax.rpc.UnaryCallable; |
29 | 28 | import com.google.cloud.orgpolicy.v2.CreatePolicyRequest; |
30 | 29 | import com.google.cloud.orgpolicy.v2.DeletePolicyRequest; |
|
42 | 41 | import io.grpc.MethodDescriptor; |
43 | 42 | import io.grpc.protobuf.ProtoUtils; |
44 | 43 | import java.io.IOException; |
45 | | -import java.util.Map; |
46 | 44 | import java.util.concurrent.TimeUnit; |
47 | 45 | import javax.annotation.Generated; |
48 | 46 |
|
@@ -176,91 +174,70 @@ protected GrpcOrgPolicyStub( |
176 | 174 | GrpcCallSettings.<ListConstraintsRequest, ListConstraintsResponse>newBuilder() |
177 | 175 | .setMethodDescriptor(listConstraintsMethodDescriptor) |
178 | 176 | .setParamsExtractor( |
179 | | - new RequestParamsExtractor<ListConstraintsRequest>() { |
180 | | - @Override |
181 | | - public Map<String, String> extract(ListConstraintsRequest request) { |
182 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
183 | | - params.put("parent", String.valueOf(request.getParent())); |
184 | | - return params.build(); |
185 | | - } |
| 177 | + request -> { |
| 178 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 179 | + params.put("parent", String.valueOf(request.getParent())); |
| 180 | + return params.build(); |
186 | 181 | }) |
187 | 182 | .build(); |
188 | 183 | GrpcCallSettings<ListPoliciesRequest, ListPoliciesResponse> listPoliciesTransportSettings = |
189 | 184 | GrpcCallSettings.<ListPoliciesRequest, ListPoliciesResponse>newBuilder() |
190 | 185 | .setMethodDescriptor(listPoliciesMethodDescriptor) |
191 | 186 | .setParamsExtractor( |
192 | | - new RequestParamsExtractor<ListPoliciesRequest>() { |
193 | | - @Override |
194 | | - public Map<String, String> extract(ListPoliciesRequest request) { |
195 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
196 | | - params.put("parent", String.valueOf(request.getParent())); |
197 | | - return params.build(); |
198 | | - } |
| 187 | + request -> { |
| 188 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 189 | + params.put("parent", String.valueOf(request.getParent())); |
| 190 | + return params.build(); |
199 | 191 | }) |
200 | 192 | .build(); |
201 | 193 | GrpcCallSettings<GetPolicyRequest, Policy> getPolicyTransportSettings = |
202 | 194 | GrpcCallSettings.<GetPolicyRequest, Policy>newBuilder() |
203 | 195 | .setMethodDescriptor(getPolicyMethodDescriptor) |
204 | 196 | .setParamsExtractor( |
205 | | - new RequestParamsExtractor<GetPolicyRequest>() { |
206 | | - @Override |
207 | | - public Map<String, String> extract(GetPolicyRequest request) { |
208 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
209 | | - params.put("name", String.valueOf(request.getName())); |
210 | | - return params.build(); |
211 | | - } |
| 197 | + request -> { |
| 198 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 199 | + params.put("name", String.valueOf(request.getName())); |
| 200 | + return params.build(); |
212 | 201 | }) |
213 | 202 | .build(); |
214 | 203 | GrpcCallSettings<GetEffectivePolicyRequest, Policy> getEffectivePolicyTransportSettings = |
215 | 204 | GrpcCallSettings.<GetEffectivePolicyRequest, Policy>newBuilder() |
216 | 205 | .setMethodDescriptor(getEffectivePolicyMethodDescriptor) |
217 | 206 | .setParamsExtractor( |
218 | | - new RequestParamsExtractor<GetEffectivePolicyRequest>() { |
219 | | - @Override |
220 | | - public Map<String, String> extract(GetEffectivePolicyRequest request) { |
221 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
222 | | - params.put("name", String.valueOf(request.getName())); |
223 | | - return params.build(); |
224 | | - } |
| 207 | + request -> { |
| 208 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 209 | + params.put("name", String.valueOf(request.getName())); |
| 210 | + return params.build(); |
225 | 211 | }) |
226 | 212 | .build(); |
227 | 213 | GrpcCallSettings<CreatePolicyRequest, Policy> createPolicyTransportSettings = |
228 | 214 | GrpcCallSettings.<CreatePolicyRequest, Policy>newBuilder() |
229 | 215 | .setMethodDescriptor(createPolicyMethodDescriptor) |
230 | 216 | .setParamsExtractor( |
231 | | - new RequestParamsExtractor<CreatePolicyRequest>() { |
232 | | - @Override |
233 | | - public Map<String, String> extract(CreatePolicyRequest request) { |
234 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
235 | | - params.put("parent", String.valueOf(request.getParent())); |
236 | | - return params.build(); |
237 | | - } |
| 217 | + request -> { |
| 218 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 219 | + params.put("parent", String.valueOf(request.getParent())); |
| 220 | + return params.build(); |
238 | 221 | }) |
239 | 222 | .build(); |
240 | 223 | GrpcCallSettings<UpdatePolicyRequest, Policy> updatePolicyTransportSettings = |
241 | 224 | GrpcCallSettings.<UpdatePolicyRequest, Policy>newBuilder() |
242 | 225 | .setMethodDescriptor(updatePolicyMethodDescriptor) |
243 | 226 | .setParamsExtractor( |
244 | | - new RequestParamsExtractor<UpdatePolicyRequest>() { |
245 | | - @Override |
246 | | - public Map<String, String> extract(UpdatePolicyRequest request) { |
247 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
248 | | - params.put("policy.name", String.valueOf(request.getPolicy().getName())); |
249 | | - return params.build(); |
250 | | - } |
| 227 | + request -> { |
| 228 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 229 | + params.put("policy.name", String.valueOf(request.getPolicy().getName())); |
| 230 | + return params.build(); |
251 | 231 | }) |
252 | 232 | .build(); |
253 | 233 | GrpcCallSettings<DeletePolicyRequest, Empty> deletePolicyTransportSettings = |
254 | 234 | GrpcCallSettings.<DeletePolicyRequest, Empty>newBuilder() |
255 | 235 | .setMethodDescriptor(deletePolicyMethodDescriptor) |
256 | 236 | .setParamsExtractor( |
257 | | - new RequestParamsExtractor<DeletePolicyRequest>() { |
258 | | - @Override |
259 | | - public Map<String, String> extract(DeletePolicyRequest request) { |
260 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
261 | | - params.put("name", String.valueOf(request.getName())); |
262 | | - return params.build(); |
263 | | - } |
| 237 | + request -> { |
| 238 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 239 | + params.put("name", String.valueOf(request.getName())); |
| 240 | + return params.build(); |
264 | 241 | }) |
265 | 242 | .build(); |
266 | 243 |
|
@@ -350,7 +327,13 @@ public UnaryCallable<DeletePolicyRequest, Empty> deletePolicyCallable() { |
350 | 327 |
|
351 | 328 | @Override |
352 | 329 | public final void close() { |
353 | | - shutdown(); |
| 330 | + try { |
| 331 | + backgroundResources.close(); |
| 332 | + } catch (RuntimeException e) { |
| 333 | + throw e; |
| 334 | + } catch (Exception e) { |
| 335 | + throw new IllegalStateException("Failed to close resource", e); |
| 336 | + } |
354 | 337 | } |
355 | 338 |
|
356 | 339 | @Override |
|
0 commit comments