Skip to content
This repository was archived by the owner on Oct 18, 2023. It is now read-only.

Commit e5907fe

Browse files
docs(samples): update AWS sample due to API changes (#576)
AwsCrossAccountRole is replaced with AwsAccessRole Fixes #571
1 parent 439d33c commit e5907fe

File tree

2 files changed

+7
-6
lines changed

2 files changed

+7
-6
lines changed

samples/snippets/src/main/java/com/example/bigqueryconnection/CreateAwsConnection.java

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
package com.example.bigqueryconnection;
1818

1919
// [START bigqueryconnection_create_aws_connection]
20+
import com.google.cloud.bigquery.connection.v1.AwsAccessRole;
2021
import com.google.cloud.bigquery.connection.v1.AwsCrossAccountRole;
2122
import com.google.cloud.bigquery.connection.v1.AwsProperties;
2223
import com.google.cloud.bigquery.connection.v1.Connection;
@@ -54,12 +55,12 @@ public static void createAwsConnection(
5455
.setConnectionId(connectionId)
5556
.build();
5657
Connection response = client.createConnection(request);
57-
AwsCrossAccountRole role = response.getAws().getCrossAccountRole();
58+
AwsAccessRole role = response.getAws().getAccessRole();
5859
System.out.println(
5960
"Aws connection created successfully : Aws userId :"
60-
+ role.getIamUserId()
61+
+ role.getIamRoleId()
6162
+ " Aws externalId :"
62-
+ role.getExternalId());
63+
+ role.getIdentity());
6364
}
6465
}
6566
}

samples/snippets/src/test/java/com/example/bigqueryconnection/CreateAwsConnectionIT.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
import static com.google.common.truth.Truth.assertThat;
2020
import static junit.framework.TestCase.assertNotNull;
2121

22-
import com.google.cloud.bigquery.connection.v1.AwsCrossAccountRole;
22+
import com.google.cloud.bigquery.connection.v1.AwsAccessRole;
2323
import com.google.cloud.bigquery.connection.v1.AwsProperties;
2424
import com.google.cloud.bigquery.connection.v1.Connection;
2525
import java.io.ByteArrayOutputStream;
@@ -83,8 +83,8 @@ public void tearDown() throws IOException {
8383
@Test
8484
public void testCreateAwsConnection() throws IOException {
8585
String iamRoleId = String.format("arn:aws:iam::%s:role/%s", AWS_ACCOUNT_ID, AWS_ROLE_ID);
86-
AwsCrossAccountRole role = AwsCrossAccountRole.newBuilder().setIamRoleId(iamRoleId).build();
87-
AwsProperties awsProperties = AwsProperties.newBuilder().setCrossAccountRole(role).build();
86+
AwsAccessRole awsRole = AwsAccessRole.newBuilder().setIamRoleId(iamRoleId).build();
87+
AwsProperties awsProperties = AwsProperties.newBuilder().setAccessRole(awsRole).build();
8888
Connection connection = Connection.newBuilder().setAws(awsProperties).build();
8989
CreateAwsConnection.createAwsConnection(PROJECT_ID, LOCATION, connectionId, connection);
9090
assertThat(bout.toString()).contains("Aws connection created successfully :");

0 commit comments

Comments
 (0)