fix(cls): add alarm flow for deploy and remove

Author: yugasun

__tests__/cls/cls.test.ts

@@ -13,6 +13,28 @@ describe('Cls', () => {
 
   let outputs: DeployOutputs;
 
+  const alarms = [
+    {
+      name: 'cls-alarm-test',
+      targets: [
+        {
+          period: 15,
+          query: '* | select count(*) as errCount',
+        },
+      ],
+      monitor: {
+        type: 'Period',
+        time: 1,
+      },
+      trigger: {
+        condition: '$1.count > 1',
+        count: 2,
+        period: 15,
+      },
+      noticeId: 'notice-4271ef11-1b09-459f-8dd1-b0a411757663',
+    },
+  ];
+
   const inputs: DeployInputs = {
     region: 'ap-guangzhou',
     name: 'cls-test',
@@ -32,14 +54,36 @@ describe('Cls', () => {
     },
   };
 
-  test('should deploy cls success', async () => {
+  test('deploy cls', async () => {
+    const res = await client.deploy(inputs);
+    expect(res).toEqual({
+      region: process.env.REGION,
+      name: inputs.name,
+      topic: inputs.topic,
+      logsetId: expect.any(String),
+      topicId: expect.any(String),
+    });
+
+    outputs = res;
+  });
+
+  test('deploy cls with alarms', async () => {
+    inputs.alarms = alarms;
     const res = await client.deploy(inputs);
     expect(res).toEqual({
       region: process.env.REGION,
       name: inputs.name,
       topic: inputs.topic,
       logsetId: expect.any(String),
       topicId: expect.any(String),
+      alarms: [
+        {
+          id: expect.stringContaining('alarm-'),
+          logsetId: expect.any(String),
+          topicId: expect.any(String),
+          ...alarms[0],
+        },
+      ],
     });
 
     outputs = res;

src/modules/cls/alarm.ts

@@ -101,8 +101,7 @@ export default class Alarm {
       } else {
         console.log(`Alarm ${id} not exist`);
       }
-    }
-    if (name) {
+    } else if (name) {
       const detail = await this.get({ name });
       if (detail) {
         await this.request({

src/modules/cls/index.ts

@@ -12,16 +12,19 @@ import {
   RemoveDashboardInputs,
   Dashboard,
   DashboardItem,
+  AlarmInputs,
 } from './interface';
 import { CapiCredentials, RegionType } from './../interface';
 import { ApiError } from '../../utils/error';
 import { dtz, TIME_FORMAT, Dayjs } from '../../utils/dayjs';
 import { createLogset, createTopic, updateIndex, getSearchSql } from './utils';
+import Alarm from './alarm';
 
 export default class Cls {
   credentials: CapiCredentials;
   region: RegionType;
   cls: ClsClient;
+  alarm: Alarm;
 
   constructor(credentials: CapiCredentials, region: RegionType = 'ap-guangzhou', expire?: number) {
     this.region = region;
@@ -34,6 +37,8 @@ export default class Cls {
       debug: false,
       expire: expire,
     });
+
+    this.alarm = new Alarm(credentials, this.region);
   }
 
   // 创建/更新 日志集
@@ -170,11 +175,25 @@ export default class Cls {
     outputs.topicId = inputs.topicId = topicOutput.topicId;
     await this.deployIndex(inputs);
 
+    // 部署告警
+    const { alarms = [] } = inputs;
+    if (alarms.length > 0) {
+      outputs.alarms = [];
+      for (let i = 0, len = alarms.length; i < len; i++) {
+        const res = await this.alarm.create({
+          ...alarms[i],
+          logsetId: outputs.logsetId,
+          topicId: outputs.topicId,
+        });
+        outputs.alarms.push(res);
+      }
+    }
+
     return outputs;
   }
 
   // 删除
-  async remove(inputs: { topicId?: string; logsetId?: string } = {}) {
+  async remove(inputs: { topicId?: string; logsetId?: string; alarms?: AlarmInputs[] } = {}) {
     try {
       console.log(`Start removing cls`);
       console.log(`Removing cls topic id ${inputs.topicId}`);
@@ -198,6 +217,15 @@ export default class Cls {
           message: res2.error.message,
         });
       }
+      const { alarms = [] } = inputs;
+      if (alarms && alarms.length > 0) {
+        for (let i = 0, len = alarms.length; i < len; i++) {
+          const cur = alarms[i];
+          console.log(`Removing alarm name ${cur.name}, id ${cur.id}`);
+          await this.alarm.delete({ id: cur.id, name: cur.name });
+          console.log(`Remove alarm name ${cur.name}, id ${cur.id} success`);
+        }
+      }
       console.log(`Removed cls id ${inputs.logsetId} success`);
     } catch (e) {
       console.log(e);