任務管理介面 (WEB)
支援在 WEB 介面 中對任務進行管理，例如：新增任務、編輯任務、啟用/禁用任務、手動執行任務 等。
任務的屬性包括：
- 任務名稱
- 執行方式
SHELLHTTP
- 表示式（*/5 * * * *）
- 命令
- 超時時間（秒）
- 重試次數
- 重試間隔（秒）
- 執行結束是否通知
- 不通知
- 失敗通知
- 結束通知
- 結果關鍵字匹配通知
- 狀態
- 備註
當執行方式為 HTTP 時，支援選擇請求方式 GET 或 POST；
當設定執行結束通知時，支援選擇通知方式 郵件 或 Webhook；
當設定郵件通知時，支援輸入郵箱地址多個用,分割；
當設定結果關鍵字匹配通知時，支援輸入關鍵字多個用,分割；
任務增加完成後，會把任務資料持久化到 MySQL 中。
任務排程器
參考了兩個開源元件：
最終選擇使用 jakecoffman/cron ，後者是在前者的基礎上做了一定的補充，例如 AddFunc() 增加了 name 引數，同時還增加了 RemoveJob(name string) 支援刪除特定的任務。
// AddFunc adds a func to the Cron to be run on the given schedule.
func (c *Cron) AddFunc(spec string, cmd func(), name string) {
c.AddJob(spec, FuncJob(cmd), name)
}
...
// RemoveJob removes a Job from the Cron based on name.
func (c *Cron) RemoveJob(name string) {
if !c.running {
i := c.entries.pos(name)
if i == -1 {
return
}
c.entries = c.entries[:i+copy(c.entries[i:], c.entries[i+1:])]
return
}
c.remove <- name
}
對其簡單封裝下就可以使用了，下面是封裝的方法，方法的具體實現與使用從 go-gin-api 中獲取。
type Server interface {
i()
// Start 啟動 cron 服務
Start()
// Stop 停止 cron 服務
Stop()
// AddTask 增加定時任務
AddTask(task *cron_task_repo.CronTask)
// RemoveTask 刪除定時任務
RemoveTask(taskId int)
// AddJob 增加定時任務執行的工作內容
AddJob(task *cron_task_repo.CronTask) cron.FuncJob
}
當呼叫 Start() 啟動服務時，會把 MySQL 中的任務列表載入到排程器中。
通過以上方法，當從 WEB 介面 操作 新增、編輯、啟用/禁用、手動執行任務時，可以動態的對排程器中的任務進行管理。
任務執行器
任務執行器指的是任務真實執行所在的機器。
我的思路是使用 Kafka 的釋出與訂閱功能，當排程器發現需要執行的任務時，將任務資訊寫到 Kafka 的 Topic 中，任務執行器訂閱相關的 Topic 獲取任務資訊然後執行任務。
如果任務的執行方式為 HTTP，那麼任務執行器可以為一組叢集，專門處理呼叫 HTTP 任務，這裡可以為一個消費組（Consumer Group），也可適具體場景而定。
如果任務的執行方式為 SHELL，那麼任務執行器必須在指令碼所在的宿主機上，這裡可以為一個具體任務的消費者。
如果任務量過多，可以考慮根據業務場景多設定幾個 Topic。
在專案中為了便於演示，不寫入到 Kafka 中，僅記錄了日誌。
func (s *server) AddJob(task *cron_task_repo.CronTask) cron.FuncJob {
return func() {
s.taskCount.Add()
defer s.taskCount.Done()
msg := fmt.Sprintf("開始執行任務：(%d)%s [%s]", task.Id, task.Name, task.Spec)
s.logger.Info(msg)
}
}
日誌目錄：/logs/go-gin-api-cron.log。
小結
本文純屬拋磚引玉，有問題，歡迎批評指正。
go-gin-api 專案安裝簡單，開箱即用，建立一個後臺任務試試吧。