test/logdb
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

ADParserSuccess Add Param successData for log to ext

Browse Source
This commit is contained in:
huangsimin 2018-12-19 10:55:17 +08:00
parent 65ccc9bcde
commit 2b7b150ec5
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
logdb.go
View File

@ -2,6 +2,7 @@ package logdb
import ( import (
"database/sql" "database/sql"
"encoding/json"
"fmt" "fmt"
"io/ioutil" "io/ioutil"
"log" "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() logdb.mutex.Lock()
defer logdb.mutex.Unlock() defer logdb.mutex.Unlock()
_, err := logdb.driver.Exec("update log_spider set status = 200 where uid = ?", uid) ext := make(map[string]string)
if err != nil { ext["success_data"] = successData
log.Println(err)
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)
}
} }
} }