From 66bef2e6f25ef4b9d0b414e928bd563c34e34e1a 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, 27 Sep 2019 12:31:47 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=B0=E5=A2=9E=E5=87=A0=E4=B8=AA=E6=96=B9?=
 =?UTF-8?q?=E6=B3=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/model/file.go   |  9 +++++++++
 backend/model/spider.go | 11 ++++++++++-
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/backend/model/file.go b/backend/model/file.go
index ca2c8ead..7aa88e3d 100644
--- a/backend/model/file.go
+++ b/backend/model/file.go
@@ -26,6 +26,15 @@ type File struct {
 	Size  int64  `json:"size"`
 }
 
+func (f *GridFs) Remove() {
+	s, gf := database.GetGridFs("files")
+	defer s.Close()
+	if err := gf.RemoveId(f.Id); err != nil {
+		log.Errorf("remove file id error: %s, id: %s", err.Error(), f.Id.Hex())
+		debug.PrintStack()
+	}
+}
+
 func GetAllGridFs() []*GridFs {
 	s, gf := database.GetGridFs("files")
 	defer s.Close()
diff --git a/backend/model/spider.go b/backend/model/spider.go
index 13ec12e4..e63c5f57 100644
--- a/backend/model/spider.go
+++ b/backend/model/spider.go
@@ -124,8 +124,17 @@ func GetSpiderList(filter interface{}, skip int, limit int) ([]Spider, int, erro
 	return spiders, count, nil
 }
 
-func GetSpiderByFileId(fileId bson.ObjectId) {
+func GetSpiderByFileId(fileId bson.ObjectId) *Spider {
+	s, c := database.GetCol("spiders")
+	defer s.Close()
 
+	var result *Spider
+	if err := c.Find(bson.M{"file_id": fileId}).One(&result); err != nil {
+		log.Errorf("get spider error: %s, file_id: %s", err.Error(), fileId.Hex())
+		debug.PrintStack()
+		return nil
+	}
+	return result
 }
 
 func GetSpiderByName(name string) *Spider {