jyMicroservices/jyBXCore/rpc/model/es/search.go

package es

import (
	"fmt"
	"jyBXCore/rpc/bxcore"
	IC "jyBXCore/rpc/init"
	"strconv"
	"strings"
	"time"

	MC "app.yhyue.com/moapp/jybase/common"
	elastic "app.yhyue.com/moapp/jybase/es"
	"github.com/zeromicro/go-zero/core/logx"
)

// GetSearchQuery  整理关键词等查询条件
func GetSearchQuery(in *bxcore.SearchReq, mustQuery string) (qstr string) {
	var (
		//搜索范围是否只有附件
		//搜索范围只选择附件，是否有附件条件无效；
		isFileSearch                            = in.SelectType == "filetext"
		wordsMusts, wordsShould, musts, mustNot []string
		findFields                              string
		selectTypeArr                           = strings.Split(in.SelectType, ",")
	)
	if selectTypeArr == nil || len(selectTypeArr) == 0 {
		findFields = `"title"`
	} else {
		findFields = fmt.Sprintf(`"%s"`, strings.Join(selectTypeArr, "\",\""))
	}
	switchBool := strings.Contains(findFields, "detail") && strings.Contains(findFields, "title") && IC.C.SearchTypeSwitch
	if mustQuery != "" {
		musts = append(musts, mustQuery)
	}
	//此时关键词中间有IC.C.JYKeyMark进行隔离
	if in.KeyWords != "" {
		var (
			keyWordsMusts []string
		)
		for _, v := range strings.Split(in.KeyWords, IC.C.JYKeyMark) {
			if elastic.ReplaceYH(v) == "" {
				continue
			}
			//单个字  搜索范围 有全文或者附件 无标题 例如：学 虚拟机 detail  搜索的时候加上标题
			//detail 正文不支持单字查询
			if len([]rune(elastic.ReplaceYH(v))) == 1 && DetailFileORTitle(findFields) {
				findFields += `,"title"`
			} else if switchBool && len([]rune(elastic.ReplaceYH(v))) > 1 {
				//标题 全文搜索 搜索类型开关打开 默认搜索全文；（全文包含标题）(单字排除)
				if strings.Contains(findFields, `"title",`) {
					findFields = strings.Replace(findFields, `"title",`, ``, -1)
				} else if strings.Contains(findFields, `,"title"`) {
					findFields = strings.Replace(findFields, `,"title"`, ``, -1)
				}
			}
			keyWordsMusts = append(keyWordsMusts, fmt.Sprintf(fmt.Sprintf(multiMatch, "%s", findFields), elastic.ReplaceYH(v)))
		}
		//搜索关键词模式；默认0：包含所有，1：包含任意
		if in.WordsMode == 1 {
			wordsShould = append(wordsShould, fmt.Sprintf(elastic.NgramMust, strings.Join(keyWordsMusts, ",")))
		} else {
			wordsMusts = append(wordsMusts, keyWordsMusts...)
		}
	}
	//附加词
	if in.AdditionalWords != "" {
		//多组附加词，每组间,号隔开。每组内如果关键词中间有空格，自动分词
		var (
			addWordsMusts []string
		)
		for _, aws := range strings.Split(in.AdditionalWords, ",") {
			var (
				addWordsMust []string
			)
			for _, v := range strings.Split(aws, IC.C.JYKeyMark) {
				if elastic.ReplaceYH(v) == "" {
					continue
				}
				//单个字  搜索范围 有全文或者附件 无标题 例如：学 虚拟机 detail  搜索的时候加上标题
				//detail 正文不支持单字查询
				if len([]rune(elastic.ReplaceYH(v))) == 1 && DetailFileORTitle(findFields) {
					findFields += `,"title"`
				} else if switchBool && len([]rune(elastic.ReplaceYH(v))) > 1 {
					//标题 全文搜索 搜索类型开关打开 默认搜索全文；（全文包含标题）(单字排除)
					if strings.Contains(findFields, `"title",`) {
						findFields = strings.Replace(findFields, `"title",`, ``, -1)
					} else if strings.Contains(findFields, `,"title"`) {
						findFields = strings.Replace(findFields, `,"title"`, ``, -1)
					}
				}
				addWordsMust = append(addWordsMust, fmt.Sprintf(fmt.Sprintf(multiMatch, "%s", findFields), elastic.ReplaceYH(v)))
				addWordsMusts = append(addWordsMusts, addWordsMust...)
				if in.WordsMode == 0 {
					wordsMusts = append(wordsMusts, addWordsMust...)
				}
				addWordsMust = []string{}
			}
			//搜索关键词模式；默认0：包含所有，1：包含任意
			if in.WordsMode == 1 {
				wordsShould = append(wordsShould, fmt.Sprintf(elastic.NgramMust, strings.Join(addWordsMusts, ",")))
				addWordsMusts = []string{}
			}
		}
	}
	//搜索关键词模式；默认0：包含所有，1：包含任意
	//包含任意一组
	if len(wordsShould) > 0 {
		musts = append(musts, fmt.Sprintf(queryBoolShould, strings.Join(wordsShould, ",")))
	} else if len(wordsMusts) > 0 {
		musts = append(musts, fmt.Sprintf(elastic.NgramMust, strings.Join(wordsMusts, ",")))
	}
	//排除词
	if notKey := strings.TrimSpace(in.ExclusionWords); notKey != "" {
		notKeyMultiMatch := fmt.Sprintf(multiMatch, "%s", findFields)
		var notKeyMustNot []string
		//多组排除词
		for _, nks := range strings.Split(notKey, ",") {
			//单组排除词 空格分割
			for _, v := range strings.Split(nks, IC.C.JYKeyMark) {
				v = strings.TrimSpace(v)
				if v == "" {
					continue
				}
				if len([]rune(elastic.ReplaceYH(v))) == 1 {
					//单个字 搜索范围 有全文或者附件 无标题 例如：学 虚拟机 detail  搜索的时候加上标题
					if DetailFileORTitle(findFields) {
						notKeyMultiMatch = fmt.Sprintf(multiMatch, "%s", findFields+`,"title"`)
					}
				}
				notKeyMustNot = append(notKeyMustNot, fmt.Sprintf(notKeyMultiMatch, elastic.ReplaceYH(v)))
			}
		}
		mustNot = append(mustNot, fmt.Sprintf(queryBoolShould, strings.Join(notKeyMustNot, ",")))
	}
	//行业
	if in.Industry != "" {
		musts = append(musts, fmt.Sprintf(queryBoolMust, `"`+strings.ReplaceAll(in.Industry, ",", `","`)+`"`))
	}
	//价格
	if in.Price != "" && len(strings.Split(in.Price, "-")) > 1 {
		minPrice, maxPrice := strings.Split(in.Price, "-")[0], strings.Split(in.Price, "-")[1]
		if minPrice != "" || maxPrice != "" {
			sq := ``
			if minPrice != "" {
				min, _ := strconv.ParseFloat(minPrice, 64)
				minPrice = fmt.Sprintf("%.0f", min*10000)
				if minPrice == "0" {
					minPrice = ""
				}
			}
			if maxPrice != "" {
				max, _ := strconv.ParseFloat(maxPrice, 64)
				maxPrice = fmt.Sprintf("%.0f", max*10000)
				if maxPrice == "0" {
					maxPrice = ""
				}
			}
			if minPrice != "" {
				sq += fmt.Sprintf(gte, minPrice)
			}
			if minPrice != "" && maxPrice != "" {
				sq += `,`
			}
			if maxPrice != "" {
				sq += fmt.Sprintf(lte, maxPrice)
			}
			if minPrice != "" || maxPrice != "" {
				query_price := fmt.Sprintf(queryBoolShould, fmt.Sprintf(queryBoolMustBoolShould, sq, sq))
				musts = append(musts, query_price)
			}
		}
	}
	//采购单位联系方式
	hasBuyerTel := in.BuyerTel
	if hasBuyerTel != "" {
		if hasBuyerTel == "y" {
			mustNot = append(mustNot, fmt.Sprintf(queryMissing, "buyertel"))
		} else {
			musts = append(musts, fmt.Sprintf(queryMissing, "buyertel"))
		}
	}
	//中标企业联系方式
	hasWinnerTel := in.WinnerTel
	if hasWinnerTel != "" {
		if hasWinnerTel == "y" {
			mustNot = append(mustNot, fmt.Sprintf(queryMissing, "winnertel"))
		} else {
			musts = append(musts, fmt.Sprintf(queryMissing, "winnertel"))
		}
	}
	//附件
	fileExists := in.FileExists
	if !isFileSearch && fileExists != "" {
		if fileExists == "1" { //有附件
			mustNot = append(mustNot, fmt.Sprintf(queryMissing, "isValidFile"))
			musts = append(musts, fmt.Sprintf(queryBoolMustTerm, 1))
		} else if fileExists == "-1" { //无附件
			musts = append(musts, fmt.Sprintf(queryMissing, "isValidFile"))
		}
	}
	// 如果是领域化数据则需要加标签
	if in.BidField != "" {
		musts = append(musts, fmt.Sprintf(queryBoolMustTermDomain, in.BidField))
	}
	qstr = fmt.Sprintf(query, strings.Join(musts, ","), strings.Join(mustNot, ","))
	logx.Info("qstr:", qstr)
	return
}

