panel/internal/service/monitor.go

package service

import (
	"fmt"
	"net/http"
	"time"

	"github.com/acepanel/panel/internal/biz"
	"github.com/acepanel/panel/internal/http/request"
	"github.com/acepanel/panel/pkg/types"
)

type MonitorService struct {
	settingRepo biz.SettingRepo
	monitorRepo biz.MonitorRepo
}

func NewMonitorService(setting biz.SettingRepo, monitor biz.MonitorRepo) *MonitorService {
	return &MonitorService{
		settingRepo: setting,
		monitorRepo: monitor,
	}
}

func (s *MonitorService) GetSetting(w http.ResponseWriter, r *http.Request) {
	setting, err := s.monitorRepo.GetSetting()
	if err != nil {
		Error(w, http.StatusInternalServerError, "%v", err)
		return
	}

	Success(w, setting)
}

func (s *MonitorService) UpdateSetting(w http.ResponseWriter, r *http.Request) {
	req, err := Bind[request.MonitorSetting](r)
	if err != nil {
		Error(w, http.StatusUnprocessableEntity, "%v", err)
		return
	}

	if err = s.monitorRepo.UpdateSetting(req); err != nil {
		Error(w, http.StatusInternalServerError, "%v", err)
		return
	}

	Success(w, nil)
}

func (s *MonitorService) Clear(w http.ResponseWriter, r *http.Request) {
	if err := s.monitorRepo.Clear(); err != nil {
		Error(w, http.StatusInternalServerError, "%v", err)
		return
	}

	Success(w, nil)
}

func (s *MonitorService) List(w http.ResponseWriter, r *http.Request) {
	req, err := Bind[request.MonitorList](r)
	if err != nil {
		Error(w, http.StatusUnprocessableEntity, "%v", err)
		return
	}

	monitors, err := s.monitorRepo.List(time.UnixMilli(req.Start), time.UnixMilli(req.End))
	if err != nil {
		Error(w, http.StatusInternalServerError, "%v", err)
		return
	}
	if len(monitors) == 0 {
		Success(w, types.MonitorDetail{})
		return
	}

	var list types.MonitorDetail

	// 用于存储每个网卡的累计流量
	netDeviceData := make(map[string]*types.Network)
	netDevicePrev := make(map[string]struct {
		sent uint64
		recv uint64
	})

	// 用于存储每个磁盘的IO数据
	diskIOData := make(map[string]*types.DiskIO)
	diskIOPrev := make(map[string]struct {
		read  uint64
		write uint64
	})

	// 初始化第一条数据的网卡和磁盘数据
	for _, net := range monitors[0].Info.Net {
		if net.Name == "lo" {
			continue
		}
		netDevicePrev[net.Name] = struct {
			sent uint64
			recv uint64
		}{sent: net.BytesSent, recv: net.BytesRecv}
		netDeviceData[net.Name] = &types.Network{Name: net.Name}
	}

	for _, disk := range monitors[0].Info.DiskIO {
		diskIOPrev[disk.Name] = struct {
			read  uint64
			write uint64
		}{read: disk.ReadBytes, write: disk.WriteBytes}
		diskIOData[disk.Name] = &types.DiskIO{Name: disk.Name}
	}

	for i, monitor := range monitors {
		// 跳过第一条数据，因为第一条数据的流量为 0
		if i == 0 {
			// MB
			list.Mem.Total = fmt.Sprintf("%.2f", float64(monitor.Info.Mem.Total)/1024/1024)
			list.SWAP.Total = fmt.Sprintf("%.2f", float64(monitor.Info.Swap.Total)/1024/1024)
			continue
		}

		// 处理网络数据
		for _, net := range monitor.Info.Net {
			if net.Name == "lo" {
				continue
			}

			// 按网卡分组
			if _, ok := netDeviceData[net.Name]; !ok {
				netDeviceData[net.Name] = &types.Network{Name: net.Name}
				netDevicePrev[net.Name] = struct {
					sent uint64
					recv uint64
				}{sent: 0, recv: 0}
			}
			prev := netDevicePrev[net.Name]
			device := netDeviceData[net.Name]
			device.Sent = append(device.Sent, fmt.Sprintf("%.2f", float64(net.BytesSent)/1024/1024))
			device.Recv = append(device.Recv, fmt.Sprintf("%.2f", float64(net.BytesRecv)/1024/1024))
			device.Tx = append(device.Tx, fmt.Sprintf("%.2f", float64(net.BytesSent-prev.sent)/60/1024/1024))
			device.Rx = append(device.Rx, fmt.Sprintf("%.2f", float64(net.BytesRecv-prev.recv)/60/1024/1024))
			netDevicePrev[net.Name] = struct {
				sent uint64
				recv uint64
			}{sent: net.BytesSent, recv: net.BytesRecv}
		}

		// 处理磁盘IO数据
		for _, disk := range monitor.Info.DiskIO {
			if _, ok := diskIOData[disk.Name]; !ok {
				diskIOData[disk.Name] = &types.DiskIO{Name: disk.Name}
				diskIOPrev[disk.Name] = struct {
					read  uint64
					write uint64
				}{read: 0, write: 0}
			}
			prev := diskIOPrev[disk.Name]
			diskData := diskIOData[disk.Name]
			diskData.ReadBytes = append(diskData.ReadBytes, fmt.Sprintf("%.2f", float64(disk.ReadBytes)/1024/1024))
			diskData.WriteBytes = append(diskData.WriteBytes, fmt.Sprintf("%.2f", float64(disk.WriteBytes)/1024/1024))
			// 监控频率为 1 分钟，所以这里除以 60 即可得到每秒的速度 (KB/s)
			diskData.ReadSpeed = append(diskData.ReadSpeed, fmt.Sprintf("%.2f", float64(disk.ReadBytes-prev.read)/60/1024))
			diskData.WriteSpeed = append(diskData.WriteSpeed, fmt.Sprintf("%.2f", float64(disk.WriteBytes-prev.write)/60/1024))
			diskIOPrev[disk.Name] = struct {
				read  uint64
				write uint64
			}{read: disk.ReadBytes, write: disk.WriteBytes}
		}

		list.Times = append(list.Times, monitor.CreatedAt.Format(time.DateTime))
		list.Load.Load1 = append(list.Load.Load1, monitor.Info.Load.Load1)
		list.Load.Load5 = append(list.Load.Load5, monitor.Info.Load.Load5)
		list.Load.Load15 = append(list.Load.Load15, monitor.Info.Load.Load15)
		list.CPU.Percent = append(list.CPU.Percent, fmt.Sprintf("%.2f", monitor.Info.Percent))
		list.Mem.Available = append(list.Mem.Available, fmt.Sprintf("%.2f", float64(monitor.Info.Mem.Available)/1024/1024))
		list.Mem.Used = append(list.Mem.Used, fmt.Sprintf("%.2f", float64(monitor.Info.Mem.Used)/1024/1024))
		list.SWAP.Used = append(list.SWAP.Used, fmt.Sprintf("%.2f", float64(monitor.Info.Swap.Used)/1024/1024))
		list.SWAP.Free = append(list.SWAP.Free, fmt.Sprintf("%.2f", float64(monitor.Info.Swap.Free)/1024/1024))
	}

	// 将 map 转换为 slice
	for _, device := range netDeviceData {
		list.Net = append(list.Net, *device)
	}
	for _, disk := range diskIOData {
		list.DiskIO = append(list.DiskIO, *disk)
	}

	Success(w, list)
}