|
30 | 30 | import com.google.api.gax.rpc.UnaryCallable; |
31 | 31 | import com.google.api.pathtemplate.PathTemplate; |
32 | 32 | import com.google.common.base.Strings; |
| 33 | +import com.google.iam.v1.GetIamPolicyRequest; |
| 34 | +import com.google.iam.v1.Policy; |
| 35 | +import com.google.iam.v1.SetIamPolicyRequest; |
| 36 | +import com.google.iam.v1.TestIamPermissionsRequest; |
| 37 | +import com.google.iam.v1.TestIamPermissionsResponse; |
33 | 38 | import com.google.longrunning.Operation; |
34 | 39 | import com.google.longrunning.stub.GrpcOperationsStub; |
35 | 40 | import com.google.protobuf.Empty; |
@@ -343,6 +348,36 @@ public class GrpcStorageControlStub extends StorageControlStub { |
343 | 348 | .setSampledToLocalTracing(true) |
344 | 349 | .build(); |
345 | 350 |
|
| 351 | + private static final MethodDescriptor<GetIamPolicyRequest, Policy> getIamPolicyMethodDescriptor = |
| 352 | + MethodDescriptor.<GetIamPolicyRequest, Policy>newBuilder() |
| 353 | + .setType(MethodDescriptor.MethodType.UNARY) |
| 354 | + .setFullMethodName("google.storage.control.v2.StorageControl/GetIamPolicy") |
| 355 | + .setRequestMarshaller(ProtoUtils.marshaller(GetIamPolicyRequest.getDefaultInstance())) |
| 356 | + .setResponseMarshaller(ProtoUtils.marshaller(Policy.getDefaultInstance())) |
| 357 | + .setSampledToLocalTracing(true) |
| 358 | + .build(); |
| 359 | + |
| 360 | + private static final MethodDescriptor<SetIamPolicyRequest, Policy> setIamPolicyMethodDescriptor = |
| 361 | + MethodDescriptor.<SetIamPolicyRequest, Policy>newBuilder() |
| 362 | + .setType(MethodDescriptor.MethodType.UNARY) |
| 363 | + .setFullMethodName("google.storage.control.v2.StorageControl/SetIamPolicy") |
| 364 | + .setRequestMarshaller(ProtoUtils.marshaller(SetIamPolicyRequest.getDefaultInstance())) |
| 365 | + .setResponseMarshaller(ProtoUtils.marshaller(Policy.getDefaultInstance())) |
| 366 | + .setSampledToLocalTracing(true) |
| 367 | + .build(); |
| 368 | + |
| 369 | + private static final MethodDescriptor<TestIamPermissionsRequest, TestIamPermissionsResponse> |
| 370 | + testIamPermissionsMethodDescriptor = |
| 371 | + MethodDescriptor.<TestIamPermissionsRequest, TestIamPermissionsResponse>newBuilder() |
| 372 | + .setType(MethodDescriptor.MethodType.UNARY) |
| 373 | + .setFullMethodName("google.storage.control.v2.StorageControl/TestIamPermissions") |
| 374 | + .setRequestMarshaller( |
| 375 | + ProtoUtils.marshaller(TestIamPermissionsRequest.getDefaultInstance())) |
| 376 | + .setResponseMarshaller( |
| 377 | + ProtoUtils.marshaller(TestIamPermissionsResponse.getDefaultInstance())) |
| 378 | + .setSampledToLocalTracing(true) |
| 379 | + .build(); |
| 380 | + |
346 | 381 | private final UnaryCallable<CreateFolderRequest, Folder> createFolderCallable; |
347 | 382 | private final UnaryCallable<DeleteFolderRequest, Empty> deleteFolderCallable; |
348 | 383 | private final UnaryCallable<GetFolderRequest, Folder> getFolderCallable; |
@@ -391,6 +426,10 @@ public class GrpcStorageControlStub extends StorageControlStub { |
391 | 426 | getOrganizationIntelligenceConfigCallable; |
392 | 427 | private final UnaryCallable<UpdateOrganizationIntelligenceConfigRequest, IntelligenceConfig> |
393 | 428 | updateOrganizationIntelligenceConfigCallable; |
| 429 | + private final UnaryCallable<GetIamPolicyRequest, Policy> getIamPolicyCallable; |
| 430 | + private final UnaryCallable<SetIamPolicyRequest, Policy> setIamPolicyCallable; |
| 431 | + private final UnaryCallable<TestIamPermissionsRequest, TestIamPermissionsResponse> |
| 432 | + testIamPermissionsCallable; |
394 | 433 |
|
395 | 434 | private final BackgroundResource backgroundResources; |
396 | 435 | private final GrpcOperationsStub operationsStub; |
@@ -430,6 +469,20 @@ public class GrpcStorageControlStub extends StorageControlStub { |
430 | 469 | PathTemplate.create("{bucket=projects/*/buckets/*}/**"); |
431 | 470 | private static final PathTemplate LIST_ANYWHERE_CACHES_0_PATH_TEMPLATE = |
432 | 471 | PathTemplate.create("{bucket=**}"); |
| 472 | + private static final PathTemplate GET_IAM_POLICY_0_PATH_TEMPLATE = |
| 473 | + PathTemplate.create("{bucket=**}"); |
| 474 | + private static final PathTemplate GET_IAM_POLICY_1_PATH_TEMPLATE = |
| 475 | + PathTemplate.create("{bucket=projects/*/buckets/*}/**"); |
| 476 | + private static final PathTemplate SET_IAM_POLICY_0_PATH_TEMPLATE = |
| 477 | + PathTemplate.create("{bucket=**}"); |
| 478 | + private static final PathTemplate SET_IAM_POLICY_1_PATH_TEMPLATE = |
| 479 | + PathTemplate.create("{bucket=projects/*/buckets/*}/**"); |
| 480 | + private static final PathTemplate TEST_IAM_PERMISSIONS_0_PATH_TEMPLATE = |
| 481 | + PathTemplate.create("{bucket=**}"); |
| 482 | + private static final PathTemplate TEST_IAM_PERMISSIONS_1_PATH_TEMPLATE = |
| 483 | + PathTemplate.create("{bucket=projects/*/buckets/*}/objects/**"); |
| 484 | + private static final PathTemplate TEST_IAM_PERMISSIONS_2_PATH_TEMPLATE = |
| 485 | + PathTemplate.create("{bucket=projects/*/buckets/*}/managedFolders/**"); |
433 | 486 |
|
434 | 487 | public static final GrpcStorageControlStub create(StorageControlStubSettings settings) |
435 | 488 | throws IOException { |
@@ -859,6 +912,44 @@ protected GrpcStorageControlStub( |
859 | 912 | return builder.build(); |
860 | 913 | }) |
861 | 914 | .build(); |
| 915 | + GrpcCallSettings<GetIamPolicyRequest, Policy> getIamPolicyTransportSettings = |
| 916 | + GrpcCallSettings.<GetIamPolicyRequest, Policy>newBuilder() |
| 917 | + .setMethodDescriptor(getIamPolicyMethodDescriptor) |
| 918 | + .setParamsExtractor( |
| 919 | + request -> { |
| 920 | + RequestParamsBuilder builder = RequestParamsBuilder.create(); |
| 921 | + builder.add(request.getResource(), "bucket", GET_IAM_POLICY_0_PATH_TEMPLATE); |
| 922 | + builder.add(request.getResource(), "bucket", GET_IAM_POLICY_1_PATH_TEMPLATE); |
| 923 | + return builder.build(); |
| 924 | + }) |
| 925 | + .build(); |
| 926 | + GrpcCallSettings<SetIamPolicyRequest, Policy> setIamPolicyTransportSettings = |
| 927 | + GrpcCallSettings.<SetIamPolicyRequest, Policy>newBuilder() |
| 928 | + .setMethodDescriptor(setIamPolicyMethodDescriptor) |
| 929 | + .setParamsExtractor( |
| 930 | + request -> { |
| 931 | + RequestParamsBuilder builder = RequestParamsBuilder.create(); |
| 932 | + builder.add(request.getResource(), "bucket", SET_IAM_POLICY_0_PATH_TEMPLATE); |
| 933 | + builder.add(request.getResource(), "bucket", SET_IAM_POLICY_1_PATH_TEMPLATE); |
| 934 | + return builder.build(); |
| 935 | + }) |
| 936 | + .build(); |
| 937 | + GrpcCallSettings<TestIamPermissionsRequest, TestIamPermissionsResponse> |
| 938 | + testIamPermissionsTransportSettings = |
| 939 | + GrpcCallSettings.<TestIamPermissionsRequest, TestIamPermissionsResponse>newBuilder() |
| 940 | + .setMethodDescriptor(testIamPermissionsMethodDescriptor) |
| 941 | + .setParamsExtractor( |
| 942 | + request -> { |
| 943 | + RequestParamsBuilder builder = RequestParamsBuilder.create(); |
| 944 | + builder.add( |
| 945 | + request.getResource(), "bucket", TEST_IAM_PERMISSIONS_0_PATH_TEMPLATE); |
| 946 | + builder.add( |
| 947 | + request.getResource(), "bucket", TEST_IAM_PERMISSIONS_1_PATH_TEMPLATE); |
| 948 | + builder.add( |
| 949 | + request.getResource(), "bucket", TEST_IAM_PERMISSIONS_2_PATH_TEMPLATE); |
| 950 | + return builder.build(); |
| 951 | + }) |
| 952 | + .build(); |
862 | 953 |
|
863 | 954 | this.createFolderCallable = |
864 | 955 | callableFactory.createUnaryCallable( |
@@ -990,6 +1081,17 @@ protected GrpcStorageControlStub( |
990 | 1081 | updateOrganizationIntelligenceConfigTransportSettings, |
991 | 1082 | settings.updateOrganizationIntelligenceConfigSettings(), |
992 | 1083 | clientContext); |
| 1084 | + this.getIamPolicyCallable = |
| 1085 | + callableFactory.createUnaryCallable( |
| 1086 | + getIamPolicyTransportSettings, settings.getIamPolicySettings(), clientContext); |
| 1087 | + this.setIamPolicyCallable = |
| 1088 | + callableFactory.createUnaryCallable( |
| 1089 | + setIamPolicyTransportSettings, settings.setIamPolicySettings(), clientContext); |
| 1090 | + this.testIamPermissionsCallable = |
| 1091 | + callableFactory.createUnaryCallable( |
| 1092 | + testIamPermissionsTransportSettings, |
| 1093 | + settings.testIamPermissionsSettings(), |
| 1094 | + clientContext); |
993 | 1095 |
|
994 | 1096 | this.backgroundResources = |
995 | 1097 | new BackgroundResourceAggregation(clientContext.getBackgroundResources()); |
@@ -1157,6 +1259,22 @@ public UnaryCallable<GetAnywhereCacheRequest, AnywhereCache> getAnywhereCacheCal |
1157 | 1259 | return updateOrganizationIntelligenceConfigCallable; |
1158 | 1260 | } |
1159 | 1261 |
|
| 1262 | + @Override |
| 1263 | + public UnaryCallable<GetIamPolicyRequest, Policy> getIamPolicyCallable() { |
| 1264 | + return getIamPolicyCallable; |
| 1265 | + } |
| 1266 | + |
| 1267 | + @Override |
| 1268 | + public UnaryCallable<SetIamPolicyRequest, Policy> setIamPolicyCallable() { |
| 1269 | + return setIamPolicyCallable; |
| 1270 | + } |
| 1271 | + |
| 1272 | + @Override |
| 1273 | + public UnaryCallable<TestIamPermissionsRequest, TestIamPermissionsResponse> |
| 1274 | + testIamPermissionsCallable() { |
| 1275 | + return testIamPermissionsCallable; |
| 1276 | + } |
| 1277 | + |
1160 | 1278 | @Override |
1161 | 1279 | public final void close() { |
1162 | 1280 | try { |
|
0 commit comments