webhook接收告警写入mysql

package main

import (
	"database/sql"
	"encoding/json"
	"fmt"
	"io"
	"math/rand"
	"net/http"
	_ "os"
	"strconv"
	"strings"
	"time"

	"github.com/gin-gonic/gin"
	_ "github.com/go-sql-driver/mysql"
)

type Alert struct {
	Receiver string   `json:"receiver"`
	Status   string   `json:"status"`
	Alerts   []Alerts `json:"alerts"`
}

type Alerts struct {
	Status       string      `json:"status"`
	Labels       Labels      `json:"labels"`
	Annotations  Annotations `json:"annotations"`
	StartsAt     string      `json:"startsAt"`
	EndsAt       string      `json:"endsAt"`
	GeneratorURL string      `json:"generatorURL"`
	Fingerprint  string      `json:"fingerprint"`
}

type Labels struct {
	Alertname    string `json:"name"`
	Device       string `json:"device"`
	FSType       string `json:"fstype"`
	Instance     string `json:"instance"`
	Job          string `json:"job"`
	MountPoint   string `json:"mountpoint"`
	Severity     string `json:"severity"`
	System       string `json:"system"`
	Yunyingshang string `json:"yunyingshang"`
	Judian       string `json:"judian"`
}

type Annotations struct {
	Description string `json:"description"`
	Summary     string `json:"summary"`
}

type Alarm struct {
	IP      string
	System  string
	ISP     string
	Desc    string
	Judian  string
	Summary string
}

func NewAlarm(value Alerts) Alarm {
	isp := value.Labels.Yunyingshang
	if len(isp) == 0 {
		isp = "wa"
	}
	ip := strings.Split(value.Labels.Instance, ":")[0]
	summary := value.Annotations.Summary
	// desc := ip + summary
	system := value.Labels.System
	if len(system) == 0 {
		system = "wa"
	}
	judian := value.Labels.Judian
	if len(judian) == 0 {
		judian = "wa"
	}
	return Alarm{
		IP:     ip,
		System: system,
		ISP:    isp,
		// Desc:   desc,
		Judian:  judian,
		Summary: summary,
	}
}

var onlyalartmID string

func generateAlarmID() (alarmID string) {
	prefix := "wamaster"
	dateStr := time.Now().Format("200601021504")
	rand.Seed(time.Now().UnixNano())
	randStr := strconv.Itoa(rand.Intn(9000) + 1000)
	alarmID = prefix + dateStr + randStr
	return
}

func InsertData(alarm Alarm) error {
	// 这里插入数据

	return nil
}

func insertAlertToDB(db *sql.DB, alert Alert) error {
	// 在这里编写插入告警数据的SQL语句，例如：
	query := `INSERT INTO hookdata (ip, status, alertid, system, isp, judian, summary, tuitime) VALUES (?, ?, ?, ?, ?, ?, ?, ?)`
	stmt, err := db.Prepare(query)
	if err != nil {
		return err
	}
	defer stmt.Close()

	for _, value := range alert.Alerts {
		onlyalartmID = generateAlarmID()
		status := 0
		alarm := NewAlarm(value)
		_ = InsertData(alarm)
		// ipmiaoshu = alarm.IP + alarm.Desc
		currentTime := time.Now().Format("2006-01-02 15:04:05") // 获取当前时间并格式化为字符串
		_, err = stmt.Exec(alarm.IP, status, onlyalartmID, alarm.System, alarm.ISP, alarm.Judian, alarm.Summary, currentTime)
		if err != nil {
			return err
		}
	}
	return nil
}

func main() {
	fmt.Println(generateAlarmID())
	// os.Exit(0)
	r := gin.Default()
	// 连接到MySQL数据库
	db, err := sql.Open("mysql", "root:111111@tcp(localhost:3306)/alert")
	if err != nil {
		panic(err)
	}
	defer db.Close()

	r.POST("/webhook", func(c *gin.Context) {
		body, err := io.ReadAll(c.Request.Body)
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "无法读取请求体"})
			return
		}

		var alert Alert
		err = json.Unmarshal(body, &alert)
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "无法解析告警信息"})
			return
		}
		//打印出告警信息进行调试
		for _, value := range alert.Alerts {
			onlyalartmID = generateAlarmID()
			status := 0
			alarm := NewAlarm(value)
			_ = InsertData(alarm)
			// ipmiaoshu = alarm.IP + alarm.Desc
			currentTime := time.Now().Format("2006-01-02 15:04:05") // 获取当前时间并格式化为字符串
			fmt.Printf("ip: %s\n status: %v\n alertid: %s\n system: %s\n isp: %s\n judian: %s\n summary: %s\n tuitime: %T\n", alarm.IP, status, onlyalartmID, alarm.System, alarm.ISP, alarm.Judian, alarm.Summary, currentTime)
		}

		// 将告警信息插入到数据库
		err = insertAlertToDB(db, alert)
		if err != nil {
			fmt.Println(err)
			c.JSON(http.StatusInternalServerError, gin.H{"error": "无法将告警信息插入到数据库"})
			return
		}
		c.JSON(http.StatusOK, gin.H{"message": "告警已接收并存储到数据库"})
	})

	r.Run(":8080") // 监听8080端口
}