feat: add cynosdb module

Author: yugasun

__tests__/cynos.test.js

@@ -0,0 +1,67 @@
+const { Cynosdb } = require('../src');
+const { getClusterDetail, sleep, generatePwd, PWD_CHARS } = require('../src/modules/cynosdb/utils');
+
+const pwdReg = new RegExp(`[${PWD_CHARS}]{8,64}`);
+
+describe('Cynosdb', () => {
+  jest.setTimeout(600000);
+  const credentials = {
+    SecretId: process.env.TENCENT_SECRET_ID,
+    SecretKey: process.env.TENCENT_SECRET_KEY,
+  };
+  const client = new Cynosdb(credentials, 'ap-guangzhou');
+
+  const inputs = {
+    region: 'ap-guangzhou',
+    zone: 'ap-guangzhou-4',
+    vpcConfig: {
+      vpcId: 'vpc-p2dlmlbj',
+      subnetId: 'subnet-a1v3k07o',
+    },
+  };
+
+  test('[generatePwd] should get random password with default length 8', () => {
+    const res = generatePwd();
+    expect(typeof res).toBe('string');
+    expect(res.length).toBe(8);
+  });
+
+  test('[generatePwd] should get random password with customize length 6', () => {
+    const res = generatePwd(6);
+    expect(typeof res).toBe('string');
+    expect(res.length).toBe(6);
+  });
+
+  test('should deploy Cynosdb success', async () => {
+    const res = await client.deploy(inputs);
+    expect(res).toEqual({
+      region: inputs.region,
+      zone: inputs.zone,
+      vpcConfig: inputs.vpcConfig,
+      instanceCount: 2,
+      adminPassword: expect.stringMatching(pwdReg),
+      clusterId: expect.stringContaining('cynosdbmysql-'),
+      connection: {
+        ip: expect.any(String),
+        port: 3306,
+        readList: [
+          {
+            ip: expect.any(String),
+            port: 3306,
+          },
+        ],
+      },
+    });
+
+    inputs.clusterId = res.clusterId;
+  });
+
+  test('should remove Cynosdb success', async () => {
+    await sleep(1000);
+    const res = await client.remove(inputs);
+
+    const detail = await getClusterDetail(client.capi, inputs.clusterId);
+    expect(res).toEqual(true);
+    expect(detail).toBeUndefined();
+  });
+});

package.json

