From fbf2638d616d4b949f92123a70de522cfe3f0f09 Mon Sep 17 00:00:00 2001
From: jiangwei1995910 <jw78667602>
Date: Tue, 9 Jul 2019 19:30:25 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B8=85=E7=90=86=E8=AE=BF=E9=97=AE=E8=AE=B0?=
 =?UTF-8?q?=E5=BD=95=E5=92=8Ccookie?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 clean_status.go | 54 ++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 49 insertions(+), 5 deletions(-)

diff --git a/clean_status.go b/clean_status.go
index f1ebf6a..9443a89 100644
--- a/clean_status.go
+++ b/clean_status.go
@@ -1,27 +1,71 @@
 package main
 
 import (
+	"context"
 	"fmt"
+	"getAwayBSG/configs"
 	"getAwayBSG/db"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+	"os"
+	"strings"
+	"time"
 )
 
 func main() {
 
-	fmt.Println("清除抓取状态（不清除状态的话爬虫会从上次停止位置继续抓取）")
-	fmt.Println("请选择需要清除哪个爬虫的的状态数据：（输入数字）")
-	fmt.Println("1.链家")
-	fmt.Println("2.智联")
 	var choice int
-	fmt.Scanln(&choice)
+	if len(os.Args) > 1 && strings.Index(os.Args[1], "lianjia") > -1 {
+		choice = 1
+	} else if len(os.Args) > 1 && strings.Index(os.Args[1], "zhilian") > -1 {
+		choice = 2
+	} else {
+		fmt.Println("清除抓取状态（不清除状态的话爬虫会从上次停止位置继续抓取）")
+		fmt.Println("请选择需要清除哪个爬虫的的状态数据：（输入数字）")
+		fmt.Println("1.链家")
+		fmt.Println("2.智联")
+		fmt.Scanln(&choice)
+
+	}
 
 	if choice == 1 {
 		db.SetLianjiaStatus(0)
+		clean_visit()
 		fmt.Println("Done!")
 	} else if choice == 2 {
 		db.SetZhilianStatus(0, 0)
+		clean_visit()
 		fmt.Println("Done!")
 	} else {
+		clean_visit()
 		fmt.Println("选择错误！")
 	}
 
 }
+
+func clean_visit() {
+	conf := configs.Config()
+
+	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
+	defer cancel()
+
+	client, _ := mongo.NewClient(options.Client().ApplyURI(conf["dburl"].(string) + "/colly"))
+	if err := client.Connect(ctx); err != nil {
+		fmt.Println(err)
+	}
+
+	odb := client.Database("colly")
+	cookies := odb.Collection("colly_cookies")
+	visit := odb.Collection("colly_visited")
+	//清除全部的cookies
+	_, err := cookies.DeleteMany(ctx, bson.M{})
+	if err != nil {
+		fmt.Println(err)
+	}
+	_, err = visit.DeleteMany(ctx, bson.M{})
+	if err != nil {
+		fmt.Println(err)
+	}
+
+}
\ No newline at end of file