From 2b7b150ec54734d6dc9e10b07197c163fb5373db Mon Sep 17 00:00:00 2001 From: huangsimin Date: Wed, 19 Dec 2018 10:55:17 +0800 Subject: [PATCH] ADParserSuccess Add Param successData for log to ext --- logdb.go | 21 +++++++++++++++++---- 1 file changed, 17 insertions(+), 4 deletions(-) diff --git a/logdb.go b/logdb.go index 716c18f..01357c6 100644 --- a/logdb.go +++ b/logdb.go @@ -2,6 +2,7 @@ package logdb import ( "database/sql" + "encoding/json" "fmt" "io/ioutil" "log" @@ -171,13 +172,25 @@ func (logdb *LogDB) ADCheckRecover(spider_id int) { } } -func (logdb *LogDB) ADParserSuccess(uid string) { +func (logdb *LogDB) ADParserSuccess(uid string, successData string) { logdb.mutex.Lock() defer logdb.mutex.Unlock() - _, err := logdb.driver.Exec("update log_spider set status = 200 where uid = ?", uid) - if err != nil { - log.Println(err) + ext := make(map[string]string) + ext["success_data"] = successData + + data, err := json.Marshal(&ext) + + if err != nil || successData == "" { + _, err := logdb.driver.Exec("update log_spider set status = 200 where uid = ?", uid) + if err != nil { + log.Println(err) + } + } else { + _, err := logdb.driver.Exec("update log_spider set status = 200, ext = ? where uid = ?", string(data), uid) + if err != nil { + log.Println(err) + } } }