@@ -74,7 +74,7 @@
     "semantic-release": "^17.0.4"
   },
   "dependencies": {
-    "@tencent-sdk/capi": "^1.1.2",
+    "@tencent-sdk/capi": "^1.1.4",
     "@ygkit/request": "^0.1.1",
     "cos-nodejs-sdk-v5": "^2.6.2",
     "moment": "^2.25.3",

src/index.js

@@ -1,33 +1,18 @@
-const Apigw = require('./modules/apigw');
-const Cdn = require('./modules/cdn');
-const Cns = require('./modules/cns');
-const Cos = require('./modules/cos');
-const Domain = require('./modules/domain');
-const MultiApigw = require('./modules/multi-apigw');
-const MultiScf = require('./modules/multi-scf');
-const Scf = require('./modules/scf');
-const Tag = require('./modules/tag');
-const Postgresql = require('./modules/postgresql');
-const Vpc = require('./modules/vpc');
-const Cam = require('./modules/cam');
-const Metrics = require('./modules/metrics');
-const Layer = require('./modules/layer');
-const Cfs = require('./modules/cfs');
-
 module.exports = {
-  Apigw,
-  Cdn,
-  Cns,
-  Cos,
-  Domain,
-  MultiApigw,
-  MultiScf,
-  Scf,
-  Tag,
-  Postgresql,
-  Vpc,
-  Cam,
-  Metrics,
-  Layer,
-  Cfs,
+  Apigw: require('./modules/apigw'),
+  Cdn: require('./modules/cdn'),
+  Cns: require('./modules/cns'),
+  Cos: require('./modules/cos'),
+  Domain: require('./modules/domain'),
+  MultiApigw: require('./modules/multi-apigw'),
+  MultiScf: require('./modules/multi-scf'),
+  Scf: require('./modules/scf'),
+  Tag: require('./modules/tag'),
+  Postgresql: require('./modules/postgresql'),
+  Vpc: require('./modules/vpc'),
+  Cam: require('./modules/cam'),
+  Metrics: require('./modules/metrics'),
+  Layer: require('./modules/layer'),
+  Cfs: require('./modules/cfs'),
+  Cynosdb: require('./modules/cynosdb'),
 };

src/modules/cynosdb/apis.js

@@ -0,0 +1,18 @@
+const { ApiFactory } = require('../../utils/api');
+
+const ACTIONS = [
+  'CreateClusters',
+  'DescribeClusterDetail',
+  'IsolateCluster',
+  'DescribeAccounts',
+  'ResetAccountPassword',
+];
+
+const APIS = ApiFactory({
+  // debug: true,
+  serviceType: 'cynosdb',
+  version: '2019-01-07',
+  actions: ACTIONS,
+});
+
+module.exports = APIS;

src/modules/cynosdb/index.js

@@ -0,0 +1,124 @@
+const { Capi } = require('@tencent-sdk/capi');
+const {
+  createCluster,
+  getClusterDetail,
+  deleteCluster,
+  generatePwd,
+  formatConnectOutput,
+  resetPwd,
+} = require('./utils');
+const { ApiError } = require('../../utils/error');
+
+class Cynosdb {
+  constructor(credentials = {}, region) {
+    this.region = region || 'ap-guangzhou';
+    this.credentials = credentials;
+    this.capi = new Capi({
+      Region: this.region,
+      AppId: this.credentials.AppId,
+      SecretId: this.credentials.SecretId,
+      SecretKey: this.credentials.SecretKey,
+      Token: this.credentials.Token,
+    });
+  }
+
+  async deploy(inputs = {}) {
+    const {
+      clusterId,
+      region,
+      zone,
+      vpcConfig,
+      projectId = 0,
+      dbVersion = '5.7',
+      dbType = 'MYSQL',
+      port = 3306,
+      cpu = 1,
+      memory = 1,
+      storageLimit = 1000,
+      instanceCount = 2,
+      adminPassword,
+    } = inputs;
+
+    const outputs = {
+      region: region,
+      zone: zone,
+      vpcConfig: vpcConfig,
+      instanceCount,
+    };
+
+    let isExisted = false;
+    let clusterDetail = null;
+    if (clusterId) {
+      clusterDetail = await getClusterDetail(this.capi, clusterId);
+      if (clusterDetail && clusterDetail.ClusterId) {
+        isExisted = true;
+        outputs.clusterId = clusterDetail.ClusterId;
+        if (adminPassword) {
+          outputs.adminPassword = adminPassword;
+        }
+      }
+    }
+    if (!isExisted) {
+      // not exist, create
+      const dbInputs = {
+        Zone: zone,
+        ProjectId: projectId,
+        DbType: dbType,
+        DbVersion: dbVersion,
+        Port: port,
+        Cpu: cpu,
+        Memory: memory,
+        StorageLimit: storageLimit,
+        InstanceCount: instanceCount,
+        PayMode: 0,
+        AutoVoucher: 1,
+        RollbackStrategy: 'noneRollback',
+        OrderSource: 'serverless',
+        VpcId: vpcConfig.vpcId,
+        SubnetId: vpcConfig.subnetId,
+        AdminPassword: adminPassword,
+        VpcId: vpcConfig.vpcId,
+        SubnetId: vpcConfig.subnetId,
+        AdminPassword: adminPassword || generatePwd(),
+      };
+
+      clusterDetail = await createCluster(this.capi, dbInputs);
+      outputs.clusterId = clusterDetail.ClusterId;
+
+      outputs.adminPassword = dbInputs.AdminPassword;
+    }
+
+    outputs.connection = formatConnectOutput(clusterDetail);
+
+    return outputs;
+  }
+
+  async remove(inputs = {}) {
+    const { clusterId } = inputs;
+
+    const clusterDetail = await getClusterDetail(this.capi, clusterId);
+    if (clusterDetail && clusterDetail.ClusterId) {
+      // need circle for deleting, after host status is 6, then we can delete it
+      await deleteCluster(this.capi, clusterId);
+    }
+    return true;
+  }
+
+  async resetPwd(inputs = {}) {
+    const { clusterId } = inputs;
+
+    const clusterDetail = await getClusterDetail(this.capi, clusterId);
+    if (clusterDetail && clusterDetail.ClusterId) {
+      // need circle for deleting, after host status is 6, then we can delete it
+      await resetPwd(this.capi, inputs);
+    } else {
+      throw ApiError({
+        type: 'PARAMETER_CYNOSDB',
+        message: `CynosDB cluster id: ${clusterId} not exist.`,
+      });
+    }
+    return true;
+  }
+}
+
+module.exports = Cynosdb;