封装cron

Author: yumaojun03

go.mod

@@ -26,6 +26,7 @@ require (
 	github.com/prometheus/client_golang v1.19.0
 	github.com/redis/go-redis/extra/redisotel/v9 v9.0.5
 	github.com/redis/go-redis/v9 v9.5.1
+	github.com/robfig/cron/v3 v3.0.1
 	github.com/rs/xid v1.5.0
 	github.com/rs/zerolog v1.32.0
 	github.com/segmentio/kafka-go v0.4.47

go.sum

@@ -205,6 +205,8 @@ github.com/redis/go-redis/v9 v9.5.1 h1:H1X4D3yHPaYrkL5X06Wh6xNVM/pX0Ft4RV0vMGvLB
 github.com/redis/go-redis/v9 v9.5.1/go.mod h1:hdY0cQFCN4fnSYT6TkisLufl/4W5UIXyv0b/CLO2V2M=
 github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
 github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
+github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
+github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
 github.com/rogpeppe/go-internal v1.11.0 h1:cWPaGQEPrBb5/AsnsZesgZZ9yb1OQ+GOISoDNXVBh4M=
 github.com/rogpeppe/go-internal v1.11.0/go.mod h1:ddIwULY96R17DhadqLgMfk9H9tvdUzkipdSkR5nkCZA=
 github.com/rs/xid v1.5.0 h1:mKX4bl4iPYJtEIxp6CYiUuLQ/8DYMoz0PUdtGgMFRVc=

ioc/config/mcron/cron.go

@@ -0,0 +1,41 @@
+package mcron
+
+import (
+	"github.com/infraboard/mcube/v2/ioc"
+	"github.com/infraboard/mcube/v2/ioc/config/log"
+	"github.com/robfig/cron/v3"
+	"github.com/rs/zerolog"
+)
+
+func init() {
+	ioc.Default().Registry(defaultConfig)
+}
+
+var defaultConfig = &config{
+	cron: cron.New(cron.WithChain(
+		cron.Recover(&LogWrapper{}),
+		cron.SkipIfStillRunning(&LogWrapper{}),
+	),
+		cron.WithLogger(&LogWrapper{}),
+	),
+}
+
+type config struct {
+	cron *cron.Cron
+	ioc.ObjectImpl
+	log *zerolog.Logger
+}
+
+func (c *config) Name() string {
+	return APP_NAME
+}
+
+func (c *config) Priority() int {
+	return PRIORITY
+}
+
+func (c *config) Init() error {
+	c.log = log.Sub(c.Name())
+	c.cron.Start()
+	return nil
+}

ioc/config/mcron/cron_test.go

@@ -0,0 +1 @@
+package mcron_test

ioc/config/mcron/interface.go

@@ -0,0 +1,21 @@
+package mcron
+
+import (
+	"github.com/infraboard/mcube/v2/ioc"
+	"github.com/robfig/cron/v3"
+)
+
+const (
+	APP_NAME = "cron"
+	PRIORITY = -199
+)
+
+func Get() *cron.Cron {
+	return ioc.Default().Get(APP_NAME).(*config).cron
+}
+
+func RunAndAddFunc(spec string, cmd func()) (cron.EntryID, error) {
+	cmd()
+
+	return Get().AddFunc(spec, cmd)
+}

ioc/config/mcron/log.go

@@ -0,0 +1,16 @@
+package mcron
+
+import "github.com/infraboard/mcube/v2/ioc/config/log"
+
+type LogWrapper struct {
+}
+
+// Info logs routine messages about cron's operation.
+func (l *LogWrapper) Info(msg string, keysAndValues ...interface{}) {
+	log.Sub(APP_NAME).Info().Msg(msg)
+}
+
+// Error logs an error condition.
+func (l *LogWrapper) Error(err error, msg string, keysAndValues ...interface{}) {
+	log.Sub(APP_NAME).Error().Msgf("%s, %s", err.Error(), msg)
+}