抽取结果保存

src/jy/extract/extract.go

@@ -174,7 +174,7 @@ func (e *ExtractTask) ExtractProcess(j *ju.Job) {
 		bs, _ := json.Marshal(j.Result)
 		log.Println("抽取结果", j.SourceMid, string(bs))
 		//分析抽取结果并保存 todo
-		AnalysisSaveResult(j.Data, j.Result, e.TaskInfo.SaveColl)
+		AnalysisSaveResult(j.Data, j.Result, e.TaskInfo)
 
 	}, func(err interface{}) {
 		log.Println(err)
@@ -564,12 +564,19 @@ func SaveExtLog() {
 }
 
 //分析抽取结果并保存
-func AnalysisSaveResult(doc *map[string]interface{}, result map[string][]*ju.ExtField, totable string) {
+func AnalysisSaveResult(doc *map[string]interface{}, result map[string][]*ju.ExtField, task *TaskInfo) {
 	//待完善
 	for key, val := range result {
 		for _, v := range val { //暂时取第一个保存
 			(*doc)[key] = v.Value
+			if key == "budget" || key == "bidamount" {
+				if qu.Int64All(v.Value) > 0 {
+					break
+				}
+			} else {
+				break
+			}
 		}
 	}
-	db.Mgo.Update(totable, `{"_id":"`+qu.BsonIdToSId((*doc)["_id"])+`"}`, doc, true, false)
+	task.DB.Update(task.SaveColl, `{"_id":"`+qu.BsonIdToSId((*doc)["_id"])+`"}`, doc, true, false)
 }