jyps/project_chuan/project.go

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"github.com/olivere/elastic/v7"
	"go.uber.org/zap"
	util "jygit.jydev.jianyu360.cn/data_processing/common_utils"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/log"
	"jygit.jydev.jianyu360.cn/data_processing/common_utils/mongodb"
	"regexp"
	"sort"
	"strings"
	"sync"
	"time"
)

func dealXlsxTest() {
	// 1. 初始化 ES 客户端
	client, err := elastic.NewClient(
		elastic.SetURL(GF.Es.URL),
		elastic.SetBasicAuth(GF.Es.Username, GF.Es.Password),
		elastic.SetSniff(false),
	)
	if err != nil {
		log.Fatal("创建 Elasticsearch 客户端失败", zap.Error(err))
	}

	// 2. 初始化 MongoDB 连接
	sess := MgoP.GetMgoConn()
	defer MgoP.DestoryMongoConn(sess)

	coll := sess.DB("qfw").C("wcc_dealXlsxData_0524")
	iter := coll.Find(nil).Select(nil).Iter()

	// 3. 并发控制
	const maxWorkers = 2
	taskCh := make(chan map[string]interface{}, 2000)
	var wg sync.WaitGroup

	// 4. 启动 worker 处理任务
	for i := 0; i < maxWorkers; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			for doc := range taskCh {
				if len(doc) == 0 {
					log.Info("aaa", zap.Any("client", client))
				}
				processOneProposedTest(doc, client)
			}
		}()
	}

	// 5. 逐条读取数据并派发任务
	log.Info("111111", zap.String("222222", "开始处理数据"))
	count := 0
	for doc := make(map[string]interface{}); iter.Next(doc); {
		count++
		if count%1000 == 0 {
			log.Info("dealProposed", zap.Int("current", count), zap.Any("projectname", doc["p1_project_name"]))
		}

		taskCh <- cloneMap(doc) // 防止 map 重用
	}
	close(taskCh)
	wg.Wait()
}

