feat: compute_instance_attach_regional_disk_force

Author: Joanna Grycz

compute/disks/attachRegionalDiskForce.js

@@ -0,0 +1,82 @@
+/*
+ * 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_force]
+  // 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 callForceAttachRegionalDisk() {
+    const [response] = await instancesClient.attachDisk({
+      instance: vmName,
+      project: projectId,
+      attachedDiskResource: new compute.AttachedDisk({
+        source: `projects/${projectId}/regions/${region}/disks/${diskName}`,
+        forceAttach: true,
+      }),
+      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} was forced to be attached to VM: ${vmName}.`
+    );
+  }
+
+  await callForceAttachRegionalDisk();
+  // [END compute_instance_attach_regional_disk_force]
+}
+
+main(...process.argv.slice(2)).catch(err => {
+  console.error(err);
+  process.exitCode = 1;
+});

compute/test/replicatedDisk.test.js

@@ -51,7 +51,8 @@ async function deleteDisk(projectId, region, diskName) {
 
 describe('Create compute regional replicated disk', async () => {
   const diskName = `replicated-disk-${uuid.v4()}`;
-  const vmName = `vm-with-replicated-disk-${uuid.v4()}`;
+  const vmName1 = `vm1-with-replicated-disk-${uuid.v4()}`;
+  const vmName2 = `vm2-with-replicated-disk-${uuid.v4()}`;
   const region = 'europe-central2';
   const zone1 = 'europe-central2-a';
   const zone2 = 'europe-central2-b';
@@ -64,8 +65,10 @@ describe('Create compute regional replicated disk', async () => {
 
   after(async () => {
     // Cleanup resources
-    execSync(`node ./deleteInstance.js ${projectId} ${zone1} ${vmName}`, {
-      cwd,
+    [vmName1, vmName2].forEach(vm => {
+      execSync(`node ./deleteInstance.js ${projectId} ${zone1} ${vm}`, {
+        cwd,
+      });
     });
     await deleteDisk(projectId, region, diskName);
   });
@@ -84,22 +87,45 @@ describe('Create compute regional replicated disk', async () => {
   it('should attach replicated disk to vm', () => {
     // Create VM, where replicated disk will be attached.
     execSync(
-      `node ./createInstance.js ${projectId} ${zone1} ${vmName} e2-small`,
+      `node ./createInstance.js ${projectId} ${zone1} ${vmName1} e2-small`,
+      {
+        cwd,
+      }
+    );
+
+    const response = execSync(
+      `node ./disks/attachRegionalDisk.js ${diskName} ${region} ${vmName1} ${zone1}`,
+      {
+        cwd,
+      }
+    );
+
+    assert(
+      response.includes(
+        `Replicated disk: ${diskName} attached to VM: ${vmName1}.`
+      )
+    );
+  });
+
+  it('should force replicated disk to be attached to vm', () => {
+    // Create VM, where replicated disk will be attached.
+    execSync(
+      `node ./createInstance.js ${projectId} ${zone1} ${vmName2} e2-small`,
       {
         cwd,
       }
     );
 
     const response = execSync(
-      `node ./disks/attachRegionalDisk.js ${diskName} ${region} ${vmName} ${zone1}`,
+      `node ./disks/attachRegionalDiskForce.js ${diskName} ${region} ${vmName2} ${zone1}`,
       {
         cwd,
       }
     );
 
     assert(
       response.includes(
-        `Replicated disk: ${diskName} attached to VM: ${vmName}.`
+        `Replicated disk: ${diskName} was forced to be attached to VM: ${vmName2}.`
       )
     );
   });