63 lines
1.1 KiB
Go
63 lines
1.1 KiB
Go
package core
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/robfig/cron"
|
|
)
|
|
|
|
var CronInfo CronServiceInfo
|
|
|
|
type CronServiceInfo struct {
|
|
Cron []string
|
|
Func []func()
|
|
}
|
|
|
|
func (c *CronServiceInfo) Register(cron string, f func()) {
|
|
c.Cron = append(c.Cron, cron)
|
|
c.Func = append(c.Func, f)
|
|
}
|
|
|
|
func CronRun() {
|
|
c := cron.New()
|
|
for k, v := range CronInfo.Cron {
|
|
err := c.AddFunc(v, wrap(CronInfo.Func[k]))
|
|
if err != nil {
|
|
fmt.Println("定时任务添加失败")
|
|
}
|
|
}
|
|
c.Start()
|
|
}
|
|
|
|
func wrap(f func()) func() {
|
|
return func() {
|
|
//defer CronRecovery(runtime.FuncForPC(reflect.ValueOf(f).Pointer()).Name())
|
|
f()
|
|
}
|
|
}
|
|
|
|
func CronRegister(rule string, f func()) {
|
|
CronInfo.Register(rule, f)
|
|
}
|
|
|
|
//func CronRecovery(name string){
|
|
// if err := recover(); err != nil {
|
|
// var code int
|
|
// var msg interface{}
|
|
// if h, ok := err.(*Exception); ok {
|
|
// code = h.Code
|
|
// msg = h.Msg
|
|
// } else if _, ok = err.(error); ok {
|
|
// msg = "未知错误"
|
|
// code = UNKNOW_ERROR
|
|
// } else {
|
|
// msg = "服务器错误"
|
|
// code = SERVER_ERROR
|
|
// }
|
|
// Log.WithFields(map[string]interface{}{
|
|
// "code":code,
|
|
// "model":"task",
|
|
// "func":name,
|
|
// }).Error(msg)
|
|
// }
|
|
//}
|