mirror of
https://gitee.com/dromara/mayfly-go
synced 2025-11-03 16:00:25 +08:00
91 lines
2.2 KiB
Go
91 lines
2.2 KiB
Go
package controllers
|
|
|
|
import (
|
|
"github.com/gorilla/websocket"
|
|
"mayfly-go/base"
|
|
"mayfly-go/machine"
|
|
"mayfly-go/models"
|
|
"net/http"
|
|
"strconv"
|
|
)
|
|
|
|
type MachineController struct {
|
|
base.Controller
|
|
}
|
|
|
|
var upGrader = websocket.Upgrader{
|
|
ReadBufferSize: 1024,
|
|
WriteBufferSize: 1024 * 1024 * 10,
|
|
CheckOrigin: func(r *http.Request) bool {
|
|
return true
|
|
},
|
|
}
|
|
|
|
func (c *MachineController) Machines() {
|
|
c.ReturnData(true, func(account *base.LoginAccount) interface{} {
|
|
return models.GetMachineList(c.GetPageParam())
|
|
})
|
|
}
|
|
|
|
func (c *MachineController) Run() {
|
|
c.ReturnData(true, func(account *base.LoginAccount) interface{} {
|
|
cmd := c.GetString("cmd")
|
|
base.NotEmpty(cmd, "cmd不能为空")
|
|
|
|
return machine.GetCli(c.GetMachineId()).Run(cmd)
|
|
})
|
|
}
|
|
|
|
// 系统基本信息
|
|
func (c *MachineController) SysInfo() {
|
|
c.ReturnData(true, func(account *base.LoginAccount) interface{} {
|
|
return machine.GetSystemInfo(machine.GetCli(c.GetMachineId()))
|
|
})
|
|
}
|
|
|
|
// top命令信息
|
|
func (c *MachineController) Top() {
|
|
c.ReturnData(true, func(account *base.LoginAccount) interface{} {
|
|
return machine.GetTop(machine.GetCli(c.GetMachineId()))
|
|
})
|
|
}
|
|
|
|
func (c *MachineController) GetProcessByName() {
|
|
c.ReturnData(true, func(account *base.LoginAccount) interface{} {
|
|
name := c.GetString("name")
|
|
base.NotEmpty(name, "name不能为空")
|
|
return machine.GetProcessByName(machine.GetCli(c.GetMachineId()), name)
|
|
})
|
|
}
|
|
|
|
//func (c *MachineController) WsSSH() {
|
|
// wsConn, err := upGrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
|
|
// if err != nil {
|
|
// panic(base.NewBizErr("获取requst responsewirte错误"))
|
|
// }
|
|
//
|
|
// cols, _ := c.GetInt("col", 80)
|
|
// rows, _ := c.GetInt("rows", 40)
|
|
//
|
|
// sws, err := machine.NewLogicSshWsSession(cols, rows, true, machine.GetCli(c.GetMachineId()), wsConn)
|
|
// if sws == nil {
|
|
// panic(base.NewBizErr("连接失败"))
|
|
// }
|
|
// //if wshandleError(wsConn, err) {
|
|
// // return
|
|
// //}
|
|
// defer sws.Close()
|
|
//
|
|
// quitChan := make(chan bool, 3)
|
|
// sws.Start(quitChan)
|
|
// go sws.Wait(quitChan)
|
|
//
|
|
// <-quitChan
|
|
//}
|
|
|
|
func (c *MachineController) GetMachineId() uint64 {
|
|
machineId, _ := strconv.Atoi(c.Ctx.Input.Param(":machineId"))
|
|
base.IsTrue(machineId > 0, "machineId错误")
|
|
return uint64(machineId)
|
|
}
|