// Copyright 2022 The Goploy Authors. All rights reserved.
// Use of this source code is governed by a GPLv3-style
// license that can be found in the LICENSE file.
package model
import (
"bytes"
"encoding/json"
"errors"
"io/ioutil"
"net/http"
sq "github.com/Masterminds/squirrel"
)
const monitorTable = "`monitor`"
type Monitor struct {
ID int64 `json:"id"`
NamespaceID int64 `json:"namespaceId"`
Name string `json:"name"`
URL string `json:"url"`
Second int `json:"second"`
Times uint16 `json:"times"`
NotifyType uint8 `json:"notifyType"`
NotifyTarget string `json:"notifyTarget"`
NotifyTimes uint16 `json:"notifyTimes"`
Description string `json:"description"`
ErrorContent string `json:"errorContent"`
State uint8 `json:"state"`
InsertTime string `json:"insertTime"`
UpdateTime string `json:"updateTime"`
}
type Monitors []Monitor
func (m Monitor) GetList() (Monitors, error) {
rows, err := sq.
Select("id, name, url, second, times, notify_type, notify_target, notify_times, description, error_content, state, insert_time, update_time").
From(monitorTable).
Where(sq.Eq{
"namespace_id": m.NamespaceID,
}).
OrderBy("id DESC").
RunWith(DB).
Query()
if err != nil {
return nil, err
}
monitors := Monitors{}
for rows.Next() {
var monitor Monitor
if err := rows.Scan(
&monitor.ID,
&monitor.Name,
&monitor.URL,
&monitor.Second,
&monitor.Times,
&monitor.NotifyType,
&monitor.NotifyTarget,
&monitor.NotifyTimes,
&monitor.Description,
&monitor.ErrorContent,
&monitor.State,
&monitor.InsertTime,
&monitor.UpdateTime); err != nil {
return nil, err
}
monitors = append(monitors, monitor)
}
return monitors, nil
}
func (m Monitor) GetData() (Monitor, error) {
var monitor Monitor
err := sq.
Select("id, name, url, second, times, notify_type, notify_target, notify_times, state").
From(monitorTable).
Where(sq.Eq{"id": m.ID}).
OrderBy("id DESC").
RunWith(DB).
QueryRow().
Scan(&monitor.ID, &monitor.Name, &monitor.URL, &monitor.Second, &monitor.Times, &monitor.NotifyType, &monitor.NotifyTarget, &monitor.NotifyTimes, &monitor.State)
if err != nil {
return monitor, errors.New("数据查询失败")
}
return monitor, nil
}
func (m Monitor) GetAllByState() (Monitors, error) {
rows, err := sq.
Select("id, name, url, second, times, notify_type, notify_target, notify_times, description").
From(monitorTable).
Where(sq.Eq{
"state": m.State,
}).
RunWith(DB).
Query()
if err != nil {
return nil, err
}
monitors := Monitors{}
for rows.Next() {
var monitor Monitor
if err := rows.Scan(
&monitor.ID,
&monitor.Name,
&monitor.URL,
&monitor.Second,
&monitor.Times,
&monitor.NotifyType,
&monitor.NotifyTarget,
&monitor.NotifyTimes,
&monitor.Description); err != nil {
return nil, err
}
monitors = append(monitors, monitor)
}
return monitors, nil
}
func (m Monitor) AddRow() (int64, error) {
result, err := sq.
Insert(monitorTable).
Columns("namespace_id", "name", "url", "second", "times", "notify_type", "notify_target", "notify_times", "description", "error_content").
Values(m.NamespaceID, m.Name, m.URL, m.Second, m.Times, m.NotifyType, m.NotifyTarget, m.NotifyTimes, m.Description, "").
RunWith(DB).
Exec()
if err != nil {
return 0, err
}
id, err := result.LastInsertId()
return id, err
}
func (m Monitor) EditRow() error {
_, err := sq.
Update(monitorTable).
SetMap(sq.Eq{
"name": m.Name,
"url": m.URL,
"second": m.Second,
"times": m.Times,
"notify_type": m.NotifyType,
"notify_target": m.NotifyTarget,
"notify_times": m.NotifyTimes,
"description": m.Description,
}).
Where(sq.Eq{"id": m.ID}).
RunWith(DB).
Exec()
return err
}
func (m Monitor) ToggleState() error {
_, err := sq.
Update(monitorTable).
SetMap(sq.Eq{
"state": sq.Expr("!state"),
}).
Where(sq.Eq{"id": m.ID}).
RunWith(DB).
Exec()
return err
}
func (m Monitor) DeleteRow() error {
_, err := sq.
Delete(monitorTable).
Where(sq.Eq{"id": m.ID}).
RunWith(DB).
Exec()
return err
}
func (m Monitor) TurnOff(errorContent string) error {
_, err := sq.
Update(monitorTable).
SetMap(sq.Eq{
"state": Disable,
"error_content": errorContent,
}).
Where(sq.Eq{"id": m.ID}).
RunWith(DB).
Exec()
return err
}
func (m Monitor) Notify(errMsg string) (string, error) {
var err error
var resp *http.Response
if m.NotifyType == NotifyWeiXin {
type markdown struct {
Content string `json:"content"`
}
type message struct {
Msgtype string `json:"msgtype"`
Markdown markdown `json:"markdown"`
}
content := "Monitor: " + m.Name + "\n "
content += "> can not access \n "
content += "> " + errMsg + " \n "
msg := message{
Msgtype: "markdown",
Markdown: markdown{
Content: content,
},
}
b, _ := json.Marshal(msg)
resp, err = http.Post(m.NotifyTarget, "application/json", bytes.NewBuffer(b))
} else if m.NotifyType == NotifyDingTalk {
type markdown struct {
Title string `json:"title"`
Text string `json:"text"`
}
type message struct {
Msgtype string `json:"msgtype"`
Markdown markdown `json:"markdown"`
}
text := "#### Monitor: " + m.Name + " can not access \n >" + errMsg
msg := message{
Msgtype: "markdown",
Markdown: markdown{
Title: m.Name,
Text: text,
},
}
b, _ := json.Marshal(msg)
resp, err = http.Post(m.NotifyTarget, "application/json", bytes.NewBuffer(b))
} else if m.NotifyType == NotifyFeiShu {
type content struct {
Text string `json:"text"`
}
type message struct {
MsgType string `json:"msg_type"`
Content content `json:"content"`
}
text := "can not access\n "
text += "detail: " + errMsg
msg := message{
MsgType: "text",
Content: content{
Text: text,
},
}
b, _ := json.Marshal(msg)
resp, err = http.Post(m.NotifyTarget, "application/json", bytes.NewBuffer(b))
} else if m.NotifyType == NotifyCustom {
type message struct {
Code int `json:"code"`
Message string `json:"message"`
Data struct {
MonitorName string `json:"monitorName"`
URL string `json:"url"`
Port int `json:"port"`
Second int `json:"second"`
Times uint16 `json:"times"`
} `json:"data"`
}
code := 0
msg := message{
Code: code,
Message: "Monitor:" + m.Name + "can not access",
}
msg.Data.MonitorName = m.Name
msg.Data.URL = m.URL
msg.Data.Second = m.Second
msg.Data.Times = m.Times
b, _ := json.Marshal(msg)
resp, err = http.Post(m.NotifyTarget, "application/json", bytes.NewBuffer(b))
}
defer resp.Body.Close()
responseData, err := ioutil.ReadAll(resp.Body)
if err != nil {
return "", err
} else {
return string(responseData), err
}
}