Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(compute): add compute disk create secondary custom sample #9644

Merged
merged 9 commits into from
Dec 30, 2024
Original file line number Diff line number Diff line change
@@ -0,0 +1,111 @@
/*
* 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;

//[START compute_disk_create_secondary_custom]
import com.google.cloud.compute.v1.Disk;
import com.google.cloud.compute.v1.DiskAsyncReplication;
import com.google.cloud.compute.v1.DisksClient;
import com.google.cloud.compute.v1.GuestOsFeature;
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.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.TimeoutException;

public class CreateSecondaryCustomDisk {
public static void main(String[] args)
throws IOException, ExecutionException, InterruptedException, TimeoutException {
// TODO(developer): Replace these variables before running the sample.
// The project that contains the primary disk.
String primaryProjectId = "PRIMARY_PROJECT_ID";
// The project that contains the secondary disk.
String secondaryProjectId = "SECONDARY_PROJECT_ID";
// Name of the primary disk you want to use.
String primaryDiskName = "PRIMARY_DISK_NAME";
// Name of the zone in which your primary disk is located.
// Learn more about zones and regions:
// https://cloud.google.com/compute/docs/disks/async-pd/about#supported_region_pairs
String primaryDiskZone = "us-central1-a";
// Name of the disk you want to create.
String secondaryDiskName = "SECONDARY_DISK_NAME";
// Name of the zone in which you want to create the secondary disk.
String secondaryDiskZone = "us-east1-c";
// Size of the new disk in gigabytes.
long diskSizeGb = 30L;
// The type of the disk you want to create. This value uses the following format:
// "projects/{projectId}/zones/{zone}/diskTypes/
// (pd-standard|pd-ssd|pd-balanced|pd-extreme)".
String diskType = String.format(
"projects/%s/zones/%s/diskTypes/pd-balanced", secondaryProjectId, secondaryDiskZone);

createSecondaryCustomDisk(primaryProjectId, secondaryProjectId, primaryDiskName,
secondaryDiskName, primaryDiskZone, secondaryDiskZone, diskSizeGb, diskType);
}

// Creates a secondary disk with specified custom parameters.
public static Status createSecondaryCustomDisk(String primaryProjectId, String secondaryProjectId,
String primaryDiskName, String secondaryDiskName, String primaryDiskZone,
String secondaryDiskZone, long diskSizeGb, String diskType)
throws IOException, ExecutionException, InterruptedException, TimeoutException {
// 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()) {
String primaryDiskSource = String.format("projects/%s/zones/%s/disks/%s",
primaryProjectId, primaryDiskZone, primaryDiskName);

DiskAsyncReplication asyncReplication = DiskAsyncReplication.newBuilder()
.setDisk(primaryDiskSource)
.build();

// Define the guest OS features.
List<GuestOsFeature> guestOsFeatures = Arrays.asList(
GuestOsFeature.newBuilder().setType("UEFI_COMPATIBLE").build(),
GuestOsFeature.newBuilder().setType("GVNIC").build(),
GuestOsFeature.newBuilder().setType("MULTI_IP_SUBNET").build());

// Define the labels.
Map<String, String> labels = new HashMap<>();
labels.put("secondary-disk-for-replication", "yes");

Disk disk = Disk.newBuilder()
.setName(secondaryDiskName)
.setSizeGb(diskSizeGb)
.setType(diskType)
.setZone(secondaryDiskZone)
.addAllGuestOsFeatures(guestOsFeatures)
.putAllLabels(labels)
.setAsyncPrimaryDisk(asyncReplication)
.build();

// Wait for the create disk operation to complete.
Operation response = disksClient.insertAsync(secondaryProjectId, secondaryDiskZone, disk)
.get(3, TimeUnit.MINUTES);

if (response.hasError()) {
throw new Error("Error creating secondary custom disks! " + response.getError());
}
return response.getStatus();
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Return the operation response.

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Returned Operation.Status as was done for other samples

}
}
}
// [END compute_disk_create_secondary_custom]
23 changes: 21 additions & 2 deletions compute/cloud-client/src/test/java/compute/disks/DisksIT.java
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,7 @@
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.PrintStream;
import java.lang.Error;
import java.util.Arrays;
import java.util.List;
import java.util.Optional;
Expand Down Expand Up @@ -79,6 +80,8 @@ public class DisksIT {
private static String SECONDARY_REGIONAL_DISK;
private static String SECONDARY_DISK;
private static final long DISK_SIZE = 10L;
private static String SECONDARY_CUSTOM_DISK;

private ByteArrayOutputStream stdOut;

// Check if the required environment variables are set.
Expand Down Expand Up @@ -109,6 +112,7 @@ public static void setup()
REGIONAL_REPLICATED_DISK = "gcloud-test-disk-replicated-" + uuid;
SECONDARY_REGIONAL_DISK = "gcloud-test-disk-secondary-regional-" + uuid;
SECONDARY_DISK = "gcloud-test-disk-secondary-" + uuid;
SECONDARY_CUSTOM_DISK = "gcloud-test-disk-custom-" + uuid;

// Cleanup existing stale resources.
Util.cleanUpExistingInstances("test-disks", PROJECT_ID, ZONE);
Expand Down Expand Up @@ -186,6 +190,7 @@ public static void cleanUp()
RegionalDelete.deleteRegionalDisk(PROJECT_ID, REGION, REGIONAL_REPLICATED_DISK);
RegionalDelete.deleteRegionalDisk(PROJECT_ID, "us-central1", SECONDARY_REGIONAL_DISK);
DeleteDisk.deleteDisk(PROJECT_ID, "us-central1-c", SECONDARY_DISK);
DeleteDisk.deleteDisk(PROJECT_ID, "us-central1-c", SECONDARY_CUSTOM_DISK);

stdOut.close();
System.setOut(out);
Expand Down Expand Up @@ -343,14 +348,28 @@ public void testCreateDiskSecondaryRegional()
@Test
public void testCreateDiskSecondaryZonal()
throws IOException, ExecutionException, InterruptedException, TimeoutException {
String diskType = String.format(
String diskType = String.format(
"projects/%s/zones/%s/diskTypes/pd-ssd", PROJECT_ID, ZONE);
Status status = CreateDiskSecondaryZonal.createDiskSecondaryZonal(
PROJECT_ID, PROJECT_ID, EMPTY_DISK_NAME, SECONDARY_DISK, ZONE,
"us-central1-c", DISK_SIZE, diskType);
"us-central1-c", DISK_SIZE, diskType);
Disk disk = Util.getDisk(PROJECT_ID, "us-central1-c", SECONDARY_DISK);

assertThat(status).isEqualTo(Status.DONE);
assertEquals(SECONDARY_DISK, disk.getName());
}

@Test
public void testCreateSecondaryCustomDisk()
throws IOException, ExecutionException, InterruptedException, TimeoutException {
String diskType = String.format(
"projects/%s/zones/%s/diskTypes/pd-ssd", PROJECT_ID, ZONE);
Status status = CreateSecondaryCustomDisk.createSecondaryCustomDisk(
PROJECT_ID, PROJECT_ID, EMPTY_DISK_NAME, SECONDARY_CUSTOM_DISK, ZONE,
"us-central1-c", DISK_SIZE, diskType);
Disk disk = Util.getDisk(PROJECT_ID, "us-central1-c", SECONDARY_CUSTOM_DISK);

assertThat(status).isEqualTo(Status.DONE);
assertEquals(SECONDARY_CUSTOM_DISK, disk.getName());
}
}
Loading