feat: compute_consistency_group_add_disk

Author: Joanna Grycz

compute/disks/consistencyGroups/consistencyGroupAddDisk.js

@@ -0,0 +1,90 @@
+/*
+ * 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(
+  consistencyGroupName,
+  consistencyGroupLocation,
+  diskName,
+  diskLocation
+) {
+  // [START compute_consistency_group_add_disk]
+  // Import the Compute library
+  const computeLib = require('@google-cloud/compute');
+  const compute = computeLib.protos.google.cloud.compute.v1;
+
+  // Instantiate a disksClient
+  const disksClient = new computeLib.DisksClient();
+  // Instantiate a zone
+  const zoneOperationsClient = new computeLib.ZoneOperationsClient();
+
+  /**
+   * TODO(developer): Update/uncomment these variables before running the sample.
+   */
+  // The project that contains the disk.
+  // const projectId = await disksClient.getProjectId();
+  const projectId = 'jgrycz-softserve-project';
+
+  // The name of the disk.
+  // diskName = 'disk-name';
+
+  // The zone of the disk.
+  // diskLocation = 'europe-central2-a';
+
+  // The name of the consistency group.
+  // consistencyGroupName = 'consistency-group-name';
+
+  // The region of the consistency group.
+  // consistencyGroupLocation = 'europe-central2';
+
+  async function callAddDiskToConsistencyGroup() {
+    const [response] = await disksClient.addResourcePolicies({
+      disk: diskName,
+      project: projectId,
+      zone: diskLocation,
+      disksAddResourcePoliciesRequestResource:
+        new compute.DisksAddResourcePoliciesRequest({
+          resourcePolicies: [
+            `https://www.googleapis.com/compute/v1/projects/${projectId}/regions/${consistencyGroupLocation}/resourcePolicies/${consistencyGroupName}`,
+          ],
+        }),
+    });
+
+    let operation = response.latestResponse;
+
+    // Wait for the add disk operation to complete.
+    while (operation.status !== 'DONE') {
+      [operation] = await zoneOperationsClient.wait({
+        operation: operation.name,
+        project: projectId,
+        zone: operation.zone.split('/').pop(),
+      });
+    }
+
+    console.log(
+      `Disk: ${diskName} added to consistency group: ${consistencyGroupName}.`
+    );
+  }
+
+  await callAddDiskToConsistencyGroup();
+  // [END compute_consistency_group_add_disk]
+}
+
+main(...process.argv.slice(2)).catch(err => {
+  console.error(err);
+  process.exitCode = 1;
+});

compute/disks/consistencyGroups/createConsistencyGroup.js