func processOneProposedTest(tmp map[string]interface{}, client *elastic.Client) {
	defer func() {
		if r := recover(); r != nil {
			log.Warn("panic in processOneProposed", zap.Any("recover", r))
		}
	}()

	id := mongodb.BsonIdToSId(tmp["_id"])
	//proposedID := mongodb.BsonIdToSId(tmp["_id"])
	projectName := util.ObjToString(tmp["p1_project_name"])
	buyer := util.ObjToString(tmp["p1_project_owner"])
	//proposed_number := util.ObjToString(tmp["proposed_number"])

	//log.Info("processOneProposed", zap.String("开始查询es", projectName))
	results, err := searchES23(client, projectName, buyer, 20, 50)
	if err != nil {
		log.Warn("searchES22 error", zap.Error(err))
		return
	}
	//log.Info("processOneProposed", zap.String("结束查询es", projectName))
	biddings := []map[string]interface{}{}

	update := map[string]interface{}{}
	// 标讯信息
	for _, re := range results {
		biddingID := util.ObjToString(re["id"])
		da := map[string]interface{}{
			"id":          re["id"],
			"title":       re["title"],
			"area":        re["area"],
			"city":        re["city"],
			"projectname": re["projectname"],
			"score":       re["score"],
			"toptype":     re["toptype"],
			"subtype":     re["subtype"],
			"buyer":       re["buyer"],
			"budget":      re["budget"],
			"buyerperson": re["buyerperson"],
			"buyertel":    re["buyertel"],
			"s_winner":    re["s_winner"],
			"bidamount":   re["bidamount"],
			"winnertel":   re["winnertel"],
			"agency":      re["agency"],
			"publishtime": re["publishtime"],
		}
		//项目信息
		where2 := map[string]interface{}{"ids": biddingID}
		if util.ObjToString(re["toptype"]) == "拟建" {
			projectset, _ := MgoP.FindOne("projectset_proposed", where2)
			if projectset != nil && len((*projectset)) > 0 {
				v3 := map[string]interface{}{
					"project_id":  mongodb.BsonIdToSId((*projectset)["_id"]),
					"projectname": (*projectset)["projectname"],
					"bidamount":   (*projectset)["bidamount"],
					"area":        (*projectset)["area"],
					"city":        (*projectset)["city"],
					"district":    (*projectset)["district"],
					"owner":       (*projectset)["owner"],
					"approvecode": (*projectset)["approvecode"],
				}
				if (*projectset)["owner"] != "" {
					where11 := map[string]interface{}{
						"company_name": (*projectset)["owner"],
					}
					std, _ := MgoQY.FindOne("qyxy_std", where11)
					v3["credit_no"] = (*std)["credit_no"]
				}
				da["project"] = v3
			}
			biddings = append(biddings, da)
		} else {
			projectset, _ := MgoP.FindOne("projectset_20230904", where2)
			if projectset != nil && len((*projectset)) > 0 {
				v3 := map[string]interface{}{
					"project_id":  mongodb.BsonIdToSId((*projectset)["_id"]),
					"projectname": (*projectset)["projectname"],
					"bidamount":   (*projectset)["bidamount"],
					"area":        (*projectset)["area"],
					"city":        (*projectset)["city"],
					"district":    (*projectset)["district"],
					"firsttime":   (*projectset)["firsttime"],
					"bidtype":     (*projectset)["bidtype"],
					"bidstatus":   (*projectset)["bidstatus"],
					"sortprice":   (*projectset)["sortprice"],
					"buyer":       (*projectset)["buyer"],
				}
				if (*projectset)["buyer"] != "" {
					where11 := map[string]interface{}{
						"company_name": (*projectset)["buyer"],
					}
					std, _ := MgoQY.FindOne("qyxy_std", where11)
					v3["credit_no"] = (*std)["credit_no"]
				}
				da["project"] = v3
			}

			biddings = append(biddings, da)
		}

	}

	if len(biddings) > 0 {
		update["bidding"] = biddings
		MgoP.UpdateById("wcc_dealXlsxData_0524", id, map[string]interface{}{"$set": update})
	}

}

// dealProposed22Concurrent 多协程处理,拟建存量数据
func dealProposed22Concurrent() {
	// 1. 初始化 ES 客户端
	client, err := elastic.NewClient(
		elastic.SetURL(GF.Es.URL),
		elastic.SetBasicAuth(GF.Es.Username, GF.Es.Password),
		elastic.SetSniff(false),
	)
	if err != nil {
		log.Fatal("创建 Elasticsearch 客户端失败", zap.Error(err))
	}

	// 2. 初始化 MongoDB 连接
	sess := MgoP.GetMgoConn()
	defer MgoP.DestoryMongoConn(sess)

	coll := sess.DB("qfw").C("projectset_proposed")
	query := map[string]interface{}{
		//"firsttime": map[string]interface{}{
		//	"$gte": 1735660800,
		//	"$lte": 1748102400,
		//},

		"firsttime": map[string]interface{}{
			"$lte": 1735660800,
		},
	}
	iter := coll.Find(query).Select(nil).Sort("-_id").Iter()

	// 3. 并发控制
	const maxWorkers = 1
	taskCh := make(chan map[string]interface{}, 2000)
	var wg sync.WaitGroup

	// 4. 启动 worker 处理任务
	for i := 0; i < maxWorkers; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			for doc := range taskCh {
				if len(doc) == 0 {
					log.Info("aaa", zap.Any("client", client))
				}
				processOneProposed(doc, client)
			}
		}()
	}

	// 5. 逐条读取数据并派发任务
	log.Info("111111", zap.String("222222", "开始处理数据"))
	count := 0
	for doc := make(map[string]interface{}); iter.Next(doc); {
		count++
		if count%1000 == 0 {
			log.Info("dealProposed", zap.Int("current", count), zap.Any("projectname", doc["projectname"]), zap.Any("_id", doc["_id"]))
		}

		//if util.ObjToString(doc["area"]) == "甘肃" {
		//	continue
		//}

		taskCh <- cloneMap(doc) // 防止 map 重用
	}
	close(taskCh)
	wg.Wait()
}

