feat: task cron config

plugin_test.go

@@ -131,7 +131,8 @@ func TestPostProcessInitializerFailed(t *testing.T) {
 func TestPostProcessTask(t *testing.T) {
 	initializers = make(map[string]Initializer)
 	tasks = make([]Task, 0)
-	RegisterTask("test", "0 0/1 * * * *", func() {})
+	viper.Set("task.cron.test", "0 0/1 * * * *")
+	RegisterTask("test", func() {})
 	if PostProcess() != nil {
 		t.Fail()
 	}
@@ -140,7 +141,7 @@ func TestPostProcessTask(t *testing.T) {
 func TestPostProcessTaskFailed(t *testing.T) {
 	initializers = make(map[string]Initializer)
 	tasks = make([]Task, 0)
-	RegisterTask("test", "", func() {})
+	RegisterTask("testFailed", func() {})
 	if PostProcess() == nil {
 		t.Fail()
 	}

task.go

@@ -4,18 +4,18 @@ import (
 	"fmt"
 
 	"github.com/robfig/cron/v3"
+	"github.com/spf13/viper"
 )
 
 type Task struct {
 	Name    string
-	Cron    string
 	Handler func()
 }
 
 var tasks = make([]Task, 0)
 
-func RegisterTask(name, cron string, handler func()) {
-	tasks = append(tasks, Task{Name: name, Cron: cron, Handler: handler})
+func RegisterTask(name string, handler func()) {
+	tasks = append(tasks, Task{Name: name, Handler: handler})
 }
 
 func runTask() error {
@@ -25,8 +25,9 @@ func runTask() error {
 	}
 	cron := cron.New(cron.WithSeconds())
 	for _, task := range tasks {
-		Log(fmt.Sprintf("Add task [%s]: [%s]", task.Cron, task.Name))
-		_, err := cron.AddFunc(task.Cron, task.Handler)
+		configCron := viper.GetString("task.cron." + task.Name)
+		Log(fmt.Sprintf("Add task [%s]: [%s]", configCron, task.Name))
+		_, err := cron.AddFunc(configCron, task.Handler)
 		if err != nil {
 			return err
 		}