2
0
mirror of https://github.com/acepanel/panel.git synced 2026-02-04 06:47:20 +08:00
Files
panel/internal/service/process.go
2025-01-01 15:33:47 +08:00

111 lines
2.4 KiB
Go

package service
import (
"net/http"
"slices"
"time"
"github.com/go-rat/chix"
"github.com/shirou/gopsutil/process"
"github.com/tnb-labs/panel/internal/http/request"
"github.com/tnb-labs/panel/pkg/types"
)
type ProcessService struct {
}
func NewProcessService() *ProcessService {
return &ProcessService{}
}
func (s *ProcessService) List(w http.ResponseWriter, r *http.Request) {
processes, err := process.Processes()
if err != nil {
Error(w, http.StatusInternalServerError, "%v", err)
return
}
data := make([]types.ProcessData, 0)
for proc := range slices.Values(processes) {
data = append(data, s.processProcess(proc))
}
paged, total := Paginate(r, data)
Success(w, chix.M{
"total": total,
"items": paged,
})
}
func (s *ProcessService) Kill(w http.ResponseWriter, r *http.Request) {
req, err := Bind[request.ProcessKill](r)
if err != nil {
Error(w, http.StatusUnprocessableEntity, "%v", err)
return
}
proc, err := process.NewProcess(req.PID)
if err != nil {
Error(w, http.StatusInternalServerError, "%v", err)
return
}
if err = proc.Kill(); err != nil {
Error(w, http.StatusInternalServerError, "%v", err)
return
}
Success(w, nil)
}
// processProcess 处理进程数据
func (s *ProcessService) processProcess(proc *process.Process) types.ProcessData {
data := types.ProcessData{
PID: proc.Pid,
}
if name, err := proc.Name(); err == nil {
data.Name = name
} else {
data.Name = "<UNKNOWN>"
}
if username, err := proc.Username(); err == nil {
data.Username = username
}
data.PPID, _ = proc.Ppid()
data.Status, _ = proc.Status()
data.Background, _ = proc.Background()
if ct, err := proc.CreateTime(); err == nil {
data.StartTime = time.Unix(ct/1000, 0).Format(time.DateTime)
}
data.NumThreads, _ = proc.NumThreads()
data.CPU, _ = proc.CPUPercent()
if mem, err := proc.MemoryInfo(); err == nil {
data.RSS = mem.RSS
data.Data = mem.Data
data.VMS = mem.VMS
data.HWM = mem.HWM
data.Stack = mem.Stack
data.Locked = mem.Locked
data.Swap = mem.Swap
}
if ioStat, err := proc.IOCounters(); err == nil {
data.DiskWrite = ioStat.WriteBytes
data.DiskRead = ioStat.ReadBytes
}
data.Nets, _ = proc.NetIOCounters(false)
data.Connections, _ = proc.Connections()
data.CmdLine, _ = proc.Cmdline()
data.OpenFiles, _ = proc.OpenFiles()
data.Envs, _ = proc.Environ()
data.OpenFiles = slices.Compact(data.OpenFiles)
data.Envs = slices.Compact(data.Envs)
return data
}