mirror of
https://github.com/TeaOSLab/EdgeAdmin.git
synced 2025-11-03 12:20:28 +08:00
312 lines
9.0 KiB
Go
312 lines
9.0 KiB
Go
// Copyright 2021 Liuxiangchao iwind.liu@gmail.com. All rights reserved.
|
||
|
||
package dashboard
|
||
|
||
import (
|
||
"fmt"
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/configloaders"
|
||
teaconst "github.com/TeaOSLab/EdgeAdmin/internal/const"
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/utils/numberutils"
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/utils/sizes"
|
||
"github.com/TeaOSLab/EdgeAdmin/internal/web/actions/actionutils"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/rpc/pb"
|
||
"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
|
||
"github.com/iwind/TeaGo/lists"
|
||
"github.com/iwind/TeaGo/maps"
|
||
"github.com/iwind/TeaGo/types"
|
||
"github.com/shirou/gopsutil/v3/disk"
|
||
"regexp"
|
||
"runtime"
|
||
)
|
||
|
||
type IndexAction struct {
|
||
actionutils.ParentAction
|
||
}
|
||
|
||
func (this *IndexAction) Init() {
|
||
this.Nav("", "", "")
|
||
}
|
||
|
||
func (this *IndexAction) RunGet(params struct{}) {
|
||
if teaconst.IsPlus {
|
||
this.RedirectURL("/dashboard/boards")
|
||
return
|
||
}
|
||
|
||
// 取得用户的权限
|
||
module, ok := configloaders.FindFirstAdminModule(this.AdminId())
|
||
if ok {
|
||
if module != "dashboard" {
|
||
for _, m := range configloaders.AllModuleMaps() {
|
||
if m.GetString("code") == module {
|
||
this.RedirectURL(m.GetString("url"))
|
||
return
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// 版本更新
|
||
this.Data["currentVersionCode"] = teaconst.Version
|
||
this.Data["newVersionCode"] = teaconst.NewVersionCode
|
||
this.Data["newVersionDownloadURL"] = teaconst.NewVersionDownloadURL
|
||
|
||
this.Show()
|
||
}
|
||
|
||
func (this *IndexAction) RunPost(params struct{}) {
|
||
// 读取看板数据
|
||
resp, err := this.RPC().AdminRPC().ComposeAdminDashboard(this.AdminContext(), &pb.ComposeAdminDashboardRequest{
|
||
ApiVersion: teaconst.APINodeVersion,
|
||
})
|
||
if err != nil {
|
||
this.ErrorPage(err)
|
||
return
|
||
}
|
||
|
||
// 检查当前服务器空间
|
||
var diskUsageWarning = ""
|
||
diskPath, diskUsage, diskUsagePercent, shouldWarning := this.checkDiskPartitions(90)
|
||
if shouldWarning {
|
||
diskUsageWarning = "当前服务器磁盘空间不足，请立即扩充容量，文件路径：" + diskPath + "，已使用：" + types.String(diskUsage/1024/1024/1024) + "G，已使用比例：" + fmt.Sprintf("%.2f%%", diskUsagePercent) + "，仅剩余空间：" + fmt.Sprintf("%.2f%%", 100-diskUsagePercent) + "。"
|
||
}
|
||
|
||
this.Data["dashboard"] = maps.Map{
|
||
"defaultClusterId": resp.DefaultNodeClusterId,
|
||
|
||
"countServers": resp.CountServers,
|
||
"countNodeClusters": resp.CountNodeClusters,
|
||
"countNodes": resp.CountNodes,
|
||
"countOfflineNodes": resp.CountOfflineNodes,
|
||
"countUsers": resp.CountUsers,
|
||
"countAPINodes": resp.CountAPINodes,
|
||
"countOfflineAPINodes": resp.CountOfflineAPINodes,
|
||
"countDBNodes": resp.CountDBNodes,
|
||
"countUserNodes": resp.CountUserNodes,
|
||
"countOfflineUserNodes": resp.CountOfflineUserNodes,
|
||
|
||
"canGoServers": configloaders.AllowModule(this.AdminId(), configloaders.AdminModuleCodeServer),
|
||
"canGoNodes": configloaders.AllowModule(this.AdminId(), configloaders.AdminModuleCodeNode),
|
||
"canGoSettings": configloaders.AllowModule(this.AdminId(), configloaders.AdminModuleCodeSetting),
|
||
"canGoUsers": configloaders.AllowModule(this.AdminId(), configloaders.AdminModuleCodeUser),
|
||
|
||
"diskUsageWarning": diskUsageWarning,
|
||
}
|
||
|
||
// 今日流量
|
||
todayTrafficBytes := int64(0)
|
||
if len(resp.DailyTrafficStats) > 0 {
|
||
todayTrafficBytes = resp.DailyTrafficStats[len(resp.DailyTrafficStats)-1].Bytes
|
||
}
|
||
todayTrafficString := numberutils.FormatBytes(todayTrafficBytes)
|
||
result := regexp.MustCompile(`^(?U)(.+)([a-zA-Z]+)$`).FindStringSubmatch(todayTrafficString)
|
||
if len(result) > 2 {
|
||
this.Data["todayTraffic"] = result[1]
|
||
this.Data["todayTrafficUnit"] = result[2]
|
||
} else {
|
||
this.Data["todayTraffic"] = todayTrafficString
|
||
this.Data["todayTrafficUnit"] = ""
|
||
}
|
||
|
||
// 24小时流量趋势
|
||
{
|
||
statMaps := []maps.Map{}
|
||
for _, stat := range resp.HourlyTrafficStats {
|
||
statMaps = append(statMaps, maps.Map{
|
||
"bytes": stat.Bytes,
|
||
"cachedBytes": stat.CachedBytes,
|
||
"countRequests": stat.CountRequests,
|
||
"countCachedRequests": stat.CountCachedRequests,
|
||
"countAttackRequests": stat.CountAttackRequests,
|
||
"attackBytes": stat.AttackBytes,
|
||
"day": stat.Hour[4:6] + "月" + stat.Hour[6:8] + "日",
|
||
"hour": stat.Hour[8:],
|
||
})
|
||
}
|
||
this.Data["hourlyTrafficStats"] = statMaps
|
||
}
|
||
|
||
// 15天流量趋势
|
||
{
|
||
statMaps := []maps.Map{}
|
||
for _, stat := range resp.DailyTrafficStats {
|
||
statMaps = append(statMaps, maps.Map{
|
||
"bytes": stat.Bytes,
|
||
"cachedBytes": stat.CachedBytes,
|
||
"countRequests": stat.CountRequests,
|
||
"countCachedRequests": stat.CountCachedRequests,
|
||
"countAttackRequests": stat.CountAttackRequests,
|
||
"attackBytes": stat.AttackBytes,
|
||
"day": stat.Day[4:6] + "月" + stat.Day[6:] + "日",
|
||
})
|
||
}
|
||
this.Data["dailyTrafficStats"] = statMaps
|
||
}
|
||
|
||
// 版本升级
|
||
if resp.NodeUpgradeInfo != nil {
|
||
this.Data["nodeUpgradeInfo"] = maps.Map{
|
||
"count": resp.NodeUpgradeInfo.CountNodes,
|
||
"version": resp.NodeUpgradeInfo.NewVersion,
|
||
}
|
||
} else {
|
||
this.Data["nodeUpgradeInfo"] = maps.Map{
|
||
"count": 0,
|
||
"version": "",
|
||
}
|
||
}
|
||
if resp.MonitorNodeUpgradeInfo != nil {
|
||
this.Data["monitorNodeUpgradeInfo"] = maps.Map{
|
||
"count": resp.MonitorNodeUpgradeInfo.CountNodes,
|
||
"version": resp.MonitorNodeUpgradeInfo.NewVersion,
|
||
}
|
||
} else {
|
||
this.Data["monitorNodeUpgradeInfo"] = maps.Map{
|
||
"count": 0,
|
||
"version": "",
|
||
}
|
||
}
|
||
if resp.ApiNodeUpgradeInfo != nil {
|
||
this.Data["apiNodeUpgradeInfo"] = maps.Map{
|
||
"count": resp.ApiNodeUpgradeInfo.CountNodes,
|
||
"version": resp.ApiNodeUpgradeInfo.NewVersion,
|
||
}
|
||
} else {
|
||
this.Data["apiNodeUpgradeInfo"] = maps.Map{
|
||
"count": 0,
|
||
"version": "",
|
||
}
|
||
}
|
||
if resp.UserNodeUpgradeInfo != nil {
|
||
this.Data["userNodeUpgradeInfo"] = maps.Map{
|
||
"count": resp.UserNodeUpgradeInfo.CountNodes,
|
||
"version": resp.UserNodeUpgradeInfo.NewVersion,
|
||
}
|
||
} else {
|
||
this.Data["userNodeUpgradeInfo"] = maps.Map{
|
||
"count": 0,
|
||
"version": 0,
|
||
}
|
||
}
|
||
if resp.AuthorityNodeUpgradeInfo != nil {
|
||
this.Data["authorityNodeUpgradeInfo"] = maps.Map{
|
||
"count": resp.AuthorityNodeUpgradeInfo.CountNodes,
|
||
"version": resp.AuthorityNodeUpgradeInfo.NewVersion,
|
||
}
|
||
} else {
|
||
this.Data["authorityNodeUpgradeInfo"] = maps.Map{
|
||
"count": 0,
|
||
"version": "",
|
||
}
|
||
}
|
||
if resp.NsNodeUpgradeInfo != nil {
|
||
this.Data["nsNodeUpgradeInfo"] = maps.Map{
|
||
"count": resp.NsNodeUpgradeInfo.CountNodes,
|
||
"version": resp.NsNodeUpgradeInfo.NewVersion,
|
||
}
|
||
} else {
|
||
this.Data["nsNodeUpgradeInfo"] = maps.Map{
|
||
"count": 0,
|
||
"version": "",
|
||
}
|
||
}
|
||
|
||
// 域名排行
|
||
{
|
||
var statMaps = []maps.Map{}
|
||
for _, stat := range resp.TopDomainStats {
|
||
statMaps = append(statMaps, maps.Map{
|
||
"serverId": stat.ServerId,
|
||
"domain": stat.Domain,
|
||
"countRequests": stat.CountRequests,
|
||
"bytes": stat.Bytes,
|
||
})
|
||
}
|
||
this.Data["topDomainStats"] = statMaps
|
||
}
|
||
|
||
// 指标
|
||
{
|
||
var chartMaps = []maps.Map{}
|
||
for _, chart := range resp.MetricDataCharts {
|
||
var statMaps = []maps.Map{}
|
||
for _, stat := range chart.MetricStats {
|
||
statMaps = append(statMaps, maps.Map{
|
||
"keys": stat.Keys,
|
||
"time": stat.Time,
|
||
"value": stat.Value,
|
||
"count": stat.SumCount,
|
||
"total": stat.SumTotal,
|
||
})
|
||
}
|
||
chartMaps = append(chartMaps, maps.Map{
|
||
"chart": maps.Map{
|
||
"id": chart.MetricChart.Id,
|
||
"name": chart.MetricChart.Name,
|
||
"widthDiv": chart.MetricChart.WidthDiv,
|
||
"isOn": chart.MetricChart.IsOn,
|
||
"maxItems": chart.MetricChart.MaxItems,
|
||
"type": chart.MetricChart.Type,
|
||
},
|
||
"item": maps.Map{
|
||
"id": chart.MetricChart.MetricItem.Id,
|
||
"name": chart.MetricChart.MetricItem.Name,
|
||
"period": chart.MetricChart.MetricItem.Period,
|
||
"periodUnit": chart.MetricChart.MetricItem.PeriodUnit,
|
||
"valueType": serverconfigs.FindMetricValueType(chart.MetricChart.MetricItem.Category, chart.MetricChart.MetricItem.Value),
|
||
"valueTypeName": serverconfigs.FindMetricValueName(chart.MetricChart.MetricItem.Category, chart.MetricChart.MetricItem.Value),
|
||
"keys": chart.MetricChart.MetricItem.Keys,
|
||
},
|
||
"stats": statMaps,
|
||
})
|
||
}
|
||
this.Data["metricCharts"] = chartMaps
|
||
}
|
||
|
||
this.Success()
|
||
}
|
||
|
||
// 检查服务器磁盘空间
|
||
func (this *IndexAction) checkDiskPartitions(thresholdPercent float64) (path string, usage uint64, usagePercent float64, shouldWarning bool) {
|
||
partitions, err := disk.Partitions(false)
|
||
if err != nil {
|
||
return
|
||
}
|
||
if !lists.ContainsString([]string{"darwin", "linux", "freebsd"}, runtime.GOOS) {
|
||
return
|
||
}
|
||
|
||
var rootFS = ""
|
||
|
||
for _, p := range partitions {
|
||
if p.Mountpoint == "/" {
|
||
rootFS = p.Fstype
|
||
break
|
||
}
|
||
}
|
||
|
||
for _, p := range partitions {
|
||
if p.Mountpoint == "/boot" {
|
||
continue
|
||
}
|
||
if p.Fstype != rootFS {
|
||
continue
|
||
}
|
||
stat, _ := disk.Usage(p.Mountpoint)
|
||
if stat != nil {
|
||
if stat.Used < 2*uint64(sizes.G) {
|
||
continue
|
||
}
|
||
if stat.UsedPercent > thresholdPercent {
|
||
path = stat.Path
|
||
usage = stat.Used
|
||
usagePercent = stat.UsedPercent
|
||
shouldWarning = true
|
||
break
|
||
}
|
||
}
|
||
}
|
||
|
||
return
|
||
}
|