feat(compute): add compute consistency group clone

compute/cloud-client/src/main/java/compute/disks/CreateReplicatedDisk.java

@@ -20,6 +20,7 @@
 import com.google.cloud.compute.v1.Disk;
 import com.google.cloud.compute.v1.InsertRegionDiskRequest;
 import com.google.cloud.compute.v1.Operation;
+import com.google.cloud.compute.v1.Operation.Status;
 import com.google.cloud.compute.v1.RegionDisksClient;
 import java.io.IOException;
 import java.util.ArrayList;
@@ -55,7 +56,7 @@ public static void main(String[] args)
   }
 
   // Create a disk for synchronous data replication between two zones in the same region
-  public static Operation.Status createReplicatedDisk(String projectId, String region,
+  public static Status createReplicatedDisk(String projectId, String region,
         List<String> replicaZones, String diskName, int diskSizeGb, String diskType)
         throws IOException, InterruptedException, ExecutionException, TimeoutException {
     // Initialize client that will be used to send requests. This client only needs to be created

compute/cloud-client/src/main/java/compute/disks/consistencygroup/AddRegionalDiskToConsistencyGroup.java

@@ -0,0 +1,77 @@
+/*
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package compute.disks.consistencygroup;
+
+// [START compute_consistency_group_add_regional_disk]
+import com.google.cloud.compute.v1.AddResourcePoliciesRegionDiskRequest;
+import com.google.cloud.compute.v1.Operation;
+import com.google.cloud.compute.v1.Operation.Status;
+import com.google.cloud.compute.v1.RegionDisksAddResourcePoliciesRequest;
+import com.google.cloud.compute.v1.RegionDisksClient;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
+
+public class AddRegionalDiskToConsistencyGroup {
+  public static void main(String[] args)
+          throws IOException, ExecutionException, InterruptedException, TimeoutException {
+    // TODO(developer): Replace these variables before running the sample.
+    // Project ID or project number of the Cloud project that contains the disk.
+    String project = "YOUR_PROJECT_ID";
+    // Region in which your disk and consistency group are located.
+    String region = "us-central1";
+    // Name of the disk.
+    String diskName = "DISK_NAME";
+    // Name of the consistency group.
+    String consistencyGroupName = "CONSISTENCY_GROUP";
+
+    addRegionalDiskToConsistencyGroup(project, region, diskName, consistencyGroupName);
+  }
+
+  // Adds regional disk to a consistency group.
+  public static Status addRegionalDiskToConsistencyGroup(
+      String project, String region, String diskName, String consistencyGroupName)
+          throws IOException, ExecutionException, InterruptedException, TimeoutException {
+    String consistencyGroupUrl = String.format(
+        "https://www.googleapis.com/compute/v1/projects/%s/regions/%s/resourcePolicies/%s",
+        project, region, consistencyGroupName);
+
+    // Initialize client that will be used to send requests. This client only needs to be created
+    // once, and can be reused for multiple requests.
+    try (RegionDisksClient disksClient = RegionDisksClient.create()) {
+      AddResourcePoliciesRegionDiskRequest request =
+              AddResourcePoliciesRegionDiskRequest.newBuilder()
+                      .setDisk(diskName)
+                      .setRegion(region)
+                      .setProject(project)
+                      .setRegionDisksAddResourcePoliciesRequestResource(
+                              RegionDisksAddResourcePoliciesRequest.newBuilder()
+                                      .addAllResourcePolicies(Arrays.asList(consistencyGroupUrl))
+                                      .build())
+                      .build();
+      Operation response = disksClient.addResourcePoliciesAsync(request).get(1, TimeUnit.MINUTES);
+
+      if (response.hasError()) {
+        throw new Error("Error adding regional disk to consistency group! " + response.getError());
+      }
+      return response.getStatus();
+    }
+  }
+}
+// [END compute_consistency_group_add_regional_disk]

compute/cloud-client/src/main/java/compute/disks/consistencygroup/AddZonalDiskToConsistencyGroup.java

@@ -0,0 +1,80 @@
+/*
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package compute.disks.consistencygroup;
+
+// [START compute_consistency_group_add_zonal_disk]
+import com.google.cloud.compute.v1.AddResourcePoliciesDiskRequest;
+import com.google.cloud.compute.v1.DisksAddResourcePoliciesRequest;
+import com.google.cloud.compute.v1.DisksClient;
+import com.google.cloud.compute.v1.Operation;
+import com.google.cloud.compute.v1.Operation.Status;
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
+
+public class AddZonalDiskToConsistencyGroup {
+  public static void main(String[] args)
+          throws IOException, ExecutionException, InterruptedException, TimeoutException {
+    // TODO(developer): Replace these variables before running the sample.
+    // Project ID or project number of the Cloud project that contains the disk.
+    String project = "YOUR_PROJECT_ID";
+    // Zone of the disk.
+    String diskLocation = "us-central1-a";
+    // Name of the disk.
+    String diskName = "DISK_NAME";
+    // Name of the consistency group.
+    String consistencyGroupName = "CONSISTENCY_GROUP";
+
+    addZonalDiskToConsistencyGroup(
+            project, diskLocation, diskName, consistencyGroupName);
+  }
+
+  // Adds zonal disk to a consistency group.
+  public static Status addZonalDiskToConsistencyGroup(
+          String project, String diskLocation, String diskName,
+          String consistencyGroupName)
+          throws IOException, ExecutionException, InterruptedException, TimeoutException {
+    String region = diskLocation.substring(0, diskLocation.lastIndexOf('-'));
+    String consistencyGroupUrl = String.format(
+            "https://www.googleapis.com/compute/v1/projects/%s/regions/%s/resourcePolicies/%s",
+            project, region, consistencyGroupName);
+
+    // Initialize client that will be used to send requests. This client only needs to be created
+    // once, and can be reused for multiple requests.
+    try (DisksClient disksClient = DisksClient.create()) {
+      AddResourcePoliciesDiskRequest request =
+              AddResourcePoliciesDiskRequest.newBuilder()
+                      .setDisk(diskName)
+                      .setZone(diskLocation)
+                      .setProject(project)
+                      .setDisksAddResourcePoliciesRequestResource(
+                              DisksAddResourcePoliciesRequest.newBuilder()
+                                      .addAllResourcePolicies(Arrays.asList(consistencyGroupUrl))
+                                      .build())
+                      .build();
+      Operation response = disksClient.addResourcePoliciesAsync(request).get(1, TimeUnit.MINUTES);
+
+      if (response.hasError()) {
+        throw new Error("Error adding disk to consistency group! " + response.getError());
+      }
+      return response.getStatus();
+    }
+  }
+}
+// [END compute_consistency_group_add_zonal_disk]

compute/cloud-client/src/main/java/compute/disks/consistencygroup/CloneRegionalDisksFromConsistencyGroup.java

@@ -0,0 +1,73 @@
+/*
+ * Copyright 2024 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package compute.disks.consistencygroup;
+
+// [START compute_consistency_group_clone_regional_disk]
+import com.google.cloud.compute.v1.BulkInsertDiskResource;
+import com.google.cloud.compute.v1.BulkInsertRegionDiskRequest;
+import com.google.cloud.compute.v1.Operation;
+import com.google.cloud.compute.v1.Operation.Status;
+import com.google.cloud.compute.v1.RegionDisksClient;
+import java.io.IOException;
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.TimeoutException;
+
+public class CloneRegionalDisksFromConsistencyGroup {
+
+  public static void main(String[] args)
+      throws IOException, ExecutionException, InterruptedException, TimeoutException {
+    // TODO(developer): Replace these variables before running the sample.
+    // Project ID or project number of the Cloud project you want to use.
+    String project = "YOUR_PROJECT_ID";
+    // Region in which your disks and consistency group are located.
+    String region = "us-central1";
+    // Name of the consistency group you want to clone disks from.
+    String consistencyGroupName = "YOUR_CONSISTENCY_GROUP_NAME";
+
+    cloneRegionalDisksFromConsistencyGroup(project, region, consistencyGroupName);
+  }
+
+  // Clones regional disks from a consistency group.
+  public static Status cloneRegionalDisksFromConsistencyGroup(
+          String project, String region, String consistencyGroupName)
+      throws IOException, InterruptedException, ExecutionException, TimeoutException {
+    String sourceConsistencyGroupPolicy = String.format(
+            "projects/%s/regions/%s/resourcePolicies/%s", project, region, consistencyGroupName);
+
+    // Initialize client that will be used to send requests. This client only needs to be created
+    // once, and can be reused for multiple requests.
+    try (RegionDisksClient disksClient = RegionDisksClient.create()) {
+      BulkInsertRegionDiskRequest request = BulkInsertRegionDiskRequest.newBuilder()
+              .setProject(project)
+              .setRegion(region)
+              .setBulkInsertDiskResourceResource(
+                      BulkInsertDiskResource.newBuilder()
+                              .setSourceConsistencyGroupPolicy(sourceConsistencyGroupPolicy)
+                              .build())
+              .build();
+
+      Operation response = disksClient.bulkInsertAsync(request).get(3, TimeUnit.MINUTES);
+
+      if (response.hasError()) {
+        throw new Error("Error cloning regional disks! " + response.getError());
+      }
+      return response.getStatus();
+    }
+  }
+}
+// [END compute_consistency_group_clone_regional_disk]