优化：核对ai版单位信息

src/check_ai.go

@@ -22,7 +22,8 @@ func ChooseCheckDataAI(tmp map[string]interface{}, update_info *map[string]inter
 	ChooseTheBestPackageField(ai_zhipu, s_toptype, s_subtype, tmp, update_info, &ext_ai_record)
 	//最终金额修正与选取
 	ChooseTheBestAmountField(tmp, update_info)
-
+	//最终单位修正与选取
+	ChooseTheBestUnitField(ai_zhipu, s_toptype, s_subtype, tmp, update_info, &ext_ai_record)
 	//字段记录
 	(*update_info)["ext_ai_record"] = ext_ai_record
 
@@ -258,6 +259,42 @@ func ChooseTheBestAmountField(tmp map[string]interface{}, update_info *map[strin
 	}
 }
 
+// 核对单位值
+func ChooseTheBestUnitField(ai_zhipu map[string]interface{}, s_toptype string, s_subtype string, tmp map[string]interface{}, update_info *map[string]interface{}, ext_ai_record *map[string]interface{}) {
+	buyer := qu.ObjToString((*update_info)["buyer"])
+	s_winner := qu.ObjToString((*update_info)["s_winner"])
+	agency := qu.ObjToString((*update_info)["agency"])
+	if buyer == "" {
+		buyer = qu.ObjToString(tmp["buyer"])
+	}
+	if s_winner == "" {
+		s_winner = qu.ObjToString(tmp["s_winner"])
+	}
+	if agency == "" {
+		agency = qu.ObjToString(tmp["agency"])
+	}
+
+	if buyer == s_winner && buyer != "" {
+		if strings.Contains(buyer, "公司") {
+			(*update_info)["buyer"] = ""
+			(*ext_ai_record)["buyer"] = tmp["buyer"]
+		} else {
+			if s_subtype == "单一" || s_subtype == "中标" || s_subtype == "成交" || s_subtype == "合同" {
+				(*update_info)["s_winner"] = ""
+				(*update_info)["winner"] = ""
+				(*ext_ai_record)["s_winner"] = tmp["s_winner"]
+				(*ext_ai_record)["winner"] = tmp["winner"]
+			}
+		}
+	}
+	if agency != "" {
+		if agency == buyer || agency == s_winner {
+			(*update_info)["agency"] = ""
+			(*ext_ai_record)["agency"] = tmp["buyer"]
+		}
+	}
+}
+
 // 筛选金额
 func filterAmount(f1 float64, f2 float64) float64 {
 	//选取一个合适的金额 ...

src/config.json

@@ -5,7 +5,7 @@
   "mongodb": {
     "addrName": "127.0.0.1:27017",
     "dbName": "zhengkun",
-    "collName": "zktest_info_0930_new",
+    "collName": "extract_test",
     "username": "",
     "password": "",
     "pool": 10

src/main.go

@@ -34,6 +34,7 @@ func main() {
 	go checkMailJob()
 	go lastUdpJob()
 
+	//test()
 	lock := make(chan bool)
 	<-lock
 }