package/go_process_manager
1
0
Fork 0
mirror of https://github.com/lzh-1625/go_process_manager.git synced 2025-10-31 19:32:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
17689440205
2025-01-26 16:43:51 +08:00
commit 460a1f428b
99 changed files with 14911 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

107
internal/app/api/proc.go Normal file
View File

@@ -0,0 +1,107 @@
package api
import (
"github.com/lzh-1625/go_process_manager/internal/app/model"
"github.com/lzh-1625/go_process_manager/internal/app/repository"
"github.com/lzh-1625/go_process_manager/internal/app/service"
"github.com/gin-gonic/gin"
)
type procApi struct{}
var ProcApi = new(procApi)
func (p *procApi) CreateNewProcess(ctx *gin.Context, req model.Process) {
index, err := repository.ProcessRepository.AddProcessConfig(req)
errCheck(ctx, err != nil, err)
req.Uuid = index
proc, err := service.ProcessCtlService.RunNewProcess(req)
errCheck(ctx, err != nil, err)
service.ProcessCtlService.AddProcess(req.Uuid, proc)
rOk(ctx, "Operation successful!", gin.H{
"id": req.Uuid,
})
}
func (p *procApi) DeleteNewProcess(ctx *gin.Context) {
uuid := getQueryInt(ctx, "uuid")
service.ProcessCtlService.KillProcess(uuid)
service.ProcessCtlService.DeleteProcess(uuid)
err := repository.ProcessRepository.DeleteProcessConfig(uuid)
errCheck(ctx, err != nil, err)
rOk(ctx, "Operation successful!", nil)
}
func (p *procApi) KillProcess(ctx *gin.Context) {
uuid := getQueryInt(ctx, "uuid")
err := service.ProcessCtlService.KillProcess(uuid)
errCheck(ctx, err != nil, err)
rOk(ctx, "Operation successful!", nil)
}
func (p *procApi) StartProcess(ctx *gin.Context) {
uuid := getQueryInt(ctx, "uuid")
prod, err := service.ProcessCtlService.GetProcess(uuid)
if err != nil { // 进程不存在则创建
proc, err := service.ProcessCtlService.RunNewProcess(repository.ProcessRepository.GetProcessConfigById(uuid))
errCheck(ctx, err != nil, err)
service.ProcessCtlService.AddProcess(uuid, proc)
rOk(ctx, "Operation successful!", nil)
return
}
errCheck(ctx, prod.State.State == 1, "The process is currently running.")
prod.ResetRestartTimes()
err = prod.Start()
errCheck(ctx, err != nil, err)
rOk(ctx, "Operation successful!", nil)
}
func (p *procApi) StartAllProcess(ctx *gin.Context) {
if isAdmin(ctx) {
service.ProcessCtlService.ProcessStartAll()
} else {
service.ProcessCtlService.ProcesStartAllByUsername(getUserName(ctx))
}
rOk(ctx, "Operation successful!", nil)
}
func (p *procApi) KillAllProcess(ctx *gin.Context) {
if isAdmin(ctx) {
service.ProcessCtlService.KillAllProcess()
} else {
service.ProcessCtlService.KillAllProcessByUserName(getUserName(ctx))
}
rOk(ctx, "Operation successful!", nil)
}
func (p *procApi) GetProcessList(ctx *gin.Context) {
if isAdmin(ctx) {
rOk(ctx, "Query successful!", service.ProcessCtlService.GetProcessList())
} else {
rOk(ctx, "Query successful!", service.ProcessCtlService.GetProcessListByUser(getUserName(ctx)))
}
}
func (p *procApi) UpdateProcessConfig(ctx *gin.Context, req model.Process) {
service.ProcessCtlService.UpdateProcessConfig(req)
err := repository.ProcessRepository.UpdateProcessConfig(req)
errCheck(ctx, err != nil, err)
rOk(ctx, "Operation successful!", nil)
}
func (p *procApi) GetProcessConfig(ctx *gin.Context) {
uuid := getQueryInt(ctx, "uuid")
data := repository.ProcessRepository.GetProcessConfigById(uuid)
errCheck(ctx, data.Uuid == 0, "No information found!")
rOk(ctx, "Query successful!", data)
}
func (p *procApi) ProcessControl(ctx *gin.Context) {
user := getUserName(ctx)
uuid := getQueryInt(ctx, "uuid")
proc, err := service.ProcessCtlService.GetProcess(uuid)
errCheck(ctx, err != nil, err)
proc.ProcessControl(user)
rOk(ctx, "Operation successful!", nil)
}