feat: compute_instance_attach_regional_disk

Author: Joanna Grycz

compute/disks/attachRegionalDisk.js

@@ -0,0 +1,79 @@
+/*
+ * 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
+ *
+ *     https://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.
+ */
+
+'use strict';
+
+async function main(diskName, region, vmName, zone) {
+  // [START compute_instance_attach_regional_disk]
+  // Import the Compute library
+  const computeLib = require('@google-cloud/compute');
+  const compute = computeLib.protos.google.cloud.compute.v1;
+
+  // Instantiate an instancesClient
+  const instancesClient = new computeLib.InstancesClient();
+  // Instantiate a zoneOperationsClient
+  const zoneOperationsClient = new computeLib.ZoneOperationsClient();
+
+  /**
+   * TODO(developer): Update/uncomment these variables before running the sample.
+   */
+  // Your project ID.
+  const projectId = await instancesClient.getProjectId();
+
+  // The zone of your VM.
+  // zone = 'europe-central2-a';
+
+  // The name of the VM to which you're adding the new replicated disk.
+  // vmName = 'vm-name';
+
+  // The name of the replicated disk
+  // diskName = 'disk-name';
+
+  // The region where the replicated disk is located.
+  // region = 'europe-central2';
+
+  async function callAttachRegionalDisk() {
+    const [response] = await instancesClient.attachDisk({
+      instance: vmName,
+      project: projectId,
+      attachedDiskResource: new compute.AttachedDisk({
+        source: `projects/${projectId}/regions/${region}/disks/${diskName}`,
+      }),
+      zone,
+    });
+
+    let operation = response.latestResponse;
+
+    // Wait for the operation to complete.
+    while (operation.status !== 'DONE') {
+      [operation] = await zoneOperationsClient.wait({
+        operation: operation.name,
+        project: projectId,
+        zone: operation.zone.split('/').pop(),
+      });
+    }
+
+    console.log(`Replicated disk: ${diskName} attached to VM: ${vmName}.`);
+  }
+
+  await callAttachRegionalDisk();
+  // [END compute_instance_attach_regional_disk]
+}
+
+main(...process.argv.slice(2)).catch(err => {
+  console.error(err);
+  process.exitCode = 1;
+});

compute/disks/createRegionalReplicatedDisk.js

@@ -42,12 +42,11 @@ async function main(diskName, region, zone1, zone2) {
   // zone2 = 'europe-central2-b';
 
   // The name of the new disk.
-  // diskName = 'primary-disk-name';
+  // diskName = 'disk-name';
 
   // The type of replicated disk.
   // The default value is `pd-standard`. For Hyperdisk, specify the value `hyperdisk-balanced-high-availability`.
   const diskType = `regions/${region}/diskTypes/pd-standard`;
-  // const diskType = 'pd-standard';
 
   // The size of the new disk in gigabytes.
   const diskSizeGb = 200;

compute/test/createRegionalReplicatedDisk.test.js renamed to compute/test/replicatedDisk.test.js

@@ -19,17 +19,16 @@
 const path = require('path');
 const assert = require('node:assert/strict');
 const uuid = require('uuid');
-const {after, describe, it} = require('mocha');
+const {after, before, describe, it} = require('mocha');
 const cp = require('child_process');
 const computeLib = require('@google-cloud/compute');
 
 const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 const cwd = path.join(__dirname, '..');
 
-async function deleteDisk(region, diskName) {
+async function deleteDisk(projectId, region, diskName) {
   const disksClient = new computeLib.RegionDisksClient();
   const regionOperationsClient = new computeLib.RegionOperationsClient();
-  const projectId = await disksClient.getProjectId();
 
   const [response] = await disksClient.delete({
     project: projectId,
@@ -52,13 +51,23 @@ async function deleteDisk(region, diskName) {
 
 describe('Create compute regional replicated disk', async () => {
   const diskName = `replicated-disk-${uuid.v4()}`;
+  const vmName = `vm-with-replicated-disk-${uuid.v4()}`;
   const region = 'europe-central2';
   const zone1 = 'europe-central2-a';
   const zone2 = 'europe-central2-b';
+  let projectId;
+
+  before(async () => {
+    const instancesClient = new computeLib.InstancesClient();
+    projectId = await instancesClient.getProjectId();
+  });
 
   after(async () => {
     // Cleanup resources
-    await deleteDisk(region, diskName);
+    execSync(`node ./deleteInstance.js ${projectId} ${zone1} ${vmName}`, {
+      cwd,
+    });
+    await deleteDisk(projectId, region, diskName);
   });
 
   it('should create a regional replicated disk', () => {
@@ -71,4 +80,27 @@ describe('Create compute regional replicated disk', async () => {
 
     assert(response.includes(`Regional replicated disk: ${diskName} created.`));
   });
+
+  it('should attach replicated disk to vm', () => {
+    // Create VM, where replicated disk will be attached.
+    execSync(
+      `node ./createInstance.js ${projectId} ${zone1} ${vmName} e2-small`,
+      {
+        cwd,
+      }
+    );
+
+    const response = execSync(
+      `node ./disks/attachRegionalDisk.js ${diskName} ${region} ${vmName} ${zone1}`,
+      {
+        cwd,
+      }
+    );
+
+    assert(
+      response.includes(
+        `Replicated disk: ${diskName} attached to VM: ${vmName}.`
+      )
+    );
+  });
 });