init

core/crontab.go

@@ -0,0 +1,62 @@
+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)
+//	}
+//}