461 lines
13 KiB
Go
461 lines
13 KiB
Go
/************************************************************
|
||
** @Description: controllers
|
||
** @Author: haodaquan
|
||
** @Date: 2018-06-09 16:11
|
||
** @Last Modified by: Bee
|
||
** @Last Modified time: 2019-02-17 22:15:15
|
||
*************************************************************/
|
||
package controllers
|
||
|
||
import (
|
||
"github.com/astaxie/beego/logs"
|
||
"github.com/george518/PPGo_Job/libs"
|
||
"github.com/george518/PPGo_Job/models"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
)
|
||
|
||
type ServerController struct {
|
||
BaseController
|
||
}
|
||
|
||
func (self *ServerController) List() {
|
||
self.Data["pageTitle"] = "执行资源管理"
|
||
self.Data["serverGroup"] = serverGroupLists(self.serverGroups, self.userId)
|
||
self.display()
|
||
}
|
||
|
||
func (self *ServerController) Add() {
|
||
self.Data["pageTitle"] = "新增执行资源"
|
||
self.Data["serverGroup"] = serverGroupLists(self.serverGroups, self.userId)
|
||
self.display()
|
||
}
|
||
|
||
func (self *ServerController) GetServerByGroupId() {
|
||
gid, _ := self.GetInt("gid", 0)
|
||
if gid == 0 {
|
||
self.ajaxMsg("groupId is not exist", MSG_ERR)
|
||
}
|
||
|
||
//列表
|
||
page, err := self.GetInt("page")
|
||
if err != nil {
|
||
page = 1
|
||
}
|
||
limit, err := self.GetInt("limit")
|
||
if err != nil {
|
||
limit = 30
|
||
}
|
||
//serverName := strings.TrimSpace(self.GetString("serverName"))
|
||
StatusText := []string{
|
||
"正常",
|
||
"<font color='red'>禁用</font>",
|
||
}
|
||
|
||
loginType := [2]string{
|
||
"密码",
|
||
"密钥",
|
||
}
|
||
|
||
serverGroup := serverGroupLists(self.serverGroups, self.userId)
|
||
|
||
self.pageSize = limit
|
||
//查询条件
|
||
filters := make([]interface{}, 0)
|
||
filters = append(filters, "status", 0)
|
||
filters = append(filters, "group_id", gid)
|
||
|
||
result, count := models.TaskServerGetList(page, self.pageSize, filters...)
|
||
list := make([]map[string]interface{}, len(result))
|
||
for k, v := range result {
|
||
row := make(map[string]interface{})
|
||
row["id"] = v.Id
|
||
row["connection_type"] = v.ConnectionType
|
||
row["server_name"] = v.ServerName
|
||
row["detail"] = v.Detail
|
||
if serverGroup[v.GroupId] == "" {
|
||
v.GroupId = 0
|
||
}
|
||
row["group_name"] = serverGroup[v.GroupId]
|
||
row["type"] = loginType[v.Type]
|
||
row["status"] = v.Status
|
||
row["status_text"] = StatusText[v.Status]
|
||
list[k] = row
|
||
}
|
||
|
||
self.ajaxList("成功", MSG_OK, count, list)
|
||
}
|
||
|
||
func (self *ServerController) Edit() {
|
||
self.Data["pageTitle"] = "编辑执行资源"
|
||
|
||
id, _ := self.GetInt("id", 0)
|
||
server, _ := models.TaskServerGetById(id)
|
||
row := make(map[string]interface{})
|
||
row["id"] = server.Id
|
||
row["connection_type"] = server.ConnectionType
|
||
row["server_name"] = server.ServerName
|
||
row["group_id"] = server.GroupId
|
||
row["server_ip"] = server.ServerIp
|
||
row["server_account"] = server.ServerAccount
|
||
row["server_outer_ip"] = server.ServerOuterIp
|
||
row["port"] = server.Port
|
||
row["type"] = server.Type
|
||
row["password"] = server.Password
|
||
row["public_key_src"] = server.PublicKeySrc
|
||
row["private_key_src"] = server.PrivateKeySrc
|
||
row["detail"] = server.Detail
|
||
self.Data["server"] = row
|
||
self.Data["serverGroup"] = serverGroupLists(self.serverGroups, self.userId)
|
||
self.display()
|
||
}
|
||
|
||
func (self *ServerController) AjaxTestServer() {
|
||
|
||
server := new(models.TaskServer)
|
||
server.ConnectionType, _ = self.GetInt("connection_type")
|
||
server.ServerName = strings.TrimSpace(self.GetString("server_name"))
|
||
server.ServerAccount = strings.TrimSpace(self.GetString("server_account"))
|
||
server.ServerOuterIp = strings.TrimSpace(self.GetString("server_outer_ip"))
|
||
server.ServerIp = strings.TrimSpace(self.GetString("server_ip"))
|
||
server.PrivateKeySrc = strings.TrimSpace(self.GetString("private_key_src"))
|
||
server.PublicKeySrc = strings.TrimSpace(self.GetString("public_key_src"))
|
||
server.Password = strings.TrimSpace(self.GetString("password"))
|
||
server.Detail = strings.TrimSpace(self.GetString("detail"))
|
||
server.Type, _ = self.GetInt("type")
|
||
server.Port, _ = self.GetInt("port")
|
||
server.GroupId, _ = self.GetInt("group_id")
|
||
|
||
var err error
|
||
|
||
if server.ConnectionType == 0 {
|
||
if server.Type == 0 {
|
||
//密码登录
|
||
err = libs.RemoteCommandByPassword(server)
|
||
}
|
||
|
||
if server.Type == 1 {
|
||
//密钥登录
|
||
err = libs.RemoteCommandByKey(server)
|
||
}
|
||
|
||
if err != nil {
|
||
self.ajaxMsg(err.Error(), MSG_ERR)
|
||
}
|
||
self.ajaxMsg("Success", MSG_OK)
|
||
} else if server.ConnectionType == 1 {
|
||
if server.Type == 0 {
|
||
//密码登录
|
||
err = libs.RemoteCommandByTelnetPassword(server)
|
||
} else {
|
||
self.ajaxMsg("Telnet方式暂不支持密钥登陆！", MSG_ERR)
|
||
}
|
||
|
||
if err != nil {
|
||
self.ajaxMsg(err.Error(), MSG_ERR)
|
||
}
|
||
self.ajaxMsg("Success", MSG_OK)
|
||
} else if server.ConnectionType == 2 {
|
||
|
||
if err := libs.RemoteAgent(server); err != nil {
|
||
self.ajaxMsg(err.Error(), MSG_ERR)
|
||
} else {
|
||
self.ajaxMsg("Success", MSG_OK)
|
||
|
||
}
|
||
}
|
||
|
||
self.ajaxMsg("未知连接方式", MSG_ERR)
|
||
}
|
||
|
||
func (self *ServerController) Copy() {
|
||
self.Data["pageTitle"] = "复制服务器资源"
|
||
|
||
id, _ := self.GetInt("id", 0)
|
||
server, _ := models.TaskServerGetById(id)
|
||
row := make(map[string]interface{})
|
||
row["id"] = server.Id
|
||
row["connection_type"] = server.ConnectionType
|
||
row["server_name"] = server.ServerName
|
||
row["group_id"] = server.GroupId
|
||
row["server_ip"] = server.ServerIp
|
||
row["server_account"] = server.ServerAccount
|
||
row["server_outer_ip"] = server.ServerOuterIp
|
||
row["port"] = server.Port
|
||
row["type"] = server.Type
|
||
row["password"] = server.Password
|
||
row["public_key_src"] = server.PublicKeySrc
|
||
row["private_key_src"] = server.PrivateKeySrc
|
||
row["detail"] = server.Detail
|
||
self.Data["server"] = row
|
||
self.Data["serverGroup"] = serverGroupLists(self.serverGroups, self.userId)
|
||
self.display()
|
||
}
|
||
|
||
func (self *ServerController) AjaxSave() {
|
||
server_id, _ := self.GetInt("id")
|
||
if server_id == 0 {
|
||
server := new(models.TaskServer)
|
||
server.ConnectionType, _ = self.GetInt("connection_type")
|
||
server.ServerName = strings.TrimSpace(self.GetString("server_name"))
|
||
server.ServerAccount = strings.TrimSpace(self.GetString("server_account"))
|
||
server.ServerOuterIp = strings.TrimSpace(self.GetString("server_outer_ip"))
|
||
server.ServerIp = strings.TrimSpace(self.GetString("server_ip"))
|
||
server.PrivateKeySrc = strings.TrimSpace(self.GetString("private_key_src"))
|
||
server.PublicKeySrc = strings.TrimSpace(self.GetString("public_key_src"))
|
||
server.Password = strings.TrimSpace(self.GetString("password"))
|
||
|
||
server.Detail = strings.TrimSpace(self.GetString("detail"))
|
||
server.Type, _ = self.GetInt("type")
|
||
server.Port, _ = self.GetInt("port")
|
||
server.GroupId, _ = self.GetInt("group_id")
|
||
|
||
server.CreateTime = time.Now().Unix()
|
||
server.UpdateTime = time.Now().Unix()
|
||
server.Status = 0
|
||
|
||
if _, err := models.TaskServerAdd(server); err != nil {
|
||
self.ajaxMsg(err.Error(), MSG_ERR)
|
||
}
|
||
self.ajaxMsg("", MSG_OK)
|
||
}
|
||
|
||
server, _ := models.TaskServerGetById(server_id)
|
||
//修改
|
||
server.Id = server_id
|
||
server.UpdateTime = time.Now().Unix()
|
||
|
||
server.ConnectionType, _ = self.GetInt("connection_type")
|
||
server.ServerName = strings.TrimSpace(self.GetString("server_name"))
|
||
server.ServerAccount = strings.TrimSpace(self.GetString("server_account"))
|
||
server.ServerOuterIp = strings.TrimSpace(self.GetString("server_outer_ip"))
|
||
server.ServerIp = strings.TrimSpace(self.GetString("server_ip"))
|
||
server.PrivateKeySrc = strings.TrimSpace(self.GetString("private_key_src"))
|
||
server.PublicKeySrc = strings.TrimSpace(self.GetString("public_key_src"))
|
||
server.Detail = strings.TrimSpace(self.GetString("detail"))
|
||
server.Password = strings.TrimSpace(self.GetString("password"))
|
||
|
||
server.Type, _ = self.GetInt("type")
|
||
server.Port, _ = self.GetInt("port")
|
||
server.GroupId, _ = self.GetInt("group_id")
|
||
|
||
if err := server.Update(); err != nil {
|
||
self.ajaxMsg(err.Error(), MSG_ERR)
|
||
}
|
||
self.ajaxMsg("", MSG_OK)
|
||
}
|
||
|
||
func (self *ServerController) AjaxDel() {
|
||
id, _ := self.GetInt("id")
|
||
|
||
if id == 1 {
|
||
self.ajaxMsg("默认分组id=1，禁止删除", MSG_ERR)
|
||
}
|
||
|
||
server, _ := models.TaskServerGetById(id)
|
||
server.UpdateTime = time.Now().Unix()
|
||
server.Status = 2
|
||
server.Id = id
|
||
|
||
//TODO 查询服务器是否用于定时任务
|
||
if err := server.Update(); err != nil {
|
||
self.ajaxMsg(err.Error(), MSG_ERR)
|
||
}
|
||
self.ajaxMsg("操作成功", MSG_OK)
|
||
}
|
||
|
||
func (self *ServerController) Table() {
|
||
//列表
|
||
page, err := self.GetInt("page")
|
||
if err != nil {
|
||
page = 1
|
||
}
|
||
limit, err := self.GetInt("limit")
|
||
if err != nil {
|
||
limit = 30
|
||
}
|
||
|
||
serverGroupId, err := self.GetInt("serverGroupId")
|
||
|
||
serverName := strings.TrimSpace(self.GetString("serverName"))
|
||
StatusText := []string{
|
||
"<i class='fa fa-refresh' style='color:#5FB878'></i>",
|
||
"<i class='fa fa-ban' style='color:#FF5722'></i>",
|
||
}
|
||
//
|
||
//loginType := [2]string{
|
||
// "密码",
|
||
// "密钥",
|
||
//}
|
||
|
||
connectionType := [3]string{
|
||
"SSH",
|
||
"Telnet",
|
||
"Agent",
|
||
}
|
||
|
||
serverGroup := serverGroupLists(self.serverGroups, self.userId)
|
||
|
||
self.pageSize = limit
|
||
//查询条件
|
||
filters := make([]interface{}, 0)
|
||
ids := []int{0, 1}
|
||
filters = append(filters, "status__in", ids)
|
||
|
||
groupsIds := make([]int, 0)
|
||
if self.userId != 1 {
|
||
groups := strings.Split(self.serverGroups, ",")
|
||
|
||
for _, v := range groups {
|
||
id, _ := strconv.Atoi(v)
|
||
if serverGroupId > 0 {
|
||
if id == serverGroupId {
|
||
groupsIds = append(groupsIds, id)
|
||
break
|
||
}
|
||
} else {
|
||
groupsIds = append(groupsIds, id)
|
||
}
|
||
}
|
||
filters = append(filters, "group_id__in", groupsIds)
|
||
} else if serverGroupId > 0 {
|
||
groupsIds = append(groupsIds, serverGroupId)
|
||
filters = append(filters, "group_id__in", groupsIds)
|
||
}
|
||
|
||
if serverName != "" {
|
||
filters = append(filters, "server_name__icontains", serverName)
|
||
}
|
||
result, count := models.TaskServerGetList(page, self.pageSize, filters...)
|
||
list := make([]map[string]interface{}, len(result))
|
||
for k, v := range result {
|
||
row := make(map[string]interface{})
|
||
row["id"] = v.Id
|
||
row["connection_type"] = connectionType[v.ConnectionType]
|
||
row["server_name"] = StatusText[v.Status] + " " + v.ServerName
|
||
row["detail"] = v.Detail
|
||
if serverGroup[v.GroupId] == "" {
|
||
v.GroupId = 0
|
||
}
|
||
row["ip_port"] = v.ServerIp + ":" + strconv.Itoa(v.Port)
|
||
row["group_name"] = serverGroup[v.GroupId]
|
||
//row["type"] = loginType[v.Type]
|
||
row["status"] = v.Status
|
||
list[k] = row
|
||
}
|
||
|
||
self.ajaxList("成功", MSG_OK, count, list)
|
||
}
|
||
|
||
//以下函数为执行器接口
|
||
//注册
|
||
func (self *ServerController) ApiSave() {
|
||
//唯一确定值 ip+port
|
||
serverIp := strings.TrimSpace(self.GetString("server_ip"))
|
||
port, _ := self.GetInt("port")
|
||
|
||
if serverIp == "" || port == 0 {
|
||
self.ajaxMsg("执行器和端口号必填", MSG_ERR)
|
||
}
|
||
|
||
defaultActName := "agent-" + serverIp + "-" + strconv.Itoa(port)
|
||
|
||
id := models.TaskServerForActuator(serverIp, port)
|
||
if id == 0 {
|
||
//新增
|
||
server := new(models.TaskServer)
|
||
server.ConnectionType, _ = self.GetInt("connection_type", 3)
|
||
server.ServerName = strings.TrimSpace(self.GetString("server_name", defaultActName))
|
||
server.ServerAccount = strings.TrimSpace(self.GetString("server_account", "agent"))
|
||
server.ServerOuterIp = strings.TrimSpace(self.GetString("server_outer_ip", ""))
|
||
server.ServerIp = strings.TrimSpace(self.GetString("server_ip"))
|
||
server.PrivateKeySrc = strings.TrimSpace(self.GetString("private_key_src", ""))
|
||
server.PublicKeySrc = strings.TrimSpace(self.GetString("public_key_src", ""))
|
||
server.Password = strings.TrimSpace(self.GetString("password", "agent"))
|
||
|
||
server.Detail = strings.TrimSpace(self.GetString("detail", ""))
|
||
server.Type, _ = self.GetInt("type", 0)
|
||
server.Port, _ = self.GetInt("port")
|
||
server.GroupId, _ = self.GetInt("group_id", 0)
|
||
server.Status = 0
|
||
|
||
server.CreateTime = time.Now().Unix()
|
||
server.UpdateTime = time.Now().Unix()
|
||
server.Status = 0
|
||
serverId, err := models.TaskServerAdd(server)
|
||
if err != nil {
|
||
self.ajaxMsg(err.Error(), MSG_ERR)
|
||
}
|
||
self.ajaxMsg(serverId, MSG_OK)
|
||
} else {
|
||
//修改状态
|
||
server, _ := models.TaskServerGetById(id)
|
||
server.UpdateTime = time.Now().Unix()
|
||
server.Status, _ = self.GetInt("status", 0)
|
||
if err := server.Update(); err != nil {
|
||
self.ajaxMsg(err.Error(), MSG_ERR)
|
||
}
|
||
self.ajaxMsg(id, MSG_OK)
|
||
}
|
||
|
||
}
|
||
|
||
//检测0-正常，1-异常，2-删除
|
||
func (self *ServerController) ApiStatus() {
|
||
//唯一确定值 ip+port
|
||
serverId := strings.TrimSpace(self.GetString("server_ip"))
|
||
port, _ := self.GetInt("port")
|
||
status, _ := self.GetInt("status", 0)
|
||
|
||
if serverId == "" || port == 0 {
|
||
self.ajaxMsg("执行器和端口号必填", MSG_ERR)
|
||
}
|
||
|
||
id := models.TaskServerForActuator(serverId, port)
|
||
if id == 0 {
|
||
self.ajaxMsg("执行器不存在", MSG_ERR)
|
||
}
|
||
|
||
if status != 0 && status != 1 {
|
||
status = 0
|
||
}
|
||
|
||
server, _ := models.TaskServerGetById(id)
|
||
server.UpdateTime = time.Now().Unix()
|
||
server.Status = status
|
||
server.Id = id
|
||
|
||
logs.Info(server)
|
||
|
||
//TODO 查询执行器是否正在使用中
|
||
if err := server.Update(); err != nil {
|
||
self.ajaxMsg(err.Error(), MSG_ERR)
|
||
}
|
||
self.ajaxMsg(id, MSG_OK)
|
||
}
|
||
|
||
//获取 不检测执行器状态
|
||
func (self *ServerController) ApiGet() {
|
||
//唯一确定值 ip+port
|
||
serverId := strings.TrimSpace(self.GetString("server_ip"))
|
||
port, _ := self.GetInt("port")
|
||
|
||
if serverId == "" || port == 0 {
|
||
self.ajaxMsg("执行器和端口号必填", MSG_ERR)
|
||
}
|
||
|
||
id := models.TaskServerForActuator(serverId, port)
|
||
if id == 0 {
|
||
self.ajaxMsg("执行器不存在", MSG_ERR)
|
||
}
|
||
|
||
server, err := models.TaskServerGetById(id)
|
||
|
||
if err != nil {
|
||
self.ajaxMsg(err.Error(), MSG_ERR)
|
||
} else {
|
||
self.ajaxMsg(server, MSG_OK)
|
||
}
|
||
}
|