初次提交

2025-10-05 16:06:51 +08:00 · 2024-06-26 20:45:23 +08:00
parent 4b388a5be1
commit 831ea9889f
57 changed files with 3945 additions and 0 deletions
--- a/api/proc.go
+++ b/api/proc.go
@@ -0,0 +1,101 @@
+package api
+
+import (
+	"msm/consts/ctxflag"
+	"msm/consts/role"
+	"msm/dao"
+	"msm/model"
+	"msm/service/process"
+	"strconv"
+
+	"github.com/gin-gonic/gin"
+)
+
+type procApi struct{}
+
+var ProcApi = new(procApi)
+
+func (p *procApi) CreateNewProcess(ctx *gin.Context) {
+	req := model.Process{}
+	ctx.ShouldBindJSON(&req)
+	index, err := dao.ProcessDao.AddProcessConfig(req)
+	errCheck(ctx, err != nil, err)
+	req.Uuid = index
+	proc, err := process.RunNewProcess(req)
+	errCheck(ctx, err != nil, err)
+	process.ProcessCtlService.AddProcess(req.Uuid, proc)
+	rOk(ctx, "创建成功", gin.H{
+		"id": req.Uuid,
+	})
+}
+
+func (p *procApi) DeleteNewProcess(ctx *gin.Context) {
+	uuid, err := strconv.Atoi(ctx.Query("uuid"))
+	errCheck(ctx, err != nil, "参数有误")
+	process.ProcessCtlService.KillProcess(uuid)
+	process.ProcessCtlService.DeleteProcess(uuid)
+	err = dao.ProcessDao.DeleteProcessConfig(uuid)
+	errCheck(ctx, err != nil, err)
+	rOk(ctx, "删除成功", nil)
+}
+
+func (p *procApi) KillProcess(ctx *gin.Context) {
+	uuid, err := strconv.Atoi(ctx.Query("uuid"))
+	errCheck(ctx, err != nil, "参数有误")
+	err = process.ProcessCtlService.KillProcess(uuid)
+	errCheck(ctx, err != nil, err)
+	rOk(ctx, "成功", nil)
+}
+
+func (p *procApi) StartProcess(ctx *gin.Context) {
+	uuid, err := strconv.Atoi(ctx.Query("uuid"))
+	errCheck(ctx, err != nil, "参数有误")
+	prod, err := process.ProcessCtlService.GetProcess(uuid)
+	if err != nil { // 进程不存在则创建
+		proc, err := process.RunNewProcess(dao.ProcessDao.GetProcessConfigById(uuid))
+		errCheck(ctx, err != nil, err)
+		process.ProcessCtlService.AddProcess(uuid, proc)
+		rOk(ctx, "成功", nil)
+		return
+	}
+	errCheck(ctx, prod.GetStateState() == 1, "进程还在运行中")
+	prod.ResetRestartTimes()
+	prod.ReStart()
+	// dao.UpdateServerAutoStart(uuid, true)
+	rOk(ctx, "成功", nil)
+}
+
+func (p *procApi) GetProcessList(ctx *gin.Context) {
+	if ctx.GetInt(ctxflag.ROLE) < int(role.USER) {
+		rOk(ctx, "进程列表获取成功", process.ProcessCtlService.GetProcessList())
+	} else {
+		rOk(ctx, "进程列表获取成功", process.ProcessCtlService.GetProcessListByUser(ctx.GetString(ctxflag.USER_NAME)))
+	}
+}
+
+func (p *procApi) UpdateProcessConfig(ctx *gin.Context) {
+	req := model.Process{}
+	ctx.ShouldBindJSON(&req)
+	process.ProcessCtlService.UpdateProcessConfig(req)
+	err := dao.ProcessDao.UpdateProcessConfig(req)
+	errCheck(ctx, err != nil, err)
+	rOk(ctx, "更改配置成功", nil)
+}
+
+func (p *procApi) GetProcessConfig(ctx *gin.Context) {
+	uuid, err := strconv.Atoi(ctx.Query("uuid"))
+	errCheck(ctx, err != nil, "参数有误")
+	data := dao.ProcessDao.GetProcessConfigById(uuid)
+	errCheck(ctx, data.Uuid == 0, "未查询到信息")
+	rOk(ctx, "success", data)
+}
+
+func (p *procApi) ProcessControl(ctx *gin.Context) {
+	user := ctx.GetString(ctxflag.USER_NAME)
+	uuid, err := strconv.Atoi(ctx.Query("uuid"))
+	errCheck(ctx, err != nil, "参数有误")
+	proc, err := process.ProcessCtlService.GetProcess(uuid)
+	errCheck(ctx, err != nil, "进程控制权获取失败")
+	proc.ProcessControl(user)
+	rOk(ctx, "获取进程控权成功", nil)
+}