package model import ( "bytes" "encoding/json" "fmt" sq "github.com/Masterminds/squirrel" "net/http" "time" ) const serverMonitorTable = "`server_monitor`" // ServerMonitor - type ServerMonitor struct { ID int64 `json:"id"` ServerID int64 `json:"serverId"` Item string `json:"item"` Formula string `json:"formula"` Operator string `json:"operator"` Value string `json:"value"` GroupCycle int `json:"groupCycle"` LastCycle int `json:"lastCycle"` SilentCycle int `json:"silentCycle"` StartTime string `json:"startTime"` EndTime string `json:"endTime"` NotifyType uint8 `json:"notifyType"` NotifyTarget string `json:"notifyTarget"` InsertTime string `json:"insertTime"` UpdateTime string `json:"updateTime"` } // ServerMonitors - type ServerMonitors []ServerMonitor // GetAll - func (sm ServerMonitor) GetAll() (ServerMonitors, error) { rows, err := sq. Select( "id", "server_id", "item", "formula", "operator", "value", "group_cycle", "last_cycle", "silent_cycle", "start_time", "end_time", "notify_type", "notify_target", "insert_time", "update_time", ). From(serverMonitorTable). Where(sq.Eq{"server_id": sm.ServerID}). OrderBy("id DESC"). RunWith(DB). Query() if err != nil { return nil, err } serverMonitors := ServerMonitors{} for rows.Next() { var serverMonitor ServerMonitor if err := rows.Scan( &serverMonitor.ID, &serverMonitor.ServerID, &serverMonitor.Item, &serverMonitor.Formula, &serverMonitor.Operator, &serverMonitor.Value, &serverMonitor.GroupCycle, &serverMonitor.LastCycle, &serverMonitor.SilentCycle, &serverMonitor.StartTime, &serverMonitor.EndTime, &serverMonitor.NotifyType, &serverMonitor.NotifyTarget, &serverMonitor.InsertTime, &serverMonitor.UpdateTime, ); err != nil { return nil, err } serverMonitors = append(serverMonitors, serverMonitor) } return serverMonitors, nil } // GetAllModBy - func (sm ServerMonitor) GetAllModBy(number int, time string) (ServerMonitors, error) { rows, err := sq. Select( "id", "server_id", "item", "formula", "operator", "value", "group_cycle", "last_cycle", "silent_cycle", "start_time", "end_time", "notify_type", "notify_target", ). From(serverMonitorTable). Where("? % `group_cycle` = 0", number). Where(sq.GtOrEq{"end_time": time}). Where(sq.LtOrEq{"start_time": time}). OrderBy("id DESC"). RunWith(DB). Query() if err != nil { return nil, err } serverMonitors := ServerMonitors{} for rows.Next() { var serverMonitor ServerMonitor if err := rows.Scan( &serverMonitor.ID, &serverMonitor.ServerID, &serverMonitor.Item, &serverMonitor.Formula, &serverMonitor.Operator, &serverMonitor.Value, &serverMonitor.GroupCycle, &serverMonitor.LastCycle, &serverMonitor.SilentCycle, &serverMonitor.StartTime, &serverMonitor.EndTime, &serverMonitor.NotifyType, &serverMonitor.NotifyTarget, ); err != nil { return nil, err } serverMonitors = append(serverMonitors, serverMonitor) } return serverMonitors, nil } // AddRow return LastInsertId func (sm ServerMonitor) AddRow() (int64, error) { result, err := sq. Insert(serverMonitorTable). Columns( "server_id", "item", "formula", "operator", "value", "group_cycle", "last_cycle", "silent_cycle", "start_time", "end_time", "notify_type", "notify_target", ). Values( sm.ServerID, sm.Item, sm.Formula, sm.Operator, sm.Value, sm.GroupCycle, sm.LastCycle, sm.SilentCycle, sm.StartTime, sm.EndTime, sm.NotifyType, sm.NotifyTarget, ). RunWith(DB). Exec() if err != nil { return 0, err } id, err := result.LastInsertId() return id, err } // EditRow - func (sm ServerMonitor) EditRow() error { _, err := sq. Update(serverMonitorTable). SetMap(sq.Eq{ "item": sm.Item, "formula": sm.Formula, "operator": sm.Operator, "value": sm.Value, "group_cycle": sm.GroupCycle, "last_cycle": sm.LastCycle, "silent_cycle": sm.SilentCycle, "start_time": sm.StartTime, "end_time": sm.EndTime, "notify_type": sm.NotifyType, "notify_target": sm.NotifyTarget, }). Where(sq.Eq{"id": sm.ID}). RunWith(DB). Exec() return err } // DeleteRow - func (sm ServerMonitor) DeleteRow() error { _, err := sq. Delete(serverMonitorTable). Where(sq.Eq{"id": sm.ID}). RunWith(DB). Exec() return err } func (sm ServerMonitor) Notify(server Server, cycleValue string) { if sm.NotifyType == NotifyWeiXin { type markdown struct { Content string `json:"content"` } type message struct { Msgtype string `json:"msgtype"` Markdown markdown `json:"markdown"` } content := fmt.Sprintf("Server: %s(%s)\n ", server.Name, server.Description) content += "Item: " + sm.Item + " warning\n " content += fmt.Sprintf("Event: %s value: %s, %s %s \n ", sm.Formula, cycleValue, sm.Operator, sm.Value) msg := message{ Msgtype: "markdown", Markdown: markdown{ Content: content, }, } b, _ := json.Marshal(msg) _, _ = http.Post(sm.NotifyTarget, "application/json", bytes.NewBuffer(b)) } else if sm.NotifyType == NotifyDingTalk { type markdown struct { Title string `json:"title"` Text string `json:"text"` } type message struct { Msgtype string `json:"msgtype"` Markdown markdown `json:"markdown"` } content := fmt.Sprintf("Server: %s(%s)\n ", server.Name, server.Description) content += "Item: " + sm.Item + "\n " content += fmt.Sprintf("Event: %s value: %s, %s %s \n ", sm.Formula, cycleValue, sm.Operator, sm.Value) msg := message{ Msgtype: "markdown", Markdown: markdown{ Title: fmt.Sprintf("%s %s warning", server.Name, sm.Item), Text: content, }, } b, _ := json.Marshal(msg) _, _ = http.Post(sm.NotifyTarget, "application/json", bytes.NewBuffer(b)) } else if sm.NotifyType == NotifyFeiShu { type message struct { Title string `json:"title"` Text string `json:"text"` } content := fmt.Sprintf("Server: %s(%s)\n ", server.Name, server.Description) content += "Item: " + sm.Item + "\n " content += fmt.Sprintf("Event: %s value: %s, %s %s \n ", sm.Formula, cycleValue, sm.Operator, sm.Value) msg := message{ Title: fmt.Sprintf("%s %s warning", server.Name, sm.Item), Text: content, } b, _ := json.Marshal(msg) _, _ = http.Post(sm.NotifyTarget, "application/json", bytes.NewBuffer(b)) } else if sm.NotifyType == NotifyCustom { type message struct { Code int `json:"code"` Message string `json:"message"` Data struct { Server Server `json:"server"` MonitorRule ServerMonitor `json:"monitorRule"` Value string `json:"value"` Time string `json:"time"` } `json:"data"` } code := 0 msg := message{ Code: code, Message: fmt.Sprintf("%s %s warning", server.Name, sm.Item), } msg.Data.Server = server msg.Data.MonitorRule = sm msg.Data.Value = cycleValue msg.Data.Time = time.Now().Format("2006-01-02 15:04:05") b, _ := json.Marshal(msg) _, _ = http.Post(sm.NotifyTarget, "application/json", bytes.NewBuffer(b)) } }