mirror of
https://github.com/0xJacky/nginx-ui.git
synced 2025-05-11 02:15:48 +02:00
95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/shirou/gopsutil/v3/cpu"
|
|
"github.com/shirou/gopsutil/v3/disk"
|
|
"github.com/shirou/gopsutil/v3/host"
|
|
"github.com/shirou/gopsutil/v3/load"
|
|
"github.com/shirou/gopsutil/v3/mem"
|
|
"net/http"
|
|
"runtime"
|
|
"strconv"
|
|
"time"
|
|
|
|
"github.com/dustin/go-humanize"
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func Analytic(c *gin.Context) {
|
|
var upGrader = websocket.Upgrader{
|
|
CheckOrigin: func(r *http.Request) bool {
|
|
return true
|
|
},
|
|
}
|
|
// upgrade http to websocket
|
|
ws, err := upGrader.Upgrade(c.Writer, c.Request, nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
defer ws.Close()
|
|
|
|
response := make(gin.H)
|
|
|
|
for {
|
|
// read
|
|
mt, message, err := ws.ReadMessage()
|
|
if err != nil {
|
|
break
|
|
}
|
|
for {
|
|
|
|
memoryStat, err := mem.VirtualMemory()
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return
|
|
}
|
|
|
|
response["memory_total"] = humanize.Bytes(memoryStat.Total)
|
|
response["memory_used"] = humanize.Bytes(memoryStat.Used)
|
|
response["memory_cached"] = humanize.Bytes(memoryStat.Cached)
|
|
response["memory_free"] = humanize.Bytes(memoryStat.Free)
|
|
|
|
response["memory_pressure"], _ = strconv.ParseFloat(fmt.Sprintf("%.2f", memoryStat.UsedPercent), 64)
|
|
|
|
cpuTimesBefore, _ := cpu.Times(false)
|
|
time.Sleep(1000 * time.Millisecond)
|
|
cpuTimesAfter, _ := cpu.Times(false)
|
|
threadNum := runtime.GOMAXPROCS(0)
|
|
|
|
cpuUserUsage := (cpuTimesAfter[0].User - cpuTimesBefore[0].User) / (float64(1000*threadNum) / 1000)
|
|
cpuSystemUsage := (cpuTimesAfter[0].System - cpuTimesBefore[0].System) / (float64(1000*threadNum) / 1000)
|
|
|
|
response["cpu_user"], _ = strconv.ParseFloat(fmt.Sprintf("%.2f",
|
|
cpuUserUsage*100), 64)
|
|
|
|
response["cpu_system"], _ = strconv.ParseFloat(fmt.Sprintf("%.2f",
|
|
cpuSystemUsage*100), 64)
|
|
|
|
response["cpu_idle"], _ = strconv.ParseFloat(fmt.Sprintf("%.2f",
|
|
(1-cpuUserUsage+cpuSystemUsage)*100), 64)
|
|
|
|
response["uptime"], _ = host.Uptime()
|
|
response["loadavg"], _ = load.Avg()
|
|
|
|
diskUsage, _ := disk.Usage(".")
|
|
|
|
response["disk_used"] = humanize.Bytes(diskUsage.Used)
|
|
response["disk_total"] = humanize.Bytes(diskUsage.Total)
|
|
response["disk_percentage"], _ = strconv.ParseFloat(fmt.Sprintf("%.2f", diskUsage.UsedPercent), 64)
|
|
|
|
m, _ := json.Marshal(response)
|
|
message = m
|
|
|
|
// write
|
|
err = ws.WriteMessage(mt, message)
|
|
if err != nil {
|
|
break
|
|
}
|
|
time.Sleep(800 * time.Microsecond)
|
|
}
|
|
}
|
|
}
|