Initial commit

core/server/cron_server/index.go

@@ -0,0 +1,34 @@
+package cron_server
+
+import (
+	"fmt"
+	"sync"
+	"time"
+)
+
+var (
+	once sync.Once
+	cc   *cron.Cron
+)
+
+// CronInstance cron single instance
+func CronInstance() *cron.Cron {
+	once.Do(func() {
+		cc = cron.New(cron.WithSeconds())
+	})
+	return cc
+}
+
+// DoOnce run job once time,this job will run after 2 second
+func DoOnce(job cron.Job, t ...time.Duration) error {
+	once := time.Now().Add(2 * time.Second)
+	if len(t) == 1 {
+		once = time.Now().Add(t[0] * time.Second)
+	}
+	onceSpec := fmt.Sprintf("%d %d %d %d %d %d", once.Second(), once.Minute(), once.Hour(), once.Day(), once.Month(), once.Weekday())
+	_, err := CronInstance().AddJob(onceSpec, job)
+	if err != nil {
+		return err
+	}
+	return nil
+}

core/server/cron_server/index_test.go

@@ -0,0 +1,16 @@
+package cron_server
+
+import "testing"
+
+func TestCronInstance(t *testing.T) {
+	t.Run("Test cron instance init", func(t *testing.T) {
+		instance := CronInstance()
+		if instance == nil {
+			t.Error("Cron Instance init fail.")
+		}
+		instance1 := CronInstance()
+		if instance != instance1 {
+			t.Error("Cron Instance is change.")
+		}
+	})
+}