// processOneProposed 处理存量数据
func processOneProposed(tmp map[string]interface{}, client *elastic.Client) {
	defer func() {
		if r := recover(); r != nil {
			log.Warn("panic in processOneProposed", zap.Any("recover", r))
		}
	}()

	proposedID := mongodb.BsonIdToSId(tmp["_id"])
	projectName := util.ObjToString(tmp["projectname"])
	buyer := util.ObjToString(tmp["owner"])
	proposed_number := util.ObjToString(tmp["proposed_number"])

	//log.Info("processOneProposed", zap.String("开始查询es", projectName))
	results, err := searchES23(client, projectName, buyer, 20, 50)
	if err != nil {
		log.Warn("searchES22 error", zap.Error(err))
		return
	}
	//log.Info("processOneProposed", zap.String("结束查询es", projectName))
	biddings := []map[string]interface{}{}

	// 标讯信息
	for _, re := range results {
		biddingID := util.ObjToString(re["id"])
		da := map[string]interface{}{
			"id":          re["id"],
			"title":       re["title"],
			"area":        re["area"],
			"city":        re["city"],
			"projectname": re["projectname"],
			"score":       re["score"],
			"toptype":     re["toptype"],
			"subtype":     re["subtype"],
			"buyer":       re["buyer"],
			"budget":      re["budget"],
			"buyerperson": re["buyerperson"],
			"buyertel":    re["buyertel"],
			"s_winner":    re["s_winner"],
			"bidamount":   re["bidamount"],
			"winnertel":   re["winnertel"],
			"agency":      re["agency"],
			"publishtime": re["publishtime"],
		}
		//项目信息
		where2 := map[string]interface{}{"ids": biddingID}
		if util.ObjToString(re["toptype"]) == "拟建" {
			projectset, _ := MgoP.FindOne("projectset_proposed", where2)
			if projectset != nil && len((*projectset)) > 0 {
				v3 := map[string]interface{}{
					"project_id":    mongodb.BsonIdToSId((*projectset)["_id"]),
					"projectname":   (*projectset)["projectname"],
					"bidamount":     (*projectset)["bidamount"],
					"area":          (*projectset)["area"],
					"city":          (*projectset)["city"],
					"district":      (*projectset)["district"],
					"owner":         (*projectset)["owner"],
					"approvecode":   (*projectset)["approvecode"],
					"approvestatus": (*projectset)["approvestatus"],
					"sourceinfourl": (*projectset)["sourceinfourl"],
				}
				if (*projectset)["owner"] != "" {
					where11 := map[string]interface{}{
						"company_name": (*projectset)["owner"],
					}
					std, _ := MgoQY.FindOne("qyxy_std", where11)
					v3["credit_no"] = (*std)["credit_no"]
				}
				da["project"] = v3
			}
			biddings = append(biddings, da)
		} else {
			projectset, _ := MgoP.FindOne("projectset_20230904", where2)
			if projectset != nil && len((*projectset)) > 0 {
				v3 := map[string]interface{}{
					"project_id":  mongodb.BsonIdToSId((*projectset)["_id"]),
					"projectname": (*projectset)["projectname"],
					"bidamount":   (*projectset)["bidamount"],
					"area":        (*projectset)["area"],
					"city":        (*projectset)["city"],
					"firsttime":   (*projectset)["firsttime"],
					"bidtype":     (*projectset)["bidtype"],
					"bidstatus":   (*projectset)["bidstatus"],
					"sortprice":   (*projectset)["sortprice"],
					"buyer":       (*projectset)["buyer"],
				}
				da["project"] = v3
			}
			biddings = append(biddings, da)
		}

	}

	insert := map[string]interface{}{
		"proposed_id":     proposedID,
		"stype":           1, //代表从拟建数据-> 匹配在建数据
		"proposed_number": proposed_number,
		"buyer":           buyer,
		"projectname":     tmp["projectname"],
		"area":            tmp["area"],
		"city":            tmp["city"],
		"district":        tmp["district"],
		"bidding":         biddings,
		"updatetime":      time.Now().Unix(),
	}

	if isValidCodeFormat(util.ObjToString(tmp["approvecode"])) {
		insert["approvecode"] = tmp["approvecode"]
	}

	if buyer != "" {
		where11 := map[string]interface{}{
			"company_name": buyer,
		}
		std, _ := MgoQY.FindOne("qyxy_std", where11)
		insert["credit_no"] = (*std)["credit_no"]
	}

	whereExist := map[string]interface{}{
		"proposed_id": proposedID,
	}
	if GF.Env.Savecoll == "" {
		GF.Env.Savecoll = "wcc_nj_zj_bidding"
	}
	exist, _ := MgoP.FindOne(GF.Env.Savecoll, whereExist)
	// 存在就更新
	if exist != nil && len(*exist) > 0 {
		exitsid := mongodb.BsonIdToSId((*exist)["_id"])
		MgoP.UpdateById(GF.Env.Savecoll, exitsid, map[string]interface{}{"$set": insert})
	} else {
		insert["comeintime"] = time.Now().Unix()
		MgoP.InsertOrUpdate("qfw", GF.Env.Savecoll, insert)
	}
}

