监控CPU磁盘网络数据库

2023-08-15 21:38:05 +08:00
parent 3672b73683
commit bfd3be9f03
3 changed files with 412 additions and 0 deletions
--- a/features/monitor/monitor/monitor.go
+++ b/features/monitor/monitor/monitor.go
@@ -0,0 +1,124 @@
+package monitor
+
+import (
+	"net/http"
+	"sort"
+	"time"
+
+	"ems.agt/lib/dborm"
+	"ems.agt/lib/services"
+	"ems.agt/restagent/config"
+	"github.com/shirou/gopsutil/disk"
+	"github.com/shirou/gopsutil/net"
+)
+
+// 点击【主机 - 监控】菜单，进入监控报表，直观的了解服务器的运行状态，包含【平均负载】、【CPU性能监控】、【内存使用监控】、【磁盘IO监控】、【网络IO监控】
+
+// 可以查看昨天，今天，最近7天，最近30天，自定义时间的监控指标情况。
+// 默认监控是开启的，可以在【面板设置 - 监控】页面中根据需求对监控进行开启和关闭。
+// 监控数据默认保存30天，可以自行修改，也可手动清理该日志。
+
+var (
+	// 可选网络
+	UriNetOpt = config.UriPrefix + "/monitor/{apiVersion}/monitor/netoptions"
+	// 可选磁盘
+	UriIoOpt = config.UriPrefix + "/monitor/{apiVersion}/monitor/iooptions"
+	// 加载
+	UriLoad = config.UriPrefix + "/monitor/{apiVersion}/monitor/load"
+)
+
+// Netoptions 可选网络
+func Netoptions(w http.ResponseWriter, r *http.Request) {
+	netStat, _ := net.IOCounters(true)
+	var options []string
+	options = append(options, "all")
+	for _, net := range netStat {
+		options = append(options, net.Name)
+	}
+	sort.Strings(options)
+	services.ResponseWithJson(w, 200, options)
+}
+
+// Iooptions 可选磁盘
+func Iooptions(w http.ResponseWriter, r *http.Request) {
+	diskStat, _ := disk.IOCounters()
+	var options []string
+	options = append(options, "all")
+	for _, net := range diskStat {
+		options = append(options, net.Name)
+	}
+	sort.Strings(options)
+	services.ResponseWithJson(w, 200, options)
+}
+
+// LoadMonitor 载入监控
+func LoadMonitor(w http.ResponseWriter, r *http.Request) {
+	// json 請求參數獲取
+	var bodyArgs MonitorSearch
+	err := services.ShouldBindJSON(r, &bodyArgs)
+	if err != nil || dborm.DbClient.XEngine == nil {
+		services.ResponseErrorWithJson(w, 400, err.Error())
+		return
+	}
+
+	loc := time.Now().Location()
+	bodyArgs.StartTime = bodyArgs.StartTime.In(loc)
+	bodyArgs.EndTime = bodyArgs.EndTime.In(loc)
+
+	var backdatas []MonitorData
+	if bodyArgs.Param == "all" || bodyArgs.Param == "cpu" || bodyArgs.Param == "memory" || bodyArgs.Param == "load" {
+		var bases []MonitorBase
+		err := dborm.DbClient.XEngine.SQL("SELECT * FROM monitor_base").
+			Where("created_at > ? AND created_at < ?", bodyArgs.StartTime, bodyArgs.EndTime).
+			Find(&bases)
+		if err != nil {
+			services.ResponseErrorWithJson(w, 400, err.Error())
+			return
+		}
+
+		var itemData MonitorData
+		itemData.Param = "base"
+		for _, base := range bases {
+			itemData.Date = append(itemData.Date, base.CreatedAt)
+			itemData.Value = append(itemData.Value, base)
+		}
+		backdatas = append(backdatas, itemData)
+	}
+	if bodyArgs.Param == "all" || bodyArgs.Param == "io" {
+		var bases []MonitorIO
+		err := dborm.DbClient.XEngine.SQL("SELECT * FROM monitor_io").
+			Where("created_at > ? AND created_at < ?", bodyArgs.StartTime, bodyArgs.EndTime).
+			Find(&bases)
+		if err != nil {
+			services.ResponseErrorWithJson(w, 400, err.Error())
+			return
+		}
+
+		var itemData MonitorData
+		itemData.Param = "io"
+		for _, base := range bases {
+			itemData.Date = append(itemData.Date, base.CreatedAt)
+			itemData.Value = append(itemData.Value, base)
+		}
+		backdatas = append(backdatas, itemData)
+	}
+	if bodyArgs.Param == "all" || bodyArgs.Param == "network" {
+		var bases []MonitorNetwork
+		err := dborm.DbClient.XEngine.SQL("SELECT * FROM monitor_network").
+			Where("name = ? AND created_at > ? AND created_at < ?", bodyArgs.Info, bodyArgs.StartTime, bodyArgs.EndTime).
+			Find(&bases)
+		if err != nil {
+			services.ResponseErrorWithJson(w, 400, err.Error())
+			return
+		}
+
+		var itemData MonitorData
+		itemData.Param = "network"
+		for _, base := range bases {
+			itemData.Date = append(itemData.Date, base.CreatedAt)
+			itemData.Value = append(itemData.Value, base)
+		}
+		backdatas = append(backdatas, itemData)
+	}
+	services.ResponseWithJson(w, 200, backdatas)
+}