mirror of
https://github.com/TeaOSLab/EdgeNode.git
synced 2025-11-03 23:20:25 +08:00
273 lines
6.9 KiB
Go
273 lines
6.9 KiB
Go
package nodes
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/nodeconfigs"
|
|
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
|
"github.com/TeaOSLab/EdgeNode/internal/caches"
|
|
teaconst "github.com/TeaOSLab/EdgeNode/internal/const"
|
|
"github.com/TeaOSLab/EdgeNode/internal/events"
|
|
"github.com/TeaOSLab/EdgeNode/internal/firewalls"
|
|
"github.com/TeaOSLab/EdgeNode/internal/monitor"
|
|
"github.com/TeaOSLab/EdgeNode/internal/remotelogs"
|
|
"github.com/TeaOSLab/EdgeNode/internal/rpc"
|
|
"github.com/TeaOSLab/EdgeNode/internal/trackers"
|
|
"github.com/TeaOSLab/EdgeNode/internal/utils"
|
|
"github.com/iwind/TeaGo/lists"
|
|
"github.com/iwind/TeaGo/maps"
|
|
"github.com/shirou/gopsutil/v3/cpu"
|
|
"github.com/shirou/gopsutil/v3/disk"
|
|
"golang.org/x/sys/unix"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
type NodeStatusExecutor struct {
|
|
isFirstTime bool
|
|
|
|
cpuUpdatedTime time.Time
|
|
cpuLogicalCount int
|
|
cpuPhysicalCount int
|
|
|
|
ticker *time.Ticker
|
|
}
|
|
|
|
func NewNodeStatusExecutor() *NodeStatusExecutor {
|
|
return &NodeStatusExecutor{
|
|
ticker: time.NewTicker(30 * time.Second),
|
|
}
|
|
}
|
|
|
|
func (this *NodeStatusExecutor) Listen() {
|
|
this.isFirstTime = true
|
|
this.cpuUpdatedTime = time.Now()
|
|
this.update()
|
|
|
|
events.OnKey(events.EventQuit, this, func() {
|
|
remotelogs.Println("NODE_STATUS", "quit executor")
|
|
this.ticker.Stop()
|
|
})
|
|
|
|
for range this.ticker.C {
|
|
this.isFirstTime = false
|
|
this.update()
|
|
}
|
|
}
|
|
|
|
func (this *NodeStatusExecutor) update() {
|
|
if sharedNodeConfig == nil {
|
|
return
|
|
}
|
|
|
|
var tr = trackers.Begin("UPLOAD_NODE_STATUS")
|
|
defer tr.End()
|
|
|
|
var status = &nodeconfigs.NodeStatus{}
|
|
status.BuildVersion = teaconst.Version
|
|
status.BuildVersionCode = utils.VersionToLong(teaconst.Version)
|
|
status.OS = runtime.GOOS
|
|
status.Arch = runtime.GOARCH
|
|
exe, _ := os.Executable()
|
|
status.ExePath = exe
|
|
status.ConfigVersion = sharedNodeConfig.Version
|
|
status.IsActive = true
|
|
status.ConnectionCount = sharedListenerManager.TotalActiveConnections()
|
|
status.CacheTotalDiskSize = caches.SharedManager.TotalDiskSize()
|
|
status.CacheTotalMemorySize = caches.SharedManager.TotalMemorySize()
|
|
status.TrafficInBytes = teaconst.InTrafficBytes
|
|
status.TrafficOutBytes = teaconst.OutTrafficBytes
|
|
var localFirewall = firewalls.Firewall()
|
|
if localFirewall != nil && !localFirewall.IsMock() {
|
|
status.LocalFirewallName = localFirewall.Name()
|
|
}
|
|
|
|
// 记录监控数据
|
|
monitor.SharedValueQueue.Add(nodeconfigs.NodeValueItemConnections, maps.Map{
|
|
"total": status.ConnectionCount,
|
|
})
|
|
|
|
hostname, _ := os.Hostname()
|
|
status.Hostname = hostname
|
|
|
|
var cpuTR = tr.Begin("cpu")
|
|
this.updateCPU(status)
|
|
cpuTR.End()
|
|
|
|
var memTR = tr.Begin("memory")
|
|
this.updateMem(status)
|
|
memTR.End()
|
|
|
|
var loadTR = tr.Begin("load")
|
|
this.updateLoad(status)
|
|
loadTR.End()
|
|
|
|
var diskTR = tr.Begin("disk")
|
|
this.updateDisk(status)
|
|
diskTR.End()
|
|
|
|
var cacheSpaceTR = tr.Begin("cache space")
|
|
this.updateCacheSpace(status)
|
|
cacheSpaceTR.End()
|
|
|
|
status.UpdatedAt = time.Now().Unix()
|
|
status.Timestamp = status.UpdatedAt
|
|
|
|
// 发送数据
|
|
jsonData, err := json.Marshal(status)
|
|
if err != nil {
|
|
remotelogs.Error("NODE_STATUS", "serial NodeStatus fail: "+err.Error())
|
|
return
|
|
}
|
|
rpcClient, err := rpc.SharedRPC()
|
|
if err != nil {
|
|
remotelogs.Error("NODE_STATUS", "failed to open rpc: "+err.Error())
|
|
return
|
|
}
|
|
_, err = rpcClient.NodeRPC.UpdateNodeStatus(rpcClient.Context(), &pb.UpdateNodeStatusRequest{
|
|
StatusJSON: jsonData,
|
|
})
|
|
if err != nil {
|
|
if rpc.IsConnError(err) {
|
|
remotelogs.Warn("NODE_STATUS", "rpc UpdateNodeStatus() failed: "+err.Error())
|
|
} else {
|
|
remotelogs.Error("NODE_STATUS", "rpc UpdateNodeStatus() failed: "+err.Error())
|
|
}
|
|
return
|
|
}
|
|
}
|
|
|
|
// 更新CPU
|
|
func (this *NodeStatusExecutor) updateCPU(status *nodeconfigs.NodeStatus) {
|
|
duration := time.Duration(0)
|
|
if this.isFirstTime {
|
|
duration = 100 * time.Millisecond
|
|
}
|
|
percents, err := cpu.Percent(duration, false)
|
|
if err != nil {
|
|
status.Error = "cpu.Percent(): " + err.Error()
|
|
return
|
|
}
|
|
if len(percents) == 0 {
|
|
return
|
|
}
|
|
status.CPUUsage = percents[0] / 100
|
|
|
|
// 记录监控数据
|
|
monitor.SharedValueQueue.Add(nodeconfigs.NodeValueItemCPU, maps.Map{
|
|
"usage": status.CPUUsage,
|
|
"cores": runtime.NumCPU(),
|
|
})
|
|
|
|
if this.cpuLogicalCount == 0 && this.cpuPhysicalCount == 0 {
|
|
this.cpuUpdatedTime = time.Now()
|
|
|
|
status.CPULogicalCount, err = cpu.Counts(true)
|
|
if err != nil {
|
|
status.Error = "cpu.Counts(): " + err.Error()
|
|
return
|
|
}
|
|
status.CPUPhysicalCount, err = cpu.Counts(false)
|
|
if err != nil {
|
|
status.Error = "cpu.Counts(): " + err.Error()
|
|
return
|
|
}
|
|
this.cpuLogicalCount = status.CPULogicalCount
|
|
this.cpuPhysicalCount = status.CPUPhysicalCount
|
|
} else {
|
|
status.CPULogicalCount = this.cpuLogicalCount
|
|
status.CPUPhysicalCount = this.cpuPhysicalCount
|
|
}
|
|
}
|
|
|
|
// 更新硬盘
|
|
func (this *NodeStatusExecutor) updateDisk(status *nodeconfigs.NodeStatus) {
|
|
partitions, err := disk.Partitions(false)
|
|
if err != nil {
|
|
remotelogs.Error("NODE_STATUS", err.Error())
|
|
return
|
|
}
|
|
lists.Sort(partitions, func(i int, j int) bool {
|
|
p1 := partitions[i]
|
|
p2 := partitions[j]
|
|
return p1.Mountpoint > p2.Mountpoint
|
|
})
|
|
|
|
// 当前TeaWeb所在的fs
|
|
rootFS := ""
|
|
rootTotal := uint64(0)
|
|
if lists.ContainsString([]string{"darwin", "linux", "freebsd"}, runtime.GOOS) {
|
|
for _, p := range partitions {
|
|
if p.Mountpoint == "/" {
|
|
rootFS = p.Fstype
|
|
usage, _ := disk.Usage(p.Mountpoint)
|
|
if usage != nil {
|
|
rootTotal = usage.Total
|
|
}
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
total := rootTotal
|
|
totalUsage := uint64(0)
|
|
maxUsage := float64(0)
|
|
for _, partition := range partitions {
|
|
if runtime.GOOS != "windows" && !strings.Contains(partition.Device, "/") && !strings.Contains(partition.Device, "\\") {
|
|
continue
|
|
}
|
|
|
|
// 跳过不同fs的
|
|
if len(rootFS) > 0 && rootFS != partition.Fstype {
|
|
continue
|
|
}
|
|
|
|
usage, err := disk.Usage(partition.Mountpoint)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
|
|
if partition.Mountpoint != "/" && (usage.Total != rootTotal || total == 0) {
|
|
total += usage.Total
|
|
}
|
|
totalUsage += usage.Used
|
|
if usage.UsedPercent >= maxUsage {
|
|
maxUsage = usage.UsedPercent
|
|
status.DiskMaxUsagePartition = partition.Mountpoint
|
|
}
|
|
}
|
|
status.DiskTotal = total
|
|
status.DiskUsage = float64(totalUsage) / float64(total)
|
|
status.DiskMaxUsage = maxUsage / 100
|
|
|
|
// 记录监控数据
|
|
monitor.SharedValueQueue.Add(nodeconfigs.NodeValueItemDisk, maps.Map{
|
|
"total": status.DiskTotal,
|
|
"usage": status.DiskUsage,
|
|
"maxUsage": status.DiskMaxUsage,
|
|
})
|
|
}
|
|
|
|
// 缓存空间
|
|
func (this *NodeStatusExecutor) updateCacheSpace(status *nodeconfigs.NodeStatus) {
|
|
var result = []maps.Map{}
|
|
cachePaths := caches.SharedManager.FindAllCachePaths()
|
|
for _, path := range cachePaths {
|
|
var stat unix.Statfs_t
|
|
err := unix.Statfs(path, &stat)
|
|
if err != nil {
|
|
return
|
|
}
|
|
result = append(result, maps.Map{
|
|
"path": path,
|
|
"total": stat.Blocks * uint64(stat.Bsize),
|
|
"avail": stat.Bavail * uint64(stat.Bsize),
|
|
"used": (stat.Blocks - stat.Bavail) * uint64(stat.Bsize),
|
|
})
|
|
}
|
|
monitor.SharedValueQueue.Add(nodeconfigs.NodeValueItemCacheDir, maps.Map{
|
|
"dirs": result,
|
|
})
|
|
}
|