// GetBidSearchQuery  整理地区、城市、发布时间、信息类型、采购单位类型 查询条件
func GetBidSearchQuery(in *bxcore.SearchReq) string {
	query := ``
	//省份
	area := in.Province
	if area != "" {
		query += `{"terms":{"area":[`
		for k, v := range strings.Split(area, ",") {
			if k > 0 {
				query += `,`
			}
			query += `"` + v + `"`
		}
		query += `]}}`
	}
	//
	city := in.City
	if city != "" {
		if len(query) > 0 {
			query += ","
		}
		query += `{"terms":{"city":[`
		for k, v := range strings.Split(city, ",") {
			if k > 0 {
				query += `,`
			}
			query += `"` + v + `"`
		}
		query += `]}}`
	}
	if query != "" {
		query = fmt.Sprintf(queryBoolShould, query)
	}
	//发布时间
	publishTime := in.PublishTime
	if publishTime != "" && len(strings.Split(publishTime, "-")) > 1 {
		if len(query) > 0 {
			query += ","
		}
		startTime := strings.Split(publishTime, "-")[0]
		endTime := strings.Split(publishTime, "-")[1]
		query += `{"range":{"publishtime":{`
		if startTime != "" {
			query += `"gte":` + startTime
		}
		if startTime != "" && endTime != "" {
			query += `,`
		}
		if endTime != "" {
			//电脑端 时间选择 开始时间当天和结束时间当天相同
			if startTime == endTime {
				et, _ := strconv.ParseInt(endTime, 0, 64)
				etTime := time.Unix(et, 0)
				endTime = fmt.Sprint(time.Date(etTime.Year(), etTime.Month(), etTime.Day()+1, 0, 0, 0, 0, time.Local).Unix())
			}
			query += `"lt":` + endTime
		}
		query += `}}}`
	}
	//信息类型-二级
	subtype := in.Subtype
	topType := MC.If(in.TopType != "", strings.Split(in.TopType, ","), []string{}).([]string)
	allType := ``
	//二级分类
	if subtype != "" {
		var typeInt = 0
		allType += `{"terms":{"subtype":[`
		for k, v := range strings.Split(subtype, ",") {
			if tType := MC.If(topTypeMap[v] != "" && in.TopType == "", topTypeMap[v], "").(string); tType != "" {
				topType = append(topType, tType)
				typeInt += 1
				continue
			}
			if k > typeInt {
				allType += `,`
			}
			allType += `"` + v + `"`
		}
		allType += `]}}`
	}
	//信息类型  一级分类
	logx.Info("topType:", topType)
	if len(topType) > 0 {
		if allType != "" {
			allType += ","
		}
		allType += `{"terms":{"toptype":[`
		for k, v := range topType {
			if k > 0 {
				allType += `,`
			}
			allType += `"` + v + `"`
		}
		allType += `]}}`
	}

	if allType != "" {
		if query != "" {
			query += ","
		}
		query += fmt.Sprintf(queryBoolShould, allType)
	}
	//采购单位类型
	buyerClass := in.BuyerClass
	if buyerClass != "" {
		if len(query) > 0 {
			query += ","
		}
		query += `{"terms":{"buyerclass":[`
		for k, v := range strings.Split(buyerClass, ",") {
			if k > 0 {
				query += `,`
			}
			query += `"` + v + `"`
		}
		query += `]}}`
	}
	return query
}

// DetailFileORTitle 包含正文或 附件 不包含标题
func DetailFileORTitle(findFields string) bool {
	return (strings.Contains(findFields, `"detail"`) || strings.Contains(findFields, `"filetext"`)) && !strings.Contains(findFields, `"title"`)
}