2017-03-10 09:24:06 +00:00
|
|
|
package service
|
|
|
|
|
|
|
|
import (
|
2017-04-07 01:22:00 +00:00
|
|
|
"github.com/ouqiang/gocron/models"
|
2017-04-02 02:38:49 +00:00
|
|
|
"io/ioutil"
|
|
|
|
"net/http"
|
|
|
|
"strconv"
|
|
|
|
"time"
|
2017-04-07 01:22:00 +00:00
|
|
|
"github.com/ouqiang/gocron/modules/logger"
|
2017-04-10 09:37:16 +00:00
|
|
|
"github.com/ouqiang/gocron/modules/ssh"
|
2017-04-13 09:35:59 +00:00
|
|
|
"github.com/jakecoffman/cron"
|
2017-03-10 09:24:06 +00:00
|
|
|
)
|
|
|
|
|
2017-04-13 09:35:59 +00:00
|
|
|
var Cron *cron.Cron
|
|
|
|
|
2017-04-02 02:19:52 +00:00
|
|
|
type Task struct{}
|
2017-03-10 09:24:06 +00:00
|
|
|
|
2017-03-24 09:55:44 +00:00
|
|
|
// 初始化任务, 从数据库取出所有任务, 添加到定时任务并运行
|
2017-04-02 02:19:52 +00:00
|
|
|
func (task *Task) Initialize() {
|
2017-04-13 09:35:59 +00:00
|
|
|
Cron = cron.New()
|
|
|
|
Cron.Start()
|
2017-04-02 02:38:49 +00:00
|
|
|
taskModel := new(models.Task)
|
|
|
|
taskList, err := taskModel.ActiveList()
|
|
|
|
if err != nil {
|
2017-04-03 07:27:19 +00:00
|
|
|
logger.Error("获取任务列表错误-", err.Error())
|
2017-04-02 02:38:49 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
if len(taskList) == 0 {
|
2017-04-03 07:27:19 +00:00
|
|
|
logger.Debug("任务列表为空")
|
2017-04-02 02:38:49 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
for _, item := range taskList {
|
|
|
|
task.Add(item)
|
|
|
|
}
|
2017-03-10 09:24:06 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// 添加任务
|
2017-04-10 09:37:16 +00:00
|
|
|
func (task *Task) Add(taskModel models.TaskHost) {
|
2017-04-02 02:38:49 +00:00
|
|
|
taskFunc := createHandlerJob(taskModel)
|
|
|
|
if taskFunc == nil {
|
2017-04-13 13:54:13 +00:00
|
|
|
logger.Error("创建任务处理Job失败,不支持的任务协议#", taskModel.Protocol)
|
2017-04-02 02:38:49 +00:00
|
|
|
return
|
|
|
|
}
|
2017-04-13 09:35:59 +00:00
|
|
|
|
|
|
|
cronName := strconv.Itoa(taskModel.Id)
|
|
|
|
Cron.RemoveJob(cronName)
|
2017-04-13 13:54:13 +00:00
|
|
|
err := Cron.AddFunc(taskModel.Spec, taskFunc, cronName)
|
|
|
|
logger.Error("添加任务到调度器失败#", err)
|
2017-03-10 09:24:06 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
type Handler interface {
|
2017-04-10 09:37:16 +00:00
|
|
|
Run(taskModel models.TaskHost) (string, error)
|
2017-03-10 09:24:06 +00:00
|
|
|
}
|
|
|
|
|
2017-03-23 05:31:16 +00:00
|
|
|
// HTTP任务
|
2017-04-02 02:19:52 +00:00
|
|
|
type HTTPHandler struct{}
|
2017-03-10 09:24:06 +00:00
|
|
|
|
2017-04-10 09:37:16 +00:00
|
|
|
func (h *HTTPHandler) Run(taskModel models.TaskHost) (result string, err error) {
|
2017-04-02 02:38:49 +00:00
|
|
|
client := &http.Client{}
|
|
|
|
if taskModel.Timeout > 0 {
|
|
|
|
client.Timeout = time.Duration(taskModel.Timeout) * time.Second
|
|
|
|
}
|
|
|
|
req, err := http.NewRequest("POST", taskModel.Command, nil)
|
|
|
|
if err != nil {
|
2017-04-13 13:54:13 +00:00
|
|
|
logger.Error("任务处理#创建HTTP请求错误-", err.Error())
|
2017-04-02 02:38:49 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
req.Header.Set("Content-type", "application/x-www-form-urlencoded")
|
2017-04-07 01:22:00 +00:00
|
|
|
req.Header.Set("User-Agent", "golang/gocron")
|
2017-04-02 02:38:49 +00:00
|
|
|
|
|
|
|
resp, err := client.Do(req)
|
|
|
|
defer func() {
|
|
|
|
if resp != nil {
|
|
|
|
resp.Body.Close()
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
if err != nil {
|
2017-04-13 13:54:13 +00:00
|
|
|
logger.Error("任务处理HTTP请求错误-", err.Error())
|
2017-04-02 02:38:49 +00:00
|
|
|
return
|
|
|
|
}
|
|
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
|
|
if err != nil {
|
2017-04-13 13:54:13 +00:00
|
|
|
logger.Error("任务处理#读取HTTP请求返回值失败-", err.Error())
|
2017-04-02 02:38:49 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return string(body), err
|
2017-03-10 09:24:06 +00:00
|
|
|
}
|
|
|
|
|
2017-04-01 12:28:30 +00:00
|
|
|
// SSH-command任务
|
2017-04-02 02:19:52 +00:00
|
|
|
type SSHCommandHandler struct{}
|
2017-03-10 09:24:06 +00:00
|
|
|
|
2017-04-10 09:37:16 +00:00
|
|
|
func (h *SSHCommandHandler) Run(taskModel models.TaskHost) (string, error) {
|
|
|
|
sshConfig := ssh.SSHConfig{
|
|
|
|
User: taskModel.Username,
|
|
|
|
Password: taskModel.Password,
|
|
|
|
Host: taskModel.Name,
|
|
|
|
Port: taskModel.Port,
|
|
|
|
ExecTimeout: taskModel.Timeout,
|
2017-04-02 02:38:49 +00:00
|
|
|
}
|
2017-04-10 09:37:16 +00:00
|
|
|
return ssh.Exec(sshConfig, taskModel.Command)
|
2017-03-23 05:31:16 +00:00
|
|
|
}
|
|
|
|
|
2017-04-10 09:37:16 +00:00
|
|
|
|
2017-04-13 09:35:59 +00:00
|
|
|
func createTaskLog(taskModel models.TaskHost) (int64, error) {
|
2017-04-02 02:38:49 +00:00
|
|
|
taskLogModel := new(models.TaskLog)
|
2017-04-10 09:37:16 +00:00
|
|
|
taskLogModel.Name = taskModel.Task.Name
|
2017-04-06 06:40:48 +00:00
|
|
|
taskLogModel.Spec = taskModel.Spec
|
|
|
|
taskLogModel.Protocol = taskModel.Protocol
|
|
|
|
taskLogModel.Command = taskModel.Command
|
|
|
|
taskLogModel.Timeout = taskModel.Timeout
|
2017-04-10 09:37:16 +00:00
|
|
|
taskLogModel.Hostname = taskModel.Name
|
2017-04-02 02:38:49 +00:00
|
|
|
taskLogModel.StartTime = time.Now()
|
|
|
|
taskLogModel.Status = models.Running
|
|
|
|
insertId, err := taskLogModel.Create()
|
2017-03-23 05:31:16 +00:00
|
|
|
|
2017-04-02 02:38:49 +00:00
|
|
|
return insertId, err
|
2017-03-24 05:06:53 +00:00
|
|
|
}
|
|
|
|
|
2017-04-13 09:35:59 +00:00
|
|
|
func updateTaskLog(taskLogId int64, result string, err error) (int64, error) {
|
2017-04-02 02:38:49 +00:00
|
|
|
taskLogModel := new(models.TaskLog)
|
|
|
|
var status models.Status
|
|
|
|
if err != nil {
|
|
|
|
result = err.Error() + " " + result
|
|
|
|
status = models.Failure
|
|
|
|
} else {
|
|
|
|
status = models.Finish
|
|
|
|
}
|
|
|
|
return taskLogModel.Update(taskLogId, models.CommonMap{
|
|
|
|
"status": status,
|
|
|
|
"result": result,
|
|
|
|
})
|
2017-03-23 05:31:16 +00:00
|
|
|
|
|
|
|
}
|
|
|
|
|
2017-04-10 09:37:16 +00:00
|
|
|
func createHandlerJob(taskModel models.TaskHost) cron.FuncJob {
|
2017-04-02 02:38:49 +00:00
|
|
|
var handler Handler = nil
|
|
|
|
switch taskModel.Protocol {
|
2017-04-13 09:35:59 +00:00
|
|
|
case models.TaskHTTP:
|
2017-04-02 02:38:49 +00:00
|
|
|
handler = new(HTTPHandler)
|
2017-04-13 09:35:59 +00:00
|
|
|
case models.TaskSSH:
|
2017-04-02 02:38:49 +00:00
|
|
|
handler = new(SSHCommandHandler)
|
|
|
|
}
|
2017-04-04 09:22:48 +00:00
|
|
|
if handler == nil {
|
|
|
|
return nil
|
|
|
|
}
|
2017-04-02 02:38:49 +00:00
|
|
|
taskFunc := func() {
|
2017-04-06 06:40:48 +00:00
|
|
|
taskLogId, err := createTaskLog(taskModel)
|
2017-04-02 02:38:49 +00:00
|
|
|
if err != nil {
|
2017-04-13 13:54:13 +00:00
|
|
|
logger.Error("任务开始执行#写入任务日志失败-", err)
|
2017-04-02 02:38:49 +00:00
|
|
|
return
|
|
|
|
}
|
2017-04-08 09:15:30 +00:00
|
|
|
// err != nil 执行失败, 失败重试3次
|
|
|
|
retryTime := 4
|
|
|
|
var result string = ""
|
|
|
|
for i := 0; i < retryTime; i++ {
|
|
|
|
result, err = handler.Run(taskModel)
|
|
|
|
if err == nil {
|
|
|
|
break
|
|
|
|
} else {
|
2017-04-13 13:54:13 +00:00
|
|
|
logger.Error("任务执行失败#tasklog.id-" + strconv.FormatInt(taskLogId, 10) + "#尝试次数-" + strconv.Itoa(i + 1) + "#" + err.Error() + " " + result)
|
2017-04-08 09:15:30 +00:00
|
|
|
}
|
|
|
|
}
|
2017-04-13 09:35:59 +00:00
|
|
|
_, err = updateTaskLog(taskLogId, result, err)
|
2017-04-02 02:38:49 +00:00
|
|
|
if err != nil {
|
2017-04-13 13:54:13 +00:00
|
|
|
logger.Error("任务结束#更新任务日志失败-", err)
|
2017-04-02 02:38:49 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return taskFunc
|
2017-04-02 02:19:52 +00:00
|
|
|
}
|