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) + } } }