From cd7f6e66b4b038ff420d148ef7d39c407f1b15aa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=99=88=E6=99=AF=E9=98=B3?= <1656488874@qq.com>
Date: Fri, 11 Oct 2019 23:53:07 +0800
Subject: [PATCH] =?UTF-8?q?fix=20=E5=AE=9A=E6=97=B6=E4=BB=BB=E5=8A=A1?=
 =?UTF-8?q?=E7=AE=A1=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/model/schedule.go  | 7 +++++++
 backend/routes/schedule.go | 7 +++----
 2 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/backend/model/schedule.go b/backend/model/schedule.go
index 951cb043..bcd051e3 100644
--- a/backend/model/schedule.go
+++ b/backend/model/schedule.go
@@ -104,6 +104,13 @@ func UpdateSchedule(id bson.ObjectId, item Schedule) error {
 		return err
 	}
 
+	node, err := GetNode(item.NodeId)
+	if err != nil {
+		log.Errorf("get node error: %s", err.Error())
+		debug.PrintStack()
+		return nil
+	}
+	item.NodeKey = node.Key
 	if err := item.Save(); err != nil {
 		return err
 	}
diff --git a/backend/routes/schedule.go b/backend/routes/schedule.go
index 4ca245b3..24df0c0f 100644
--- a/backend/routes/schedule.go
+++ b/backend/routes/schedule.go
@@ -1,7 +1,6 @@
 package routes
 
 import (
-	"crawlab/constants"
 	"crawlab/model"
 	"crawlab/services"
 	"github.com/gin-gonic/gin"
@@ -49,9 +48,9 @@ func PostSchedule(c *gin.Context) {
 	newItem.Id = bson.ObjectIdHex(id)
 
 	// 如果node_id为空，则置为空ObjectId
-	if newItem.NodeId == "" {
-		newItem.NodeId = bson.ObjectIdHex(constants.ObjectIdNull)
-	}
+	//if newItem.NodeId == "" {
+	//	newItem.NodeId = bson.ObjectIdHex(constants.ObjectIdNull)
+	//}
 
 	// 更新数据库
 	if err := model.UpdateSchedule(bson.ObjectIdHex(id), newItem); err != nil {