TRKJ/mayfly-go
2
0
Fork 0
mirror of https://gitee.com/dromara/mayfly-go synced 2025-11-04 00:10:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

增加webssh及数据库查询

Browse Source
This commit is contained in:
meilin.huang
2021-01-08 15:37:32 +08:00
parent 111612b7f2
commit 4c2e6b6155
60 changed files with 4347 additions and 2222 deletions
Download Patch File Download Diff File Expand all files Collapse all files

91
controllers/machine.go
View File

@@ -1,19 +1,22 @@
package controllers
import (
"github.com/gorilla/websocket"
"mayfly-go/base"
"mayfly-go/base/ctx"
"mayfly-go/base/model"
"mayfly-go/machine"
"mayfly-go/models"
"net/http"
"strconv"
"github.com/gorilla/websocket"
)
type MachineController struct {
base.Controller
}
var upGrader = websocket.Upgrader{
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024 * 1024 * 10,
CheckOrigin: func(r *http.Request) bool {
@@ -22,69 +25,81 @@ var upGrader = websocket.Upgrader{
}
func (c *MachineController) Machines() {
c.ReturnData(true, func(account *base.LoginAccount) interface{} {
c.ReturnData(true, func(account *ctx.LoginAccount) interface{} {
return models.GetMachineList(c.GetPageParam())
})
}
func (c *MachineController) Run() {
c.ReturnData(true, func(account *base.LoginAccount) interface{} {
c.ReturnData(true, func(account *ctx.LoginAccount) interface{} {
cmd := c.GetString("cmd")
base.NotEmpty(cmd, "cmd不能为空")
model.NotEmpty(cmd, "cmd不能为空")
return machine.GetCli(c.GetMachineId()).Run(cmd)
res, err := c.getCli().Run(cmd)
model.BizErrIsNil(err, "执行命令失败")
return res
})
}
// 系统基本信息
func (c *MachineController) SysInfo() {
c.ReturnData(true, func(account *base.LoginAccount) interface{} {
return machine.GetSystemInfo(machine.GetCli(c.GetMachineId()))
c.ReturnData(true, func(account *ctx.LoginAccount) interface{} {
res, err := c.getCli().GetSystemInfo()
model.BizErrIsNil(err, "获取系统基本信息失败")
return res
})
}
// top命令信息
func (c *MachineController) Top() {
c.ReturnData(true, func(account *base.LoginAccount) interface{} {
return machine.GetTop(machine.GetCli(c.GetMachineId()))
c.ReturnData(true, func(account *ctx.LoginAccount) interface{} {
return c.getCli().GetTop()
})
}
func (c *MachineController) GetProcessByName() {
c.ReturnData(true, func(account *base.LoginAccount) interface{} {
c.ReturnData(true, func(account *ctx.LoginAccount) interface{} {
name := c.GetString("name")
base.NotEmpty(name, "name不能为空")
return machine.GetProcessByName(machine.GetCli(c.GetMachineId()), name)
model.NotEmpty(name, "name不能为空")
res, err := c.getCli().GetProcessByName(name)
model.BizErrIsNil(err, "获取失败")
return res
})
}
//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) WsSSH() {
wsConn, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
if err != nil {
panic(model.NewBizErr("获取requst responsewirte错误"))
}
cols, _ := c.GetInt("cols", 80)
rows, _ := c.GetInt("rows", 40)
sws, err := machine.NewLogicSshWsSession(cols, rows, c.getCli(), wsConn)
if sws == nil {
panic(model.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错误")
model.IsTrue(machineId > 0, "machineId错误")
return uint64(machineId)
}
func (c *MachineController) getCli() *machine.Cli {
cli, err := machine.GetCli(c.GetMachineId())
model.BizErrIsNil(err, "获取客户端错误")
return cli
}