mirror of https://github.com/ouqiang/gocron
137 lines
2.7 KiB
Go
137 lines
2.7 KiB
Go
package service
|
|
|
|
import (
|
|
"scheduler/models"
|
|
"scheduler/utils"
|
|
"net/http"
|
|
"io/ioutil"
|
|
"strconv"
|
|
"time"
|
|
"sync"
|
|
)
|
|
|
|
type Host struct {
|
|
sync.RWMutex
|
|
hosts []models.Host
|
|
}
|
|
|
|
var host Host = &Host{
|
|
sync.RWMutex{},
|
|
hosts: initHosts(),
|
|
}
|
|
|
|
func GetHosts() []models.Host {
|
|
host.RLock()
|
|
defer host.RUnlock()
|
|
|
|
return host.hosts
|
|
}
|
|
|
|
func SetHosts(h []models.Host) {
|
|
host.Lock()
|
|
defer host.Unlock()
|
|
|
|
host.hosts = h
|
|
}
|
|
|
|
func initHosts() []models.Host {
|
|
// 获取所有主机
|
|
hostModel := new(models.Host)
|
|
list, err := hostModel.List()
|
|
if err != nil {
|
|
utils.RecordLog("获取主机列表失败-", err.Error())
|
|
return nil
|
|
}
|
|
if len(list) == 0 {
|
|
utils.RecordLog("主机列表为空")
|
|
return nil
|
|
}
|
|
|
|
return list
|
|
}
|
|
|
|
type Task struct {}
|
|
|
|
// 初始化任务，从数据库取出所有任务添加到定时任务
|
|
func(task *Task) Initialize() {
|
|
taskModel := new(models.Task)
|
|
taskList, err := taskModel.List()
|
|
if err != nil {
|
|
utils.RecordLog("获取任务列表错误-", err.Error())
|
|
}
|
|
if len(taskList) == 0 {
|
|
utils.RecordLog("任务列表为空")
|
|
}
|
|
for _, item := range(taskList) {
|
|
task.Add(item)
|
|
}
|
|
}
|
|
|
|
// 添加任务
|
|
func(task *Task) Add(taskModel models.Task) {
|
|
var taskFunc func() = nil;
|
|
switch taskModel.Protocol {
|
|
case models.HTTP:
|
|
taskFunc = func() {
|
|
var handler Handler = new(HTTPHandler)
|
|
handler.Run(taskModel)
|
|
}
|
|
case models.SSH:
|
|
taskFunc = func() {
|
|
var handler Handler = new(SSHHandler)
|
|
handler.Run(taskModel)
|
|
}
|
|
default:
|
|
utils.RecordLog("任务协议不存在-协议编号: ", taskModel.Protocol)
|
|
}
|
|
if (taskFunc != nil) {
|
|
utils.DefaultCronTask.Add(strconv.Itoa(taskModel.Id), taskModel.Spec, taskFunc)
|
|
}
|
|
}
|
|
|
|
type Handler interface {
|
|
Run(taskModel models.Task)
|
|
}
|
|
|
|
type HTTPHandler struct {}
|
|
|
|
// 执行HTTP任务
|
|
func(h *HTTPHandler) Run(taskModel models.Task) {
|
|
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 {
|
|
utils.RecordLog("创建HTTP请求错误-", err.Error())
|
|
return
|
|
}
|
|
req.Header.Set("Content-type", "application/x-www-form-urlencoded")
|
|
req.Header.Set("User-Agent", "golang-cron/scheduler")
|
|
|
|
resp, err := client.Do(req)
|
|
defer resp.Body.Close()
|
|
if err != nil {
|
|
utils.RecordLog("HTTP请求错误-", err.Error())
|
|
return
|
|
}
|
|
body, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
utils.RecordLog("读取HTTP请求返回值失败-", err.Error())
|
|
}
|
|
|
|
_, err = taskModel.Update(
|
|
taskModel.Id,
|
|
models.CommonMap{
|
|
"status": 0,
|
|
"result" : string(body),
|
|
});
|
|
if err != nil {
|
|
utils.RecordLog("更新任务日志失败-", err.Error())
|
|
}
|
|
}
|
|
|
|
type SSHHandler struct {}
|
|
|
|
// 执行SSH任务
|
|
func(ssh *SSHHandler) Run(taskModel models.Task) {} |