✨ feat(notification): 添加通知方式

model/notification.go

@@ -0,0 +1,100 @@
+package model
+
+import (
+	"crypto/tls"
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"net/url"
+	"strings"
+	"time"
+)
+
+const (
+	_ = iota
+	NotificationRequestTypeJSON
+	NotificationRequestTypeForm
+)
+
+const (
+	_ = iota
+	NotificationRequestMethodGET
+	NotificationRequestMethodPOST
+)
+
+type NotificatonSender struct {
+	Rule   *Rule
+	Server *Server
+	State  *State
+}
+
+type Notification struct {
+	Common
+	Name          string
+	URL           string
+	RequestMethod int
+	RequestType   int
+	RequestBody   string `gorm:"type:longtext" `
+	VerifySSL     *bool
+}
+
+func (n *Notification) Send(sender *NotificatonSender, message string) {
+	var verifySSL bool
+
+	if n.VerifySSL != nil && *n.VerifySSL {
+		verifySSL = true
+	}
+
+	var err error
+	transCfg := &http.Transport{
+		TLSClientConfig: &tls.Config{InsecureSkipVerify: verifySSL},
+	}
+	client := &http.Client{Transport: transCfg, Timeout: time.Minute * 10}
+	var reqURL *url.URL
+	reqURL, err = url.Parse(n.URL)
+	var data map[string]string
+	if err == nil && (n.RequestMethod == NotificationRequestMethodGET || n.RequestType == NotificationRequestTypeForm) {
+		err = json.Unmarshal([]byte(n.RequestBody), &data)
+	}
+
+	if err == nil {
+		if n.RequestMethod == NotificationRequestMethodGET {
+			for k, v := range data {
+				reqURL.Query().Set(k, replaceParamsInString(v, sender))
+			}
+			client.Get(reqURL.String())
+		} else {
+			if n.RequestType == NotificationRequestTypeForm {
+				params := url.Values{}
+				for k, v := range data {
+					params.Add(k, replaceParamsInString(v, sender))
+				}
+				client.PostForm(reqURL.String(), params)
+			} else {
+				jsonValue := replaceParamsInJSON(n.RequestBody, sender)
+				if err == nil {
+					client.Post(reqURL.String(), "application/json", strings.NewReader(jsonValue))
+				}
+			}
+		}
+	}
+}
+
+func replaceParamsInString(str string, sender *NotificatonSender) string {
+	str = strings.ReplaceAll(str, "#CPU#", fmt.Sprintf("%2f%%", sender.State.CPU))
+	return str
+}
+
+func replaceParamsInJSON(str string, sender *NotificatonSender) string {
+	str = strings.ReplaceAll(str, "#CPU#", fmt.Sprintf("%2f%%", sender.State.CPU))
+	return str
+}
+
+func jsonEscape(raw interface{}) string {
+	b, _ := json.Marshal(raw)
+	strb := string(b)
+	if strings.HasPrefix(strb, "\"") {
+		return strb[1 : len(strb)-1]
+	}
+	return strb
+}

model/rule.go

@@ -0,0 +1,10 @@
+package model
+
+type Rule struct {
+	Common
+	Name     string
+	Type     string // 指标类型，cpu、memory、swap、disk、net_in、net_out、net_all、transfer_in、transfer_out、transfer_all、offline
+	Min      uint64 // 最小阈值
+	Max      uint64 // 最大阈值
+	Duration uint64 // 持续时间
+}

model/server.go

@@ -2,6 +2,7 @@ package model
 
 import (
 	"fmt"
+	"html/template"
 	"time"
 
 	pb "github.com/naiba/nezha/proto"
@@ -21,6 +22,6 @@ type Server struct {
 	StreamClose chan<- error                    `gorm:"-" json:"-"`
 }
 
-func (s Server) Marshal() string {
-	return fmt.Sprintf(`{"ID":%d,"Name":"%s","Secret":"%s"}`, s.ID, s.Name, s.Secret)
+func (s Server) Marshal() template.JS {
+	return template.JS(fmt.Sprintf(`{"ID":%d,"Name":"%s","Secret":"%s"}`, s.ID, s.Name, s.Secret))
 }