109 lines
3.4 KiB
Go
109 lines
3.4 KiB
Go
package handler
|
|
|
|
import (
|
|
"runtime"
|
|
"time"
|
|
|
|
"kra/internal/server/httpx"
|
|
"kra/internal/service"
|
|
"kra/internal/service/dto"
|
|
"kra/internal/worker"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/shirou/gopsutil/v4/cpu"
|
|
"github.com/shirou/gopsutil/v4/disk"
|
|
"github.com/shirou/gopsutil/v4/mem"
|
|
)
|
|
|
|
type SystemConfig struct {
|
|
system *service.SystemService
|
|
settings *service.SettingsService
|
|
scheduler *worker.TaskScheduler
|
|
}
|
|
|
|
func NewSystemConfig(system *service.SystemService, settings *service.SettingsService, scheduler *worker.TaskScheduler) *SystemConfig {
|
|
return &SystemConfig{system: system, settings: settings, scheduler: scheduler}
|
|
}
|
|
|
|
func (h *SystemConfig) GetSecurity(c *gin.Context) {
|
|
value, err := h.settings.Security(c.Request.Context())
|
|
if err != nil {
|
|
httpx.Fail(c, "获取安全配置失败")
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, value, "获取成功")
|
|
}
|
|
|
|
func (h *SystemConfig) SetSecurity(c *gin.Context) {
|
|
var req dto.SecurityConfigRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
httpx.Fail(c, err.Error())
|
|
return
|
|
}
|
|
result, err := h.settings.SaveSecurityRequest(c.Request.Context(), &req)
|
|
if err != nil {
|
|
httpx.Fail(c, "设置安全配置失败")
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, result, "设置成功")
|
|
}
|
|
|
|
func (h *SystemConfig) Get(c *gin.Context) {
|
|
httpx.Write(c, httpx.CodeSuccess, h.system.SystemConfig(), "获取成功")
|
|
}
|
|
|
|
func (h *SystemConfig) Set(c *gin.Context) {
|
|
var req dto.SetSystemConfigRequest
|
|
if err := c.ShouldBindJSON(&req); err != nil {
|
|
httpx.Fail(c, err.Error())
|
|
return
|
|
}
|
|
if err := h.system.SaveSystemConfig(c.Request.Context(), &req); err != nil {
|
|
httpx.Fail(c, "设置失败")
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{}, "设置成功")
|
|
}
|
|
|
|
func (h *SystemConfig) Reload(c *gin.Context) {
|
|
if err := h.system.ReloadConfig(c.Request.Context()); err != nil {
|
|
httpx.Fail(c, "重载系统失败:"+err.Error())
|
|
return
|
|
}
|
|
if err := h.scheduler.Reload(c.Request.Context()); err != nil {
|
|
httpx.Fail(c, "系统配置已重载，但定时任务重载失败:"+err.Error())
|
|
return
|
|
}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{}, "重载系统成功")
|
|
}
|
|
|
|
func (h *SystemConfig) ServerInfo(c *gin.Context) {
|
|
physicalCores, err := cpu.Counts(false)
|
|
if err != nil {
|
|
httpx.Fail(c, "获取失败")
|
|
return
|
|
}
|
|
cpuPercent, err := cpu.Percent(200*time.Millisecond, true)
|
|
if err != nil {
|
|
httpx.Fail(c, "获取失败")
|
|
return
|
|
}
|
|
memory, err := mem.VirtualMemory()
|
|
if err != nil {
|
|
httpx.Fail(c, "获取失败")
|
|
return
|
|
}
|
|
diskInfo := make([]gin.H, 0)
|
|
for _, mountPoint := range h.system.DiskMountPoints() {
|
|
usage, usageErr := disk.Usage(mountPoint)
|
|
if usageErr != nil {
|
|
httpx.Fail(c, "获取失败")
|
|
return
|
|
}
|
|
diskInfo = append(diskInfo, gin.H{"mountPoint": mountPoint, "usedMb": usage.Used / 1024 / 1024, "usedGb": usage.Used / 1024 / 1024 / 1024, "totalMb": usage.Total / 1024 / 1024, "totalGb": usage.Total / 1024 / 1024 / 1024, "usedPercent": int(usage.UsedPercent)})
|
|
}
|
|
usedMB, totalMB, memoryPercent := memory.Used/1024/1024, memory.Total/1024/1024, int(memory.UsedPercent)
|
|
server := gin.H{"os": gin.H{"goos": runtime.GOOS, "numCpu": runtime.NumCPU(), "compiler": runtime.Compiler, "goVersion": runtime.Version(), "numGoroutine": runtime.NumGoroutine()}, "cpu": gin.H{"cpus": cpuPercent, "cores": physicalCores}, "ram": gin.H{"usedMb": usedMB, "totalMb": totalMB, "usedPercent": memoryPercent}, "disk": diskInfo}
|
|
httpx.Write(c, httpx.CodeSuccess, gin.H{"server": server}, "获取成功")
|
|
}
|