fix 定时任务参数错误的问题

backend/model/node.go

@@ -4,6 +4,7 @@ import (
 	"crawlab/constants"
 	"crawlab/database"
 	"crawlab/services/register"
+	"errors"
 	"github.com/apex/log"
 	"github.com/globalsign/mgo"
 	"github.com/globalsign/mgo/bson"
@@ -158,16 +159,19 @@ func GetNodeList(filter interface{}) ([]Node, error) {
 
 func GetNode(id bson.ObjectId) (Node, error) {
 	var node Node
+
 	if id.Hex() == "" {
-		return node, nil
+		log.Infof("id is empty")
+		debug.PrintStack()
+		return node, errors.New("id is empty")
 	}
+
 	s, c := database.GetCol("nodes")
 	defer s.Close()
+
 	if err := c.FindId(id).One(&node); err != nil {
-		if err != mgo.ErrNotFound {
-			log.Errorf(err.Error())
-			debug.PrintStack()
-		}
+		log.Errorf(err.Error())
+		debug.PrintStack()
 		return node, err
 	}
 	return node, nil

backend/model/schedule.go

@@ -45,6 +45,27 @@ func (sch *Schedule) Delete() error {
 	return c.RemoveId(sch.Id)
 }
 
+func (sch *Schedule) SyncNodeIdAndSpiderId(node Node, spider Spider) {
+	sch.syncNodeId(node)
+	sch.syncSpiderId(spider)
+}
+
+func (sch *Schedule) syncNodeId(node Node) {
+	if node.Id.Hex() == sch.NodeId.Hex() {
+		return
+	}
+	sch.NodeId = node.Id
+	_ = sch.Save()
+}
+
+func (sch *Schedule) syncSpiderId(spider Spider) {
+	if spider.Id.Hex() == sch.SpiderId.Hex() {
+		return
+	}
+	sch.SpiderId = spider.Id
+	_ = sch.Save()
+}
+
 func GetScheduleList(filter interface{}) ([]Schedule, error) {
 	s, c := database.GetCol("schedules")
 	defer s.Close()
@@ -103,13 +124,11 @@ func UpdateSchedule(id bson.ObjectId, item Schedule) error {
 	if err := c.FindId(id).One(&result); err != nil {
 		return err
 	}
-
 	node, err := GetNode(item.NodeId)
 	if err != nil {
-		log.Errorf("get node error: %s", err.Error())
-		debug.PrintStack()
-		return nil
+		return err
 	}
+
 	item.NodeKey = node.Key
 	if err := item.Save(); err != nil {
 		return err
@@ -123,9 +142,7 @@ func AddSchedule(item Schedule) error {
 
 	node, err := GetNode(item.NodeId)
 	if err != nil {
-		log.Errorf("get node error: %s", err.Error())
-		debug.PrintStack()
-		return nil
+		return err
 	}
 
 	item.Id = bson.NewObjectId()

backend/routes/schedule.go

@@ -45,13 +45,14 @@ func PostSchedule(c *gin.Context) {
 		HandleError(http.StatusBadRequest, c, err)
 		return
 	}
+
+	// 验证cron表达式
+	if err := services.ParserCron(newItem.Cron); err != nil {
+		HandleError(http.StatusOK, c, err)
+		return
+	}
+
 	newItem.Id = bson.ObjectIdHex(id)
-
-	// 如果node_id为空，则置为空ObjectId
-	//if newItem.NodeId == "" {
-	//	newItem.NodeId = bson.ObjectIdHex(constants.ObjectIdNull)
-	//}
-
 	// 更新数据库
 	if err := model.UpdateSchedule(bson.ObjectIdHex(id), newItem); err != nil {
 		HandleError(http.StatusInternalServerError, c, err)
@@ -79,10 +80,11 @@ func PutSchedule(c *gin.Context) {
 		return
 	}
 
-	// 如果node_id为空，则置为空ObjectId
-	//if item.NodeId == "" {
-	//	item.NodeId = bson.ObjectIdHex(constants.ObjectIdNull)
-	//}
+	// 验证cron表达式
+	if err := services.ParserCron(item.Cron); err != nil {
+		HandleError(http.StatusOK, c, err)
+		return
+	}
 
 	// 更新数据库
 	if err := model.AddSchedule(item); err != nil {

backend/services/schedule.go

@@ -5,7 +5,7 @@ import (
 	"crawlab/lib/cron"
 	"crawlab/model"
 	"github.com/apex/log"
-	uuid "github.com/satori/go.uuid"
+	"github.com/satori/go.uuid"
 	"runtime/debug"
 )
 
@@ -31,6 +31,9 @@ func AddTask(s model.Schedule) func() {
 			return
 		}
 
+		// 同步ID到定时任务
+		s.SyncNodeIdAndSpiderId(node, *spider)
+
 		// 生成任务ID
 		id := uuid.NewV4()
 
@@ -119,6 +122,18 @@ func (s *Scheduler) RemoveAll() {
 	}
 }
 
+// 验证cron表达式是否正确
+func ParserCron(spec string) error {
+	parser := cron.NewParser(
+		cron.Second | cron.Minute | cron.Hour | cron.Dom | cron.Month | cron.Dow | cron.Descriptor,
+	)
+
+	if _, err := parser.Parse(spec); err != nil {
+		return err
+	}
+	return nil
+}
+
 func (s *Scheduler) Update() error {
 	// 删除所有定时任务
 	s.RemoveAll()