func cloneMap(src map[string]interface{}) map[string]interface{} {
	dst := make(map[string]interface{}, len(src))
	for k, v := range src {
		dst[k] = v
	}
	return dst
}

func dealProposed22() {
	url := GF.Es.URL
	//url := "http://127.0.0.1:19908"
	username := GF.Es.Username
	password := GF.Es.Password
	//index := "bidding" //索引名称
	// 创建 Elasticsearch 客户端
	client, err := elastic.NewClient(
		elastic.SetURL(url),
		elastic.SetBasicAuth(username, password),
		elastic.SetSniff(false),
	)
	if err != nil {
		log.Info("创建 Elasticsearch 客户端失败", zap.Error(err))
	}

	//
	sess := MgoP.GetMgoConn()
	defer MgoP.DestoryMongoConn(sess)

	log.Info("dealProposed", zap.Any("开始处理：拟建数据表", "projectset_proposed"))
	where := map[string]interface{}{
		"firsttime": map[string]interface{}{
			"$gte": 1735660800,
		},
	}
	queryMgo := sess.DB("qfw").C("projectset_proposed").Find(&where).Select(nil).Iter()
	count := 0
	for tmp := make(map[string]interface{}); queryMgo.Next(tmp); count++ {
		if count%1000 == 0 {
			log.Info("dealProposed", zap.Any("current", count), zap.Any("projectname", tmp["projectname"]))
		}

		proposed_id := mongodb.BsonIdToSId(tmp["_id"])
		project_name := util.ObjToString(tmp["projectname"])
		buyer := util.ObjToString(tmp["owner"])
		results, err := searchES22(client, project_name, buyer, 60, 10)
		if err != nil {
			log.Info("searchES22", zap.Error(err))
		}

		projectIds := make([]string, 0) //拟建对应的在建项目ID
		biddingIds := make([]string, 0) //拟建项目对在建项目中的标讯ids
		biddings := make([]map[string]interface{}, 0)

		for _, re := range results {
			bidding_id := util.ObjToString(re["id"])
			biddingIds = append(biddingIds, bidding_id)
			bidding := map[string]interface{}{
				"id":          re["id"],
				"title":       re["title"],
				"projectname": re["projectname"],
				"score":       re["score"],
				"toptype":     re["toptype"],
				"subtype":     re["subtype"],
			}
			biddings = append(biddings, bidding)
		}

		for _, bid := range biddingIds {
			where2 := map[string]interface{}{
				"ids": bid,
			}
			projectset, _ := MgoP.FindOne("projectset_20230904", where2)
			if projectset != nil && len((*projectset)) > 0 {
				projectIds = append(projectIds, mongodb.BsonIdToSId((*projectset)["_id"]))
			}
		}

		insert := map[string]interface{}{
			"proposed_id":  proposed_id,
			"bidding_ids":  removeDuplicates(biddingIds),
			"project_ids":  removeDuplicates(projectIds),
			"biddings":     biddings,
			"project_name": tmp["projectname"],
		}

		MgoP.InsertOrUpdate("qfw", "wcc_dealProposed22", insert)
	}
}

