|
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; |
27 | 28 | import com.google.api.gax.rpc.UnaryCallable; |
| 29 | +import com.google.common.collect.ImmutableMap; |
28 | 30 | import com.google.iam.v1.GetIamPolicyRequest; |
29 | 31 | import com.google.iam.v1.Policy; |
30 | 32 | import com.google.iam.v1.SetIamPolicyRequest; |
|
44 | 46 | import io.grpc.MethodDescriptor; |
45 | 47 | import io.grpc.protobuf.ProtoUtils; |
46 | 48 | import java.io.IOException; |
| 49 | +import java.util.Map; |
47 | 50 | import java.util.concurrent.TimeUnit; |
48 | 51 | import javax.annotation.Generated; |
49 | 52 |
|
@@ -196,45 +199,135 @@ protected GrpcPublisherStub( |
196 | 199 | GrpcCallSettings<Topic, Topic> createTopicTransportSettings = |
197 | 200 | GrpcCallSettings.<Topic, Topic>newBuilder() |
198 | 201 | .setMethodDescriptor(createTopicMethodDescriptor) |
| 202 | + .setParamsExtractor( |
| 203 | + new RequestParamsExtractor<Topic>() { |
| 204 | + @Override |
| 205 | + public Map<String, String> extract(Topic request) { |
| 206 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 207 | + params.put("name", String.valueOf(request.getName())); |
| 208 | + return params.build(); |
| 209 | + } |
| 210 | + }) |
199 | 211 | .build(); |
200 | 212 | GrpcCallSettings<UpdateTopicRequest, Topic> updateTopicTransportSettings = |
201 | 213 | GrpcCallSettings.<UpdateTopicRequest, Topic>newBuilder() |
202 | 214 | .setMethodDescriptor(updateTopicMethodDescriptor) |
| 215 | + .setParamsExtractor( |
| 216 | + new RequestParamsExtractor<UpdateTopicRequest>() { |
| 217 | + @Override |
| 218 | + public Map<String, String> extract(UpdateTopicRequest request) { |
| 219 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 220 | + params.put("topic.name", String.valueOf(request.getTopic().getName())); |
| 221 | + return params.build(); |
| 222 | + } |
| 223 | + }) |
203 | 224 | .build(); |
204 | 225 | GrpcCallSettings<PublishRequest, PublishResponse> publishTransportSettings = |
205 | 226 | GrpcCallSettings.<PublishRequest, PublishResponse>newBuilder() |
206 | 227 | .setMethodDescriptor(publishMethodDescriptor) |
| 228 | + .setParamsExtractor( |
| 229 | + new RequestParamsExtractor<PublishRequest>() { |
| 230 | + @Override |
| 231 | + public Map<String, String> extract(PublishRequest request) { |
| 232 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 233 | + params.put("topic", String.valueOf(request.getTopic())); |
| 234 | + return params.build(); |
| 235 | + } |
| 236 | + }) |
207 | 237 | .build(); |
208 | 238 | GrpcCallSettings<GetTopicRequest, Topic> getTopicTransportSettings = |
209 | 239 | GrpcCallSettings.<GetTopicRequest, Topic>newBuilder() |
210 | 240 | .setMethodDescriptor(getTopicMethodDescriptor) |
| 241 | + .setParamsExtractor( |
| 242 | + new RequestParamsExtractor<GetTopicRequest>() { |
| 243 | + @Override |
| 244 | + public Map<String, String> extract(GetTopicRequest request) { |
| 245 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 246 | + params.put("topic", String.valueOf(request.getTopic())); |
| 247 | + return params.build(); |
| 248 | + } |
| 249 | + }) |
211 | 250 | .build(); |
212 | 251 | GrpcCallSettings<ListTopicsRequest, ListTopicsResponse> listTopicsTransportSettings = |
213 | 252 | GrpcCallSettings.<ListTopicsRequest, ListTopicsResponse>newBuilder() |
214 | 253 | .setMethodDescriptor(listTopicsMethodDescriptor) |
| 254 | + .setParamsExtractor( |
| 255 | + new RequestParamsExtractor<ListTopicsRequest>() { |
| 256 | + @Override |
| 257 | + public Map<String, String> extract(ListTopicsRequest request) { |
| 258 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 259 | + params.put("project", String.valueOf(request.getProject())); |
| 260 | + return params.build(); |
| 261 | + } |
| 262 | + }) |
215 | 263 | .build(); |
216 | 264 | GrpcCallSettings<ListTopicSubscriptionsRequest, ListTopicSubscriptionsResponse> |
217 | 265 | listTopicSubscriptionsTransportSettings = |
218 | 266 | GrpcCallSettings |
219 | 267 | .<ListTopicSubscriptionsRequest, ListTopicSubscriptionsResponse>newBuilder() |
220 | 268 | .setMethodDescriptor(listTopicSubscriptionsMethodDescriptor) |
| 269 | + .setParamsExtractor( |
| 270 | + new RequestParamsExtractor<ListTopicSubscriptionsRequest>() { |
| 271 | + @Override |
| 272 | + public Map<String, String> extract(ListTopicSubscriptionsRequest request) { |
| 273 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 274 | + params.put("topic", String.valueOf(request.getTopic())); |
| 275 | + return params.build(); |
| 276 | + } |
| 277 | + }) |
221 | 278 | .build(); |
222 | 279 | GrpcCallSettings<DeleteTopicRequest, Empty> deleteTopicTransportSettings = |
223 | 280 | GrpcCallSettings.<DeleteTopicRequest, Empty>newBuilder() |
224 | 281 | .setMethodDescriptor(deleteTopicMethodDescriptor) |
| 282 | + .setParamsExtractor( |
| 283 | + new RequestParamsExtractor<DeleteTopicRequest>() { |
| 284 | + @Override |
| 285 | + public Map<String, String> extract(DeleteTopicRequest request) { |
| 286 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 287 | + params.put("topic", String.valueOf(request.getTopic())); |
| 288 | + return params.build(); |
| 289 | + } |
| 290 | + }) |
225 | 291 | .build(); |
226 | 292 | GrpcCallSettings<SetIamPolicyRequest, Policy> setIamPolicyTransportSettings = |
227 | 293 | GrpcCallSettings.<SetIamPolicyRequest, Policy>newBuilder() |
228 | 294 | .setMethodDescriptor(setIamPolicyMethodDescriptor) |
| 295 | + .setParamsExtractor( |
| 296 | + new RequestParamsExtractor<SetIamPolicyRequest>() { |
| 297 | + @Override |
| 298 | + public Map<String, String> extract(SetIamPolicyRequest request) { |
| 299 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 300 | + params.put("resource", String.valueOf(request.getResource())); |
| 301 | + return params.build(); |
| 302 | + } |
| 303 | + }) |
229 | 304 | .build(); |
230 | 305 | GrpcCallSettings<GetIamPolicyRequest, Policy> getIamPolicyTransportSettings = |
231 | 306 | GrpcCallSettings.<GetIamPolicyRequest, Policy>newBuilder() |
232 | 307 | .setMethodDescriptor(getIamPolicyMethodDescriptor) |
| 308 | + .setParamsExtractor( |
| 309 | + new RequestParamsExtractor<GetIamPolicyRequest>() { |
| 310 | + @Override |
| 311 | + public Map<String, String> extract(GetIamPolicyRequest request) { |
| 312 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 313 | + params.put("resource", String.valueOf(request.getResource())); |
| 314 | + return params.build(); |
| 315 | + } |
| 316 | + }) |
233 | 317 | .build(); |
234 | 318 | GrpcCallSettings<TestIamPermissionsRequest, TestIamPermissionsResponse> |
235 | 319 | testIamPermissionsTransportSettings = |
236 | 320 | GrpcCallSettings.<TestIamPermissionsRequest, TestIamPermissionsResponse>newBuilder() |
237 | 321 | .setMethodDescriptor(testIamPermissionsMethodDescriptor) |
| 322 | + .setParamsExtractor( |
| 323 | + new RequestParamsExtractor<TestIamPermissionsRequest>() { |
| 324 | + @Override |
| 325 | + public Map<String, String> extract(TestIamPermissionsRequest request) { |
| 326 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 327 | + params.put("resource", String.valueOf(request.getResource())); |
| 328 | + return params.build(); |
| 329 | + } |
| 330 | + }) |
238 | 331 | .build(); |
239 | 332 |
|
240 | 333 | this.createTopicCallable = |
|
0 commit comments