mirror of https://github.com/ouqiang/gocron
181 lines
5.2 KiB
Go
181 lines
5.2 KiB
Go
package host
|
|
|
|
import (
|
|
"gopkg.in/macaron.v1"
|
|
"github.com/ouqiang/gocron/models"
|
|
"github.com/ouqiang/gocron/modules/utils"
|
|
"github.com/ouqiang/gocron/modules/logger"
|
|
"strconv"
|
|
"github.com/ouqiang/gocron/modules/ssh"
|
|
"github.com/ouqiang/gocron/service"
|
|
"github.com/Unknwon/paginater"
|
|
"fmt"
|
|
"html/template"
|
|
"github.com/ouqiang/gocron/routers/base"
|
|
)
|
|
|
|
func Index(ctx *macaron.Context) {
|
|
hostModel := new(models.Host)
|
|
queryParams := parseQueryParams(ctx)
|
|
total, err := hostModel.Total(queryParams)
|
|
hosts, err := hostModel.List(queryParams)
|
|
if err != nil {
|
|
logger.Error(err)
|
|
}
|
|
name, ok := queryParams["name"].(string)
|
|
var safeNameHTML = ""
|
|
if ok {
|
|
safeNameHTML = template.HTMLEscapeString(name)
|
|
}
|
|
PageParams := fmt.Sprintf("id=%d&name=%s&page_size=%d",
|
|
queryParams["Id"], safeNameHTML, queryParams["PageSize"]);
|
|
queryParams["PageParams"] = template.URL(PageParams)
|
|
p := paginater.New(int(total), queryParams["PageSize"].(int), queryParams["Page"].(int), 5)
|
|
ctx.Data["Pagination"] = p
|
|
ctx.Data["Title"] = "主机列表"
|
|
ctx.Data["Hosts"] = hosts
|
|
ctx.Data["Params"] = queryParams
|
|
ctx.HTML(200, "host/index")
|
|
}
|
|
|
|
func Create(ctx *macaron.Context) {
|
|
ctx.Data["Title"] = "添加主机"
|
|
ctx.HTML(200, "host/host_form")
|
|
}
|
|
|
|
func Edit(ctx *macaron.Context) {
|
|
ctx.Data["Title"] = "编辑主机"
|
|
hostModel := new(models.Host)
|
|
id := ctx.ParamsInt(":id")
|
|
err := hostModel.Find(id)
|
|
if err != nil {
|
|
logger.Errorf("获取主机详情失败#主机id-%d", id)
|
|
}
|
|
ctx.Data["Host"] = hostModel
|
|
ctx.HTML(200, "host/host_form")
|
|
}
|
|
|
|
func Ping(ctx *macaron.Context) string {
|
|
id := ctx.ParamsInt(":id")
|
|
hostModel := new(models.Host)
|
|
err := hostModel.Find(id)
|
|
json := utils.JsonResponse{}
|
|
if err != nil || hostModel.Id <= 0{
|
|
return json.CommonFailure("主机不存在", err)
|
|
}
|
|
|
|
sshConfig := ssh.SSHConfig{}
|
|
sshConfig.User = hostModel.Username
|
|
sshConfig.Host = hostModel.Name
|
|
sshConfig.Port = hostModel.Port
|
|
sshConfig.ExecTimeout = 5
|
|
sshConfig.AuthType = hostModel.AuthType
|
|
var password string
|
|
var privateKey string
|
|
if hostModel.AuthType == ssh.HostPassword {
|
|
password, err = hostModel.GetPasswordByHost(hostModel.Name)
|
|
if err != nil {
|
|
return json.CommonFailure(err.Error(), err)
|
|
}
|
|
sshConfig.Password = password
|
|
} else {
|
|
privateKey, err = hostModel.GetPrivateKeyByHost(hostModel.Name)
|
|
if err != nil {
|
|
return json.CommonFailure(err.Error(), err)
|
|
}
|
|
sshConfig.PrivateKey = privateKey
|
|
}
|
|
|
|
_, err = ssh.Exec(sshConfig, "pwd")
|
|
if err != nil {
|
|
return json.CommonFailure("连接失败-" + err.Error(), err)
|
|
}
|
|
|
|
return json.Success("连接成功", nil)
|
|
}
|
|
|
|
type HostForm struct {
|
|
Id int16
|
|
Name string `binding:"Required;MaxSize(100)"`
|
|
Alias string `binding:"Required;MaxSize(32)"`
|
|
Username string `binding:"Required;MaxSize(32)"`
|
|
Port int `binding:"Required;Range(1-65535)"`
|
|
AuthType ssh.HostAuthType `binding:"Required:Range(1,2)"`
|
|
Remark string
|
|
}
|
|
|
|
func Store(ctx *macaron.Context, form HostForm) string {
|
|
json := utils.JsonResponse{}
|
|
hostModel := new(models.Host)
|
|
id := form.Id
|
|
nameExist, err := hostModel.NameExists(form.Name, form.Id)
|
|
if err != nil {
|
|
return json.CommonFailure("操作失败", err)
|
|
}
|
|
if nameExist {
|
|
return json.CommonFailure("主机名已存在")
|
|
}
|
|
|
|
hostModel.Name = form.Name
|
|
hostModel.Alias = form.Alias
|
|
hostModel.Username = form.Username
|
|
hostModel.Port = form.Port
|
|
hostModel.Remark = form.Remark
|
|
hostModel.AuthType = form.AuthType
|
|
isCreate := false
|
|
if id > 0 {
|
|
_, err = hostModel.UpdateBean(id)
|
|
} else {
|
|
isCreate = true
|
|
id, err = hostModel.Create()
|
|
}
|
|
if err != nil {
|
|
return json.CommonFailure("保存失败", err)
|
|
}
|
|
|
|
taskModel := new(models.TaskHost)
|
|
tasks, err := taskModel.ActiveListByHostId(id)
|
|
if err != nil {
|
|
return json.CommonFailure("刷新任务主机信息失败", err)
|
|
}
|
|
if !isCreate && len(tasks) > 0 {
|
|
serviceTask := new(service.Task)
|
|
serviceTask.BatchAdd(tasks)
|
|
}
|
|
|
|
return json.Success("保存成功", nil)
|
|
}
|
|
|
|
func Remove(ctx *macaron.Context) string {
|
|
id, err := strconv.Atoi(ctx.Params(":id"))
|
|
json := utils.JsonResponse{}
|
|
if err != nil {
|
|
return json.CommonFailure("参数错误", err)
|
|
}
|
|
taskModel := new(models.Task)
|
|
exist,err := taskModel.HostIdExist(int16(id))
|
|
if err != nil {
|
|
return json.CommonFailure("操作失败", err)
|
|
}
|
|
if exist {
|
|
return json.CommonFailure("有任务引用此主机,不能删除")
|
|
}
|
|
|
|
hostModel := new(models.Host)
|
|
_, err =hostModel.Delete(id)
|
|
if err != nil {
|
|
return json.CommonFailure("操作失败", err)
|
|
}
|
|
|
|
return json.Success("操作成功", nil)
|
|
}
|
|
|
|
// 解析查询参数
|
|
func parseQueryParams(ctx *macaron.Context) (models.CommonMap) {
|
|
var params models.CommonMap = models.CommonMap{}
|
|
params["Id"] = ctx.QueryInt("id")
|
|
params["Name"] = ctx.QueryTrim("name")
|
|
base.ParsePageAndPageSize(ctx, params)
|
|
|
|
return params
|
|
} |