go 動態定時任務 API 接口(支持其他語言調用)

go 動態定時任務 API 接口

使用

• 連接數據庫表，啓動服務 初始化所有定時任務，然後通過 API接口 動態修改 定時任務。
• 數據庫表

CREATE TABLE `go_cron` (
  `id` int(10) unsigned NOT NULL AUTO_INCREMENT,
  `time` varchar(100) DEFAULT '' COMMENT '定時時間',
  `ip` varchar(100) DEFAULT '' COMMENT '內網IP',
  `shell` varchar(50) DEFAULT '' COMMENT '命令',
  `env` varchar(50) DEFAULT '' COMMENT '環境',
  `run` int(10)  DEFAULT '0' COMMENT '是否運行',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=17 DEFAULT CHARSET=utf8;

代碼

package main

import (
    "fmt"
    "github.com/Unknwon/com"
    "github.com/gin-gonic/gin"
    "github.com/gogf/gf/g/os/gcron"
    "github.com/gogf/gf/g/os/glog"
    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/mysql"
    "strconv"
)

var db *gorm.DB

type Cron struct {
    ID        int    `gorm:"primary_key" json:"id"`
    Time      string `json:"time"`
    Ip        string `json:"ip"`
    Shell string `json:"shell"`
    Env       string `json:"env"`
    Run       int    `json:"run"`
}

func GetCron(id int) (user Cron) {
    db.Where("id = ?", id).First(&user)
    return
}

func GetCronAll() ([] *Cron) {
    var cron []*Cron
    db.Where("run = ?", 0).Find(&cron)
    return cron
}

func InitDb() {
    var err error
    db, err = gorm.Open("mysql", fmt.Sprintf("root:123456@tcp(192.168.100.50)/go?charset=utf8&parseTime=True&loc=Local", ))

    if err != nil {
        glog.Println(err)
    }

    gorm.DefaultTableNameHandler = func(db *gorm.DB, defaultTableName string) string {
        return "go_" + defaultTableName
    }

    db.SingularTable(true)
    db.DB().SetMaxIdleConns(10)
    db.DB().SetMaxOpenConns(100)
}

func main() {
    InitDb()

    r := gin.Default()

    gc := gcron.New()
    data := GetCronAll()

    for _, v := range data {
        _, err := gc.Add(v.Time, func() { glog.Println(v.shell) }, strconv.Itoa(v.ID))
        if err != nil {
            fmt.Println("初始化定時任務錯誤 ",err)
        }
    }

    r.GET("/crons/:id", func(c *gin.Context) {
        id := com.StrTo(c.Param("id")).MustInt()
        run := com.StrTo(c.Query("run")).MustInt()
        if run == 0 {
            gc.Start(strconv.Itoa(id))
            glog.Println("啓動", id)
            c.JSON(200, gin.H{
                "data": 0,
            })
        } else if run == 1 {
            glog.Println("關閉", id)
            gc.Stop(strconv.Itoa(id))
            c.JSON(200, gin.H{
                "data": 1,
            })
        } else {
            glog.Println("參數錯誤", id)
            c.JSON(400, gin.H{
                "data": "參數不對, 參數爲 run=0/1",
            })
        }
    })

    r.POST("/crons/:id", func(c *gin.Context) {
        id := com.StrTo(c.Param("id")).MustInt()
        data := GetCron(id)
        _, err := gc.Add(data.Time, func() { glog.Println(data.shell) }, strconv.Itoa(data.ID))
        if err != nil {
            glog.Println("新增任務錯誤 ",err)
            c.JSON(400, gin.H{
                "data":err,
            })
        }else {
            glog.Println("新增任務 成功 ")
            c.JSON(200, gin.H{
                "data":0,
            })
        }
    })

    r.PUT("/crons/:id", func(c *gin.Context) {
        id := com.StrTo(c.Param("id")).MustInt()
        data := GetCron(id)
        gc.Stop(strconv.Itoa(id))
        gc.Remove(strconv.Itoa(id))
        _, err := gc.Add(data.Time, func() { glog.Println(data.shell) }, strconv.Itoa(data.ID))
        if err != nil {
            glog.Println("更新 任務錯誤 ",err)
            c.JSON(400, gin.H{
                "data":err,
            })
        }else {
            glog.Println("更新 任務 成功 ")
            c.JSON(200, gin.H{
                "data":0,
            })
        }
    })

    r.DELETE("/crons/:id", func (c *gin.Context) {
        id := com.StrTo(c.Param("id")).MustInt()

        fmt.Println(gc.Search(strconv.Itoa(id)))
        glog.Println("刪除", id)
        gc.Stop(strconv.Itoa(id))
        fmt.Println(gc.Search(strconv.Itoa(id)))
        gc.Remove(strconv.Itoa(id))
        fmt.Println(gc.Search(strconv.Itoa(id)))

        err := gc.Search(strconv.Itoa(id))
        if err != nil {
            c.JSON(400, gin.H{
                "data": "刪除成功",
            })
        }else {
            c.JSON(200, gin.H{
                "data": 0,
            })
        }

    })
    _ = r.Run("127.0.0.1:8080")
}