fix 删除spider没有删除task的问题

This commit is contained in:
陈景阳
2019-09-02 20:49:00 +08:00
parent 41c3b98333
commit 6675d075f2
2 changed files with 22 additions and 1 deletions

View File

@@ -190,6 +190,21 @@ func RemoveTask(id string) error {
return nil
}
func RemoveTaskBySpiderId(id string) error {
tasks, err := GetTaskList(bson.M{"spider_id": id}, 0, constants.Infinite, "-create_ts")
if err != nil {
log.Error("get tasks error:" + err.Error())
}
for _, task := range tasks {
if err := RemoveTask(task.Id); err != nil {
log.Error("remove task error:" + err.Error())
continue
}
}
return nil
}
func GetTaskCount(query interface{}) (int, error) {
s, c := database.GetCol("tasks")
defer s.Close()
@@ -207,7 +222,7 @@ func GetDailyTaskStats(query bson.M) ([]TaskDailyItem, error) {
defer s.Close()
// 起始日期
startDate := time.Now().Add(- 30 * 24 * time.Hour)
startDate := time.Now().Add(-30 * 24 * time.Hour)
endDate := time.Now()
// query

View File

@@ -229,6 +229,12 @@ func DeleteSpider(c *gin.Context) {
return
}
// 删除爬虫对应的task任务
if err := model.RemoveTaskBySpiderId(spider.Id.Hex()); err != nil {
HandleError(http.StatusInternalServerError, c, err)
return
}
c.JSON(http.StatusOK, Response{
Status: "ok",
Message: "success",