From 12b03cba8a54b29c9bb84ec878f36b2cdd65c806 Mon Sep 17 00:00:00 2001
From: marvzhang <tikazyq@163.com>
Date: Sun, 9 Feb 2020 09:02:13 +0800
Subject: [PATCH] fixed https://github.com/crawlab-team/crawlab/issues/524

---
 backend/model/spider.go  |  4 ++++
 backend/routes/spider.go | 14 ++++++++++++--
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/backend/model/spider.go b/backend/model/spider.go
index 2baeb6ed..c1544259 100644
--- a/backend/model/spider.go
+++ b/backend/model/spider.go
@@ -78,6 +78,10 @@ func (spider *Spider) Add() error {
 	spider.CreateTs = time.Now()
 	spider.UpdateTs = time.Now()
 
+	if !spider.ProjectId.Valid() {
+		spider.ProjectId = bson.ObjectIdHex(constants.ObjectIdNull)
+	}
+
 	if err := c.Insert(&spider); err != nil {
 		return err
 	}
diff --git a/backend/routes/spider.go b/backend/routes/spider.go
index 2b6dfd63..1abb6463 100644
--- a/backend/routes/spider.go
+++ b/backend/routes/spider.go
@@ -295,7 +295,12 @@ func UploadSpider(c *gin.Context) {
 		if cmd != "" {
 			spider.Cmd = cmd
 		}
-		_ = spider.Add()
+		if err := spider.Add(); err != nil {
+			log.Error("add spider error: " + err.Error())
+			debug.PrintStack()
+			HandleError(http.StatusInternalServerError, c, err)
+			return
+		}
 	} else {
 		if name != "" {
 			spider.Name = name
@@ -311,7 +316,12 @@ func UploadSpider(c *gin.Context) {
 		}
 		// 更新file_id
 		spider.FileId = fid
-		_ = spider.Save()
+		if err := spider.Add(); err != nil {
+			log.Error("add spider error: " + err.Error())
+			debug.PrintStack()
+			HandleError(http.StatusInternalServerError, c, err)
+			return
+		}
 	}
 
 	// 发起同步