SCF+腾讯云API+企业微信机器人实现CDB慢查询提醒

func main() {
	cloudfunction.Start(MainFunc)
}  

{"errorCode":-1,"errorMessage":"Time limit exceeded"}

set GOOS=linux
set GOARCH=amd64
go build -o main main.go

type InstanceResponse struct {
	Response struct {
		TotalCount int `json:"TotalCount"`
		Items      []struct {
			InstanceID    string `json:"InstanceId"`
			ResourceID    string `json:"ResourceId"`
			RegionID      int    `json:"RegionId"`
			RegionName    string `json:"RegionName"`
			QPS           int    `json:"Qps"`
			Region        string `json:"Region"`
			InitFlag      int    `json:"InitFlag"`
			InstanceType  int    `json:"InstanceType"`
			InstanceName  string `json:"InstanceName"`
			Vip           string `json:"Vip"`
			Vport         int    `json:"Vport"`
			WanStatus     int    `json:"WanStatus"`
			WanDomain     string `json:"WanDomain"`
			WanPort       int    `json:"WanPort"`
			Status        int    `json:"Status"`
			CdbError      int    `json:"CdbError"`
			TaskStatus    int    `json:"TaskStatus"`
			EngineVersion string `json:"EngineVersion"`
			CreateTime    string `json:"CreateTime"`
			DeadlineTime  string `json:"DeadlineTime"`
			IsolateTime   string `json:"IsolateTime"`
			DeviceType    string `json:"DeviceType"`
			Memory        int    `json:"Memory"`
			Volume        int    `json:"Volume"`
			CPU           int    `json:"Cpu"`
			AutoRenew     int    `json:"AutoRenew"`
			ZoneID        int    `json:"ZoneId"`
			Zone          string `json:"Zone"`
			ZoneName      string `json:"ZoneName"`
			VpcID         int    `json:"VpcId"`
			SubnetID      int    `json:"SubnetId"`
			UniqVpcID     string `json:"UniqVpcId"`
			UniqSubnetID  string `json:"UniqSubnetId"`
			ProjectID     int    `json:"ProjectId"`
			PayType       int    `json:"PayType"`
			ProtectMode   int    `json:"ProtectMode"`
			DeployMode    int    `json:"DeployMode"`
			SlaveInfo     struct {
				First struct {
					Region string `json:"Region"`
					ZoneID int    `json:"ZoneId"`
					Zone   string `json:"Zone"`
					Vip    string `json:"Vip"`
					Vport  int    `json:"Vport"`
				} `json:"First"`
				Second interface{} `json:"Second"`
			} `json:"SlaveInfo"`
			MasterInfo    interface{}   `json:"MasterInfo"`
			RoInfo        []interface{} `json:"RoInfo"`
			RoGroups      []interface{} `json:"RoGroups"`
			DrInfo        []interface{} `json:"DrInfo"`
			BackupZoneID  int           `json:"BackupZoneId"`
			ExClusterID   string        `json:"ExClusterId"`
			OfflineTime   string        `json:"OfflineTime"`
			HourFeeStatus int           `json:"HourFeeStatus"`
			RoVipInfo     interface{}   `json:"RoVipInfo"`
			PhysicalID    string        `json:"PhysicalId"`
		} `json:"Items"`
		RequestID string `json:"RequestId"`
	} `json:"Response"`
}

type SlowLogResponse struct {
	Response struct {
		TotalCount int `json:"TotalCount"`
		Items      []struct {
			Name        string `json:"Name"`
			IntranetURL string `json:"IntranetUrl"`
			InternetURL string `json:"InternetUrl"`
			Size        int    `json:"Size"`
			Type        string `json:"Type"`
			Date        string `json:"Date"`
		} `json:"Items"`
		RequestID string `json:"RequestId"`
	} `json:"Response"`
}

func main() {
	cloudfunction.Start(MainFunc)
}