// dealProposed 处理拟建数据表
func dealProposed() {
	sess := MgoP.GetMgoConn()
	defer MgoP.DestoryMongoConn(sess)

	log.Info("dealProposed", zap.Any("开始处理：拟建数据表", "projectset_proposed"))
	where := map[string]interface{}{
		"firsttime": map[string]interface{}{
			"$gte": 1735660800,
		},
	}
	queryMgo := sess.DB("qfw").C("projectset_proposed").Find(&where).Select(nil).Iter()
	count := 0

	for tmp := make(map[string]interface{}); queryMgo.Next(tmp); count++ {
		if count%1000 == 0 {
			log.Info("dealProposed", zap.Any("current", count), zap.Any("projectname", tmp["projectname"]))
		}

		proposed_id := mongodb.BsonIdToSId(tmp["_id"])

		insert := make(map[string]interface{})
		insert["proposed_id"] = proposed_id
		var nzj_follw_records = make([]DwdFnzjFollowRecord, 0)
		err := JianyuSubjectDB.Where("proposed_id = ? ", proposed_id).Find(&nzj_follw_records).Error
		if err != nil {
			log.Info("dealProposed", zap.Error(err))
		}
		//拟建标讯，没有找到对应的在建项目数据
		if len(nzj_follw_records) == 0 {
			insert["has_bidding"] = false
			MgoP.InsertOrUpdate("qfw", "wcc_ok_project_proposed", insert)
			continue
		}

		projectIds := make([]string, 0) //拟建对应的在建项目ID
		biddingIds := make([]string, 0) //拟建项目对在建项目中的标讯ids
		for _, v := range nzj_follw_records {
			biddingIds = append(biddingIds, v.InfoID)
		}

		biddingIds = removeDuplicates(biddingIds)
		insert["bidding_ids"] = biddingIds

		for _, bid := range biddingIds {
			where2 := map[string]interface{}{
				"ids": bid,
			}
			projectset, _ := MgoP.FindOne("projectset_20230904", where2)
			if projectset != nil && len((*projectset)) > 0 {
				projectIds = append(projectIds, mongodb.BsonIdToSId((*projectset)["_id"]))
			}
		}

		if len(projectIds) > 0 {
			insert["project_ids"] = projectIds
		} else {
			insert["has_project"] = false
		}

		MgoP.InsertOrUpdate("qfw", "wcc_ok_project_proposed", insert)
	}

	log.Info("dealProposed", zap.Any("数据处理完毕：拟建数据表", "projectset_proposed"))
}

// removeDuplicates 去除重复字符串
func removeDuplicates(arr []string) []string {
	uniqueMap := make(map[string]bool)
	var result []string
	for _, str := range arr {
		if !uniqueMap[str] {
			uniqueMap[str] = true
			result = append(result, str)
		}
	}
	return result
}