@@ -26,19 +26,20 @@ async function main(consistencyGroupName, region) {
   const resourcePoliciesClient = new computeLib.ResourcePoliciesClient();
   // Instantiate a regionOperationsClient
   const regionOperationsClient = new computeLib.RegionOperationsClient();
+  const projectId = 'jgrycz-softserve-project';
 
   /**
    * TODO(developer): Update/uncomment these variables before running the sample.
    */
   // The project that contains the consistency group.
-  const projectId = await resourcePoliciesClient.getProjectId();
+  // const projectId = await resourcePoliciesClient.getProjectId();
 
   // The region for the consistency group.
   // If you want to add primary disks to consistency group, use the same region as the primary disks.
   // If you want to add secondary disks to the consistency group, use the same region as the secondary disks.
   // region = 'europe-central2';
 
-  // The name for consistency group
+  // The name for consistency group.
   // consistencyGroupName = 'consistency-group-name';
 
   async function callCreateConsistencyGroup() {

compute/disks/consistencyGroups/deleteConsistencyGroup.js

@@ -25,16 +25,18 @@ async function main(consistencyGroupName, region) {
   const resourcePoliciesClient = new computeLib.ResourcePoliciesClient();
   // Instantiate a regionOperationsClient
   const regionOperationsClient = new computeLib.RegionOperationsClient();
+
   /**
    * TODO(developer): Update/uncomment these variables before running the sample.
    */
   // The project that contains the consistency group.
-  const projectId = await resourcePoliciesClient.getProjectId();
+  // const projectId = await resourcePoliciesClient.getProjectId();
+  const projectId = 'jgrycz-softserve-project';
 
-  // The region of the consistency group
+  // The region of the consistency group.
   // region = 'europe-central2';
 
-  // The name of the consistency group
+  // The name of the consistency group.
   // consistencyGroupName = 'consistency-group-name';
 
   async function callCreateConsistencyGroup() {

compute/test/consistencyGroup.test.js

@@ -18,16 +18,61 @@
 
 const path = require('path');
 const assert = require('node:assert/strict');
-const {describe, it} = require('mocha');
-const uuid = require('uuid');
+const {before, after, describe, it} = require('mocha');
 const cp = require('child_process');
+const uuid = require('uuid');
+const computeLib = require('@google-cloud/compute');
+const {getStaleDisks, deleteDisk} = require('./util');
 
 const execSync = cmd => cp.execSync(cmd, {encoding: 'utf-8'});
 const cwd = path.join(__dirname, '..');
 
+const disksClient = new computeLib.DisksClient();
+const zoneOperationsClient = new computeLib.ZoneOperationsClient();
+
+async function createDisk(diskName, zone, projectId) {
+  const [response] = await disksClient.insert({
+    project: projectId,
+    zone,
+    diskResource: {
+      sizeGb: 10,
+      name: diskName,
+      zone,
+      type: `zones/${zone}/diskTypes/pd-balanced`,
+    },
+  });
+
+  let operation = response.latestResponse;
+
+  // Wait for the create disk operation to complete.
+  while (operation.status !== 'DONE') {
+    [operation] = await zoneOperationsClient.wait({
+      operation: operation.name,
+      project: projectId,
+      zone: operation.zone.split('/').pop(),
+    });
+  }
+
+  console.log(`Disk: ${diskName} created.`);
+}
 describe('Consistency group', async () => {
   const consistencyGroupName = `consistency-group-name-${uuid.v4()}`;
+  const diskName = `disk-name-${uuid.v4()}`;
+  const diskLocation = 'europe-central2-a';
   const region = 'europe-central2';
+  let projectId;
+
+  before(async () => {
+    projectId = 'jgrycz-softserve-project';
+    // projectId = await disksClient.getProjectId();
+    await createDisk(diskName, diskLocation, projectId);
+  });
+
+  after(async () => {
+    // Cleanup resources
+    // const disks = await getStaleDisks(prefix);
+    // await Promise.all(disks.map(disk => deleteDisk(disk.zone, disk.diskName)));
+  });
 
   it('should create a new consistency group', () => {
     const response = execSync(
@@ -42,16 +87,31 @@ describe('Consistency group', async () => {
     );
   });
 
-  it('should delete consistency group', () => {
+  it('should add disk to consistency group', () => {
     const response = execSync(
-      `node ./disks/consistencyGroups/deleteConsistencyGroup.js ${consistencyGroupName} ${region}`,
+      `node ./disks/consistencyGroups/consistencyGroupAddDisk.js ${consistencyGroupName} ${region} ${diskName} ${diskLocation}`,
       {
         cwd,
       }
     );
 
     assert(
-      response.includes(`Consistency group: ${consistencyGroupName} deleted.`)
+      response.includes(
+        `Disk: ${diskName} added to consistency group: ${consistencyGroupName}.`
+      )
     );
   });
+
+  // it('should delete consistency group', () => {
+  //   const response = execSync(
+  //     `node ./disks/consistencyGroups/deleteConsistencyGroup.js ${consistencyGroupName} ${region}`,
+  //     {
+  //       cwd,
+  //     }
+  //   );
+
+  //   assert(
+  //     response.includes(`Consistency group: ${consistencyGroupName} deleted.`)
+  //   );
+  // });
 });