first

controllers/machine.go

@@ -0,0 +1,90 @@
+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)
+}