func MainFunc() {
	credential := common.NewCredential(
		"",
		"",
	)
	cpf := profile.NewClientProfile()
	cpf.HttpProfile.Endpoint = "cdb.tencentcloudapi.com"
	client, _ := cdb.NewClient(credential, "ap-guangzhou", cpf)

	insRequest := cdb.NewDescribeDBInstancesRequest()
	insParams := `{"Limit":100}`

	err := insRequest.FromJsonString(insParams)
	if err != nil {
		panic(err)
	}

	responseIns, err := client.DescribeDBInstances(insRequest)
	if _, ok := err.(*errors.TencentCloudSDKError); ok {
		fmt.Println(err)
		return
	}
	if err != nil {
		panic(err)
	}

	var response InstanceResponse
	err = json.Unmarshal([]byte(responseIns.ToJsonString()), &response)
	if err != nil {
		fmt.Println(err)
		return
	}

	instanceCount := len(response.Response.Items)

	if instanceCount == 0 {
		fmt.Println("no instance")
		return
	}

	var wg sync.WaitGroup
	wg.Add(instanceCount)

	slowLogChan := make(chan SlowLogResponse, instanceCount)
	for idx := 0; idx < instanceCount; idx++ {
		go SlowLogRequest(response.Response.Items[idx].InstanceID, slowLogChan, &wg)
	}

	wg.Wait()
	close(slowLogChan)

	var msgBuf bytes.Buffer //消息buf
	for res := range slowLogChan {
		if len(res.Response.Items) > 1 && res.Response.Items[1].Size > 0 {
			logName, err := url.QueryUnescape(res.Response.Items[1].Name)
			if err != nil {
				continue
			}

			msgBuf.WriteString(fmt.Sprintf(">Name: %s \\n", logName))
			msgBuf.WriteString(fmt.Sprintf(">Size: %d B\\n", res.Response.Items[1].Size))
			msgBuf.WriteString(fmt.Sprintf("[点击下载](%s) \\n\\n", res.Response.Items[1].InternetURL))
		}
	}

	msgStr := msgBuf.String()
	if msgStr == "<nil>" || msgStr == "" {
		msgStr = "<font color=\"success\">很好~</font>昨天没有产生慢查询日志\\n"
	} else {
		msgStr = "### 昨日CDB慢查询提醒 \\n\\n" + msgStr
	}

	sendMarkdownToUs(msgStr)
	fmt.Println("exit...")
	return
}

func sendMarkdownToUs(msgStr string) {
	param := fmt.Sprintf(`{"msgtype": "markdown",
    "markdown": {
        "content":"%s"
	}}`, msgStr)

	fmt.Println(param)
	resp, err := http.Post("https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=你的企业微信机器人key",
		"application/json",
		strings.NewReader(param))

	if err != nil {
		fmt.Println(err)
		return
	}

	defer resp.Body.Close()
	_, err = ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Println("send msg error")
	}

	return
}

func SlowLogRequest(instanceId string, ch chan SlowLogResponse, wg *sync.WaitGroup) {
	defer func() {
		if err := recover(); err != nil {
			fmt.Println(err)
		}
		wg.Done()
	}()

	credential := common.NewCredential(
		"",
		"",
	)
	cpf := profile.NewClientProfile()
	cpf.HttpProfile.Endpoint = "cdb.tencentcloudapi.com"
	client, _ := cdb.NewClient(credential, "ap-guangzhou", cpf)

	request := cdb.NewDescribeSlowLogsRequest()

	params := fmt.Sprintf(`{"InstanceId":"%s", "Limit":100}`, instanceId)
	err := request.FromJsonString(params)

	if err != nil {
		panic(err)
	}

	response, err := client.DescribeSlowLogs(request)
	if _, ok := err.(*errors.TencentCloudSDKError); ok {
		fmt.Println(err)
		return
	}
	if err != nil {
		panic(err)
	}

	var slowLogResponse SlowLogResponse
	err = json.Unmarshal([]byte(response.ToJsonString()), &slowLogResponse)
	if err != nil {
		fmt.Println(err)
		return
	}

	ch <- slowLogResponse

	return
}