func searchES23(client *elastic.Client, projectName, buyer2 string, scoreThreshold float64, maxResults int) ([]map[string]interface{}, error) {
	fieldsToTry := []string{"projectname.pname", "title", "detail"}

	filtersToTry := [][]elastic.Query{
		{elastic.NewTermsQuery("subtype", "中标", "成交", "合同", "单一")},
		{elastic.NewTermsQuery("toptype", "招标", "预告", "采购意向")},
		{elastic.NewTermsQuery("toptype", "拟建")},
	}

	var allResults []*elastic.SearchHit
	seenIDs := make(map[string]bool)

	for _, field := range fieldsToTry {
		for _, filter := range filtersToTry {
			// 构建查询：使用 MultiMatchQuery + phrase
			query := elastic.NewBoolQuery().
				Must(elastic.NewMultiMatchQuery(projectName, field).Type("phrase")).
				Filter(filter...)

			// 执行查询
			searchResult, err := client.Search().
				Index("bidding").
				Query(query).
				Size(70).
				Do(context.Background())
			if err != nil {
				return nil, err
			}

			// 去重处理
			for _, hit := range searchResult.Hits.Hits {
				if !seenIDs[hit.Id] {
					seenIDs[hit.Id] = true
					allResults = append(allResults, hit)
				}
			}
			if len(allResults) >= maxResults {
				break
			}
		}
		if len(allResults) >= maxResults {
			break
		}
	}

	var results []map[string]interface{}
	seenProjectNames := make(map[string]bool)
	seenProjectCodes := make(map[string]bool)
	bidamountMap := make(map[float64]bool)
	//subtypeMap := make(map[string]bool)

	for _, hit := range allResults {
		var doc map[string]interface{}
		if err := json.Unmarshal(hit.Source, &doc); err != nil {
			log.Info("解析文档失败", zap.Error(err))
			continue
		}

		projectNameValue := util.ObjToString(doc["projectname"])
		if projectNameValue == "" {
			continue
		}

		projectCode := util.ObjToString(doc["projectcode"])
		if projectCode != "" {
			if seenProjectCodes[projectCode] {
				continue
			}
			seenProjectCodes[projectCode] = true
		}

		bidamount := util.Float64All(doc["bidamount"])
		if bidamount != 0 {
			if bidamountMap[bidamount] {
				continue
			}
			bidamountMap[bidamount] = true
		}

		// 相似度筛选
		score := *hit.Score
		doc["score"] = score //相似度
		if score < scoreThreshold {
			continue
		}

		doc["score"] = score
		detail := util.ObjToString(doc["detail"])

		// 字段中必须包含 projectName
		if buyer2 != "" {
			if !strings.Contains(detail, projectName) && !strings.Contains(detail, buyer2) {
				continue
			}
		}

		if seenProjectNames[projectNameValue] {
			continue
		}
		seenProjectNames[projectNameValue] = true

		results = append(results, doc)

		if len(results) >= maxResults {
			break
		}
	}
	// 排序：按 score 降序
	sort.Slice(results, func(i, j int) bool {
		si := util.Float64All(results[i]["score"])
		sj := util.Float64All(results[j]["score"])
		return si > sj
	})

	return results, nil
}

