mirror of
https://github.com/acepanel/panel.git
synced 2026-02-04 05:31:44 +08:00
111 lines
2.4 KiB
Go
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
|
|
}
|