|
@@ -238,3 +238,140 @@ func EsMultiMatchFunctionScoreQueryTimeSort(indexName, keyWord string, startSize
|
|
}
|
|
}
|
|
return
|
|
return
|
|
}
|
|
}
|
|
|
|
+
|
|
|
|
+func EsArticleSearch(keyWord string, startSize, pageSize int, orderColumn string) (result []*models.SearchItem, total int64, err error) {
|
|
|
|
+ indexName := utils.IndexName
|
|
|
|
+ client := utils.Client
|
|
|
|
+ keyWordArr, err := GetIndustryMapNameSliceV3(keyWord)
|
|
|
|
+ keyWordArr = RemoveDuplicatesAndEmpty(keyWordArr)
|
|
|
|
+ keyWordLen := len(keyWordArr)
|
|
|
|
+ if keyWordLen <= 0 {
|
|
|
|
+ keyWordArr = append(keyWordArr, keyWord)
|
|
|
|
+ keyWordLen = len(keyWordArr)
|
|
|
|
+ }
|
|
|
|
+ mustMap := make([]interface{}, 0)
|
|
|
|
+ shouldMap := make(map[string]interface{}, 0)
|
|
|
|
+ shouldMapquery := make([]interface{}, 0)
|
|
|
|
+ // @Param OrderColumn query int true "排序字段 ,Comprehensive综合 ,Matching匹配度 ,PublishDate 发布时间 "
|
|
|
|
+ //keyWordWeight := GetWeight(keyWordLen)
|
|
|
|
+ lenkeyWordArr := len(keyWordArr)
|
|
|
|
+ for k, v := range keyWordArr {
|
|
|
|
+ if v != "" {
|
|
|
|
+ shouldMapquery = append(shouldMapquery, map[string]interface{}{
|
|
|
|
+ "function_score": map[string]interface{}{
|
|
|
|
+ "query": map[string]interface{}{
|
|
|
|
+ "multi_match": map[string]interface{}{
|
|
|
|
+ "boost": (lenkeyWordArr - k) * 100, //给查询的值赋予权重
|
|
|
|
+ "fields": []interface{}{"Title"},
|
|
|
|
+ "query": v,
|
|
|
|
+ },
|
|
|
|
+ },
|
|
|
|
+ },
|
|
|
|
+ })
|
|
|
|
+
|
|
|
|
+ shouldMapquery = append(shouldMapquery, map[string]interface{}{
|
|
|
|
+ "function_score": map[string]interface{}{
|
|
|
|
+ "query": map[string]interface{}{
|
|
|
|
+ "multi_match": map[string]interface{}{
|
|
|
|
+ "boost": (lenkeyWordArr-k)*100 - 1, //给查询的值赋予权重
|
|
|
|
+ "fields": []interface{}{"BodyText"},
|
|
|
|
+ "query": v,
|
|
|
|
+ },
|
|
|
|
+ },
|
|
|
|
+ },
|
|
|
|
+ })
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ shouldMap = map[string]interface{}{
|
|
|
|
+ "should": shouldMapquery,
|
|
|
|
+ }
|
|
|
|
+ //排序
|
|
|
|
+ sortMap := make([]interface{}, 0)
|
|
|
|
+ //时间
|
|
|
|
+ sortMap = append(sortMap, map[string]interface{}{
|
|
|
|
+ "PublishDate": map[string]interface{}{
|
|
|
|
+ "order": "desc",
|
|
|
|
+ },
|
|
|
|
+ })
|
|
|
|
+ //高亮
|
|
|
|
+ highlightMap := make(map[string]interface{}, 0)
|
|
|
|
+ highlightMap = map[string]interface{}{
|
|
|
|
+ "fields": map[string]interface{}{
|
|
|
|
+ "BodyText": map[string]interface{}{},
|
|
|
|
+ "Title": map[string]interface{}{},
|
|
|
|
+ },
|
|
|
|
+ //样式 红色
|
|
|
|
+ "post_tags": []interface{}{"</font>"},
|
|
|
|
+ "pre_tags": []interface{}{"<font color='red'>"},
|
|
|
|
+ }
|
|
|
|
+
|
|
|
|
+ mustMap = append(mustMap, map[string]interface{}{
|
|
|
|
+ "bool": shouldMap,
|
|
|
|
+ })
|
|
|
|
+ queryMap := map[string]interface{}{
|
|
|
|
+ "query": map[string]interface{}{
|
|
|
|
+ "bool": map[string]interface{}{
|
|
|
|
+ "must": mustMap,
|
|
|
|
+ },
|
|
|
|
+ },
|
|
|
|
+ }
|
|
|
|
+ if orderColumn == "PublishDate" {
|
|
|
|
+ queryMap["sort"] = sortMap
|
|
|
|
+ }
|
|
|
|
+ queryMap["from"] = startSize
|
|
|
|
+ queryMap["size"] = pageSize
|
|
|
|
+ queryMap["highlight"] = highlightMap
|
|
|
|
+ jsonBytes, _ := json.Marshal(queryMap)
|
|
|
|
+ fmt.Println(string(jsonBytes))
|
|
|
|
+ request := client.Search(indexName).Source(queryMap) // sets the JSON request
|
|
|
|
+ searchByMatch, err := request.Do(context.Background())
|
|
|
|
+ if searchByMatch != nil {
|
|
|
|
+ if searchByMatch.Hits != nil {
|
|
|
|
+ for _, v := range searchByMatch.Hits.Hits {
|
|
|
|
+ var isAppend bool
|
|
|
|
+ articleJson, err := v.Source.MarshalJSON()
|
|
|
|
+ if err != nil {
|
|
|
|
+ return nil, 0, err
|
|
|
|
+ }
|
|
|
|
+ article := new(models.CygxArticleEs)
|
|
|
|
+ err = json.Unmarshal(articleJson, &article)
|
|
|
|
+ if err != nil {
|
|
|
|
+ return nil, 0, err
|
|
|
|
+ }
|
|
|
|
+ searchItem := new(models.SearchItem)
|
|
|
|
+ searchItem.ArticleId, _ = strconv.Atoi(v.Id)
|
|
|
|
+ if len(v.Highlight["BodyText"]) > 0 {
|
|
|
|
+ searchItem.Body = v.Highlight["BodyText"]
|
|
|
|
+ } else {
|
|
|
|
+ bodyRune := []rune(article.BodyText)
|
|
|
|
+ bodyRuneLen := len(bodyRune)
|
|
|
|
+ if bodyRuneLen > 100 {
|
|
|
|
+ bodyRuneLen = 100
|
|
|
|
+ }
|
|
|
|
+ body := string(bodyRune[:bodyRuneLen])
|
|
|
|
+ searchItem.Body = []string{body}
|
|
|
|
+ }
|
|
|
|
+ var title string
|
|
|
|
+ if len(v.Highlight["Title"]) > 0 {
|
|
|
|
+ title = v.Highlight["Title"][0]
|
|
|
|
+ } else {
|
|
|
|
+ title = article.Title
|
|
|
|
+ }
|
|
|
|
+ searchItem.Title = title
|
|
|
|
+ searchItem.PublishDate = article.PublishDate
|
|
|
|
+ searchItem.ExpertBackground = article.ExpertBackground
|
|
|
|
+ searchItem.CategoryId = article.CategoryId
|
|
|
|
+ for _, v_result := range result {
|
|
|
|
+ if v_result.ArticleId == searchItem.ArticleId {
|
|
|
|
+ isAppend = true
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ if !isAppend {
|
|
|
|
+ result = append(result, searchItem)
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+ total = searchByMatch.Hits.TotalHits.Value
|
|
|
|
+ }
|
|
|
|
+ return
|
|
|
|
+}
|