func searchES22(client *elastic.Client, projectName, buyer2 string, scoreThreshold float64, maxResults int) ([]map[string]interface{}, error) {
	fieldsToTry := []string{"projectname.pname", "title", "detail"}
	filtersToTry := [][]elastic.Query{
		{elastic.NewTermsQuery("subtype", "中标", "成交", "合同", "单一")},
		{elastic.NewTermsQuery("toptype", "招标", "预告", "采购意向")},
		{elastic.NewTermsQuery("toptype", "拟建")},
	}

	var allResults []*elastic.SearchHit
	seenIDs := make(map[string]bool)

	for _, field := range fieldsToTry {
		for _, filter := range filtersToTry {
			// 构建查询
			query := elastic.NewBoolQuery().
				Must(elastic.NewMatchQuery(field, projectName)).
				Filter(filter...)

			// 执行查询
			searchResult, err := client.Search().
				Index("bidding").
				Query(query).
				Size(70). // 多取一些，后面做筛选和去重
				Do(context.Background())
			if err != nil {
				return nil, err
			}

			for _, hit := range searchResult.Hits.Hits {
				if !seenIDs[hit.Id] {
					allResults = append(allResults, hit)
					seenIDs[hit.Id] = true
				}
			}

			if len(allResults) >= maxResults {
				break
			}
		}
		if len(allResults) >= maxResults {
			break
		}
	}

	var results []map[string]interface{}
	seenProjectNames := make(map[string]bool)
	seenProjectCodes := make(map[string]bool)
	bidamountMap := make(map[float64]bool)

	for _, hit := range allResults {
		var doc map[string]interface{}
		if err := json.Unmarshal(hit.Source, &doc); err != nil {
			log.Info("解析文档失败", zap.Error(err))
			continue
		}

		projectNameValue := util.ObjToString(doc["projectname"])
		if projectNameValue == "" {
			continue
		}

		projectCode := util.ObjToString(doc["projectcode"])
		if seenProjectCodes[projectCode] {
			continue
		}
		seenProjectCodes[projectCode] = true

		bidamount := util.Float64All(doc["bidamount"])
		if bidamountMap[bidamount] {
			continue
		}
		bidamountMap[bidamount] = true

		// 相似度筛选
		score := *hit.Score
		doc["score"] = score //相似度
		if score < scoreThreshold {
			continue
		}

		//id := util.ObjToString(doc["id"])
		//doc["jyhref"] = GetJyURLByID(id)

		//if site := util.ObjToString(doc["site"]); site == "中华人民共和国自然资源部" {
		//	doc["title"] = "土地出让" + "-" + util.ObjToString(doc["title"])
		//}

		// enrich: total_investment
		//if bidData, _ := MgoB.FindById("bidding", id, nil); bidData != nil {
		//	if util.Float64All((*bidData)["total_investment"]) > 0 {
		//		doc["total_investment"] = (*bidData)["total_investment"]
		//	}
		//}

		doc["score"] = score
		detail := util.ObjToString(doc["detail"])

		// 字段中必须包含 projectName
		if buyer2 != "" {
			if !strings.Contains(detail, projectName) && !strings.Contains(detail, buyer2) {
				continue
			}
		}

		if seenProjectNames[projectNameValue] {
			continue
		}
		seenProjectNames[projectNameValue] = true

		results = append(results, doc)

		if len(results) >= maxResults {
			break
		}
	}

	return results, nil
}

func searchES(client *elastic.Client, projectName, buyer2 string) ([]map[string]interface{}, error) {
	query := elastic.NewBoolQuery().
		Must(
			//elastic.NewMatchQuery("projectname.pname", projectName), // 模糊匹配 projectname
			//elastic.NewMatchQuery("title", projectName), // 模糊匹配 projectname
			elastic.NewMatchQuery("detail", projectName), // 模糊匹配 projectname
			//elastic.NewTermQuery("area", "安徽"),                       // 过滤区域
			elastic.NewTermsQuery("subtype", "中标", "成交", "合同", "单一"), // 过滤 subtype
			//elastic.NewTermsQuery("toptype", "招标", "预告", "采购意向"), // 过滤 subtype
			//elastic.NewTermsQuery("toptype", "拟建"), // 过滤 subtype
		)

	searchResult, err := client.Search().
		Index("bidding").
		Query(query).
		Size(70). // 先取 12 条，确保足够数据
		Do(context.Background())

	if err != nil {
		return nil, err
	}

	// 结果集
	var results []map[string]interface{}
	seenProjectNames := make(map[string]bool) // 用于去重
	seenProjectCode := make(map[string]bool)  // 用于去重
	bidamountMap := make(map[float64]bool)
	for _, hit := range searchResult.Hits.Hits {
		var doc map[string]interface{}
		err := json.Unmarshal(hit.Source, &doc)
		if err != nil {
			log.Info("解析文档失败", zap.Error(err))
			continue
		}

		// 获取 `projectname`，防止 key 不存在时的错误
		projectNameValue, ok := doc["projectname"].(string)

		bidamount := util.Float64All(doc["bidamount"])
		if !ok {
			log.Info("⚠️ 缺少 projectname 字段，跳过：", zap.Any("projectname", doc["projectname"]))
			continue
		}

		projectCodeValue := util.ObjToString(doc["projectcode"])

		if seenProjectCode[projectCodeValue] {
			continue
		}
		if projectCodeValue != "" {
			seenProjectCode[projectCodeValue] = true
		}

		// **处理额外字段**
		id := util.ObjToString(doc["id"])
		bidData, _ := MgoB.FindById("bidding", id, nil)
		if util.Float64All((*bidData)["total_investment"]) > 0 {
			doc["total_investment"] = (*bidData)["total_investment"]
		}
		doc["jyhref"] = GetJyURLByID(id)
		score := *hit.Score

		site := util.ObjToString(doc["site"])
		if site == "中华人民共和国自然资源部" {
			doc["title"] = "土地出让" + "-" + util.ObjToString(doc["title"])
		}
		doc["score"] = score //相似度

		detail := util.ObjToString(doc["detail"])
		if !strings.Contains(detail, projectName) {
			continue
		}

		// **去重逻辑**：如果 `projectname` 已经出现过，则跳过
		if seenProjectNames[projectNameValue] {
			continue
		}

		if bidamountMap[bidamount] {
			continue
		}
		// **记录该 `projectname`，避免重复**
		seenProjectNames[projectNameValue] = true
		bidamountMap[bidamount] = true

		// **加入结果集**
		results = append(results, doc)

		// **如果已经找到 6 条不同 `projectname`，就跳出循环**
		if len(results) >= 10 {
			break
		}
	}

	//2、判断正文包含采购单位
	for _, hit := range searchResult.Hits.Hits {
		var doc map[string]interface{}
		err := json.Unmarshal(hit.Source, &doc)
		if err != nil {
			log.Info("解析文档失败：", zap.Error(err))
			continue
		}

		// 获取 `projectname`，防止 key 不存在时的错误
		projectNameValue, ok := doc["projectname"].(string)
		bidamount := util.Float64All(doc["bidamount"])
		if !ok {
			log.Info("⚠️ 缺少 projectname 字段，跳过：", zap.Any("projectname", doc["projectname"]))
			continue
		}

		// **处理额外字段**
		id := util.ObjToString(doc["id"])
		doc["jyhref"] = GetJyURLByID(id)
		score := *hit.Score
		doc["score"] = score //相似度
		site := util.ObjToString(doc["site"])
		if site == "中华人民共和国自然资源部" {
			doc["title"] = "土地出让" + "-" + util.ObjToString(doc["title"])
		}

		//判断正文包含采购单位
		detail := util.ObjToString(doc["detail"])
		if !strings.Contains(detail, buyer2) {
			continue
		}
		// **去重逻辑**：如果 `projectname` 已经出现过，则跳过
		if seenProjectNames[projectNameValue] {
			continue
		}
		if bidamountMap[bidamount] {
			continue
		}

		// **记录该 `projectname`，避免重复**
		seenProjectNames[projectNameValue] = true
		bidamountMap[bidamount] = true

		// **加入结果集**
		results = append(results, doc)

		// **如果已经找到 6 条不同 `projectname`，就跳出循环**
		if len(results) >= 10 {
			break
		}
	}

	return results, nil
}

// GetJyURLByID 获取剑鱼地址
func GetJyURLByID(id string) string {
	var Url = "https://www.jianyu360.com/article/content/%s.html"
	url := fmt.Sprintf(Url, util.CommonEncodeArticle("content", id))

	return url
}

// GetIdByURL 解密url，获取bidding ID
func GetIdByURL(url string) string {
	if strings.Contains(url, "work-bench") {
		return ""
	}
	if strings.Contains(url, "/article/content") {
		urls := strings.Split(url, "content/")
		res := strings.Split(urls[1], ".html")
		ids := util.CommonDecodeArticle("content", res[0])
		return ids[0]
	}
	if strings.HasSuffix(url, "appid") {
		urls := strings.Split(url, "entservice/")
		res := strings.Split(urls[1], ".html")
		se := util.SimpleEncrypt{Key: "entservice"}
		id := se.DecodeString(res[0])
		return id
	}

	return ""
}

// isValidCodeFormat 判断 拟建项目编码
func isValidCodeFormat(s string) bool {
	pattern := `^\d{4}-\d{6}-\d{2}-\d{2}-\d{6}$`
	matched, err := regexp.MatchString(pattern, s)
	if err != nil {
		return false
	}
	return matched
}