package elastic
import (
"context"
"encoding/json"
"errors"
"eta_gn/eta_api/models"
"eta_gn/eta_api/models/data_manage"
"eta_gn/eta_api/utils"
"fmt"
"github.com/olivere/elastic/v7"
"strconv"
"strings"
)
// indexName:索引名称
// mappingJson:表结构
func EsCreateIndex(indexName, mappingJson string) (err error) {
client := utils.EsClient
//定义表结构
exists, err := client.IndexExists(indexName).Do(context.Background()) //<5>
if err != nil {
return
}
if !exists {
resp, err := client.CreateIndex(indexName).BodyJson(mappingJson).Do(context.Background())
//BodyJson(bodyJson).Do(context.Background())
if err != nil {
fmt.Println("CreateIndex Err:" + err.Error())
return err
}
fmt.Println(resp.Index, resp.ShardsAcknowledged, resp.Acknowledged)
} else {
fmt.Println(indexName + " 已存在")
}
return
}
// 删除数据
func EsDeleteData(indexName, docId string) (err error) {
client := utils.EsClient
resp, err := client.Delete().Index(indexName).Id(docId).Do(context.Background())
fmt.Println(resp)
if err != nil {
return
}
if resp.Status == 0 {
fmt.Println("删除成功")
} else {
fmt.Println("AddData", resp.Status, resp.Result)
}
return
}
func MappingModify(indexName, mappingJson string) {
client := utils.EsClient
result, err := client.PutMapping().Index(indexName).BodyString(mappingJson).Do(context.Background())
fmt.Println(err)
fmt.Println(result)
return
}
// EsAddOrEditEdbInfoData 新增/修改es中的指标数据
func EsAddOrEditEdbInfoData(indexName, docId string, item *data_manage.EdbInfoList) (err error) {
defer func() {
if err != nil {
fmt.Println("EsAddOrEditData Err:", err.Error())
}
}()
client := utils.EsClient
resp, err := client.Index().Index(indexName).Id(docId).BodyJson(item).Do(context.Background())
if err != nil {
fmt.Println("新增失败:", err.Error())
return err
}
fmt.Println(resp)
if resp.Status == 0 {
fmt.Println("新增成功", resp.Result)
err = nil
} else {
fmt.Println("AddData", resp.Status, resp.Result)
}
return
}
// SearchEdbInfoData 查询es中的指标数据
func SearchEdbInfoData(keywordStr string, from, size, filterSource, source int, edbInfoType int8, frequency string, noPermissionEdbInfoIdList, noPermissionEdbClassifyIdList, collectEdbInfoIdList, edbTypeList []int, edbAuth, sysUserId int) (total int64, list []*data_manage.EdbInfoList, err error) {
indexName := utils.DATA_INDEX_NAME
list = make([]*data_manage.EdbInfoList, 0)
defer func() {
if err != nil {
fmt.Println("SearchEdbInfoData Err:", err.Error())
}
}()
highlight := elastic.NewHighlight()
highlight = highlight.Fields(elastic.NewHighlighterField("EdbCode"), elastic.NewHighlighterField("EdbName"))
highlight = highlight.PreTags("").PostTags("")
mustMap := make([]interface{}, 0)
mustNotMap := make([]interface{}, 0)
switch filterSource {
case 2:
mustMap = append(mustMap, map[string]interface{}{
"term": map[string]interface{}{
"Frequency.keyword": "月度",
//"Frequency.keyword": "月度",
},
})
case 3:
case 4:
mustMap = append(mustMap, map[string]interface{}{
"term": map[string]interface{}{
"EdbType": 1,
},
})
case 5:
mustMap = append(mustMap, map[string]interface{}{
"term": map[string]interface{}{
"Source": 6,
},
})
case 6:
mustMap = append(mustMap, map[string]interface{}{
"match": map[string]interface{}{
"Frequency.keyword": "年度",
},
})
}
//指标来源
if source > 0 {
mustMap = append(mustMap, map[string]interface{}{
"term": map[string]interface{}{
"Source": source,
//"Frequency.keyword": "月度",
},
})
}
if frequency != "" {
mustMap = append(mustMap, map[string]interface{}{
"term": map[string]interface{}{
"Frequency.keyword": frequency,
//"Frequency.keyword": "月度",
},
})
}
// noPermissionEdbInfoIdList 无权限指标id
if len(noPermissionEdbInfoIdList) > 0 {
mustNotMap = append(mustNotMap, map[string]interface{}{
"terms": map[string]interface{}{
"EdbInfoId": noPermissionEdbInfoIdList,
//"Frequency.keyword": "月度",
},
})
}
// noPermissionEdbInfoIdList 无权限指标id
if len(noPermissionEdbClassifyIdList) > 0 {
mustNotMap = append(mustNotMap, map[string]interface{}{
"terms": map[string]interface{}{
"ClassifyId": noPermissionEdbClassifyIdList,
//"Frequency.keyword": "月度",
},
})
}
// collectEdbInfoIdList 收藏的指标id
if len(collectEdbInfoIdList) > 0 {
mustMap = append(mustMap, map[string]interface{}{
"terms": map[string]interface{}{
"EdbInfoId": collectEdbInfoIdList,
},
})
}
// 指标类型:普通指标、预测指标(小于0 代表不区分指标是普通还是预测)
if edbInfoType >= 0 {
mustMap = append(mustMap, map[string]interface{}{
"term": map[string]interface{}{
"EdbInfoType": edbInfoType,
},
})
}
// 指标类型:0-基础+计算;1-基础指标;2-计算指标;3-预测指标
if len(edbTypeList) > 0 {
mustMap = append(mustMap, map[string]interface{}{
"terms": map[string]interface{}{
"EdbType": edbTypeList,
},
})
}
shouldMapList := make([]map[string]interface{}, 0)
// 指标名称、编码匹配
if keywordStr != `` {
// 默认使用中文名字字段去匹配
keywordNameKey := `EdbName`
shouldMap := map[string]interface{}{
"should": []interface{}{
map[string]interface{}{
"match": map[string]interface{}{
"EdbCode": keywordStr,
//"Frequency.keyword": "月度",
},
},
map[string]interface{}{
"match": map[string]interface{}{
keywordNameKey: keywordStr,
//"Frequency.keyword": "月度",
},
},
},
}
shouldMapList = append(shouldMapList, shouldMap)
}
// 指标与用户的权限匹配
if sysUserId > 0 {
shouldTermList := make([]map[string]interface{}, 0)
// 自己的指标
shouldTermList = append(shouldTermList, map[string]interface{}{
"term": map[string]interface{}{
"SysUserId": sysUserId,
},
})
// 分享给我的指标
shouldTermList = append(shouldTermList, map[string]interface{}{
"terms": map[string]interface{}{
"SharedUserIdList": []int{sysUserId},
},
})
// 公开的指标
//shouldTermList = append(shouldTermList,map[string]interface{}{
// "term": map[string]interface{}{
// "PublicStatus": 2,
// },
//})
shouldMap := map[string]interface{}{
"should": shouldTermList,
}
shouldMapList = append(shouldMapList, shouldMap)
}
// 排序
sortList := make([]interface{}, 0)
// 如果没有关键字,那么就走指标id倒序
if keywordStr == `` {
sortEdbInfoId := map[string]interface{}{
"EdbInfoId": map[string]interface{}{
"order": "desc",
},
}
sortList = append(sortList, sortEdbInfoId)
}
return searchEdbInfoData(indexName, mustMap, mustNotMap, shouldMapList, sortList, from, size)
}
// searchEdbInfoData 查询es中的指标数据
func searchEdbInfoData(indexName string, mustMap, mustNotMap []interface{}, shouldMapList []map[string]interface{}, sortList []interface{}, from, size int) (total int64, list []*data_manage.EdbInfoList, err error) {
list = make([]*data_manage.EdbInfoList, 0)
defer func() {
if err != nil {
fmt.Println("EsAddOrEditData Err:", err.Error())
}
}()
client := utils.EsClient
//queryString := elastic.NewQueryStringQuery(keywordStr)
//boolQueryJson, err := json.Marshal(queryString)
//if err != nil {
// fmt.Println("boolQueryJson err:", err)
//} else {
// fmt.Println("boolQueryJson ", string(boolQueryJson))
//}
highlight := elastic.NewHighlight()
highlight = highlight.Fields(elastic.NewHighlighterField("EdbCode"), elastic.NewHighlighterField("EdbName"))
highlight = highlight.PreTags("").PostTags("")
//query := elastic.RawStringQuery(`{"match_all":{}}`)
//关键字匹配
for _, shouldMap := range shouldMapList {
mustMap = append(mustMap, map[string]interface{}{
"bool": shouldMap,
})
}
queryMap := map[string]interface{}{
"query": map[string]interface{}{
"bool": map[string]interface{}{
"must": mustMap,
"must_not": mustNotMap,
//"should": shouldMap,
},
},
}
//根据条件数量统计
requestTotalHits := client.Count(indexName).BodyJson(queryMap)
total, err = requestTotalHits.Do(context.Background())
if err != nil {
return
}
queryMap["from"] = from
queryMap["size"] = size
// 如果有指定排序,那么就按照排序来
if len(sortList) > 0 {
queryMap["sort"] = sortList
}
jsonBytes, _ := json.Marshal(queryMap)
fmt.Println(string(jsonBytes))
//queryString := elastic.NewMatchQuery("EdbCode", keywordStr)
//request := client.Search(indexName).Highlight(highlight).From(from).Size(size).Query(queryString)
request := client.Search(indexName).Highlight(highlight).Source(queryMap) // sets the JSON request
//requestJson, err := json.Marshal(request)
//if err != nil {
// fmt.Println("requestJson err:", err)
//}
//fmt.Println("requestJson ", string(requestJson))
searchMap := make(map[string]string)
searchResp, err := request.Do(context.Background())
if err != nil {
return
}
fmt.Println(searchResp)
fmt.Println(searchResp.Status)
if searchResp.Status != 0 {
return
}
//total = searchResp.TotalHits()
if searchResp.Hits != nil {
for _, v := range searchResp.Hits.Hits {
if _, ok := searchMap[v.Id]; !ok {
itemJson, tmpErr := v.Source.MarshalJSON()
if tmpErr != nil {
err = tmpErr
fmt.Println("movieJson err:", err)
return
}
edbInfoItem := new(data_manage.EdbInfoList)
tmpErr = json.Unmarshal(itemJson, &edbInfoItem)
if tmpErr != nil {
fmt.Println("json.Unmarshal movieJson err:", tmpErr)
err = tmpErr
return
}
if len(v.Highlight["EdbCode"]) > 0 {
edbInfoItem.EdbCode = v.Highlight["EdbCode"][0]
}
if len(v.Highlight["EdbName"]) > 0 {
edbInfoItem.EdbCode = v.Highlight["EdbName"][0]
}
list = append(list, edbInfoItem)
searchMap[v.Id] = v.Id
}
}
}
return
}
// EsDeleteEdbInfoData 删除es中的指标数据
func EsDeleteEdbInfoData(indexName, docId string) (err error) {
defer func() {
if err != nil {
fmt.Println("EsDeleteEdbInfoData Err:", err.Error())
}
}()
client := utils.EsClient
resp, err := client.Delete().Index(indexName).Id(docId).Do(context.Background())
fmt.Println(resp)
if err != nil {
return
}
if resp.Status == 0 {
fmt.Println("删除成功")
} else {
fmt.Println("AddData", resp.Status, resp.Result)
}
return
}
// EsAddOrEditReport 新增编辑es报告
func EsAddOrEditReport(indexName, docId string, item *models.ElasticReportDetail) (err error) {
defer func() {
if err != nil {
fmt.Println("EsAddOrEditReport Err:", err.Error())
}
}()
client := utils.EsClient
// docId为报告ID+章节ID
searchById, err := client.Get().Index(indexName).Id(docId).Do(context.Background())
if err != nil && !strings.Contains(err.Error(), "404") {
fmt.Println("Get Err" + err.Error())
return
}
if searchById != nil && searchById.Found {
resp, err := client.Update().Index(indexName).Id(docId).Doc(map[string]interface{}{
"ReportId": item.ReportId,
"ReportChapterId": item.ReportChapterId,
"Title": item.Title,
"Abstract": item.Abstract,
"BodyContent": item.BodyContent,
"PublishTime": item.PublishTime,
"PublishState": item.PublishState,
"Author": item.Author,
"ClassifyIdFirst": item.ClassifyIdFirst,
"ClassifyNameFirst": item.ClassifyNameFirst,
"ClassifyIdSecond": item.ClassifyIdSecond,
"ClassifyNameSecond": item.ClassifyNameSecond,
"Categories": item.Categories,
"StageStr": item.StageStr,
}).Do(context.Background())
if err != nil {
return err
}
//fmt.Println(resp.Status, resp.Result)
if resp.Status == 0 {
fmt.Println("修改成功" + docId)
err = nil
} else {
fmt.Println("EditData", resp.Status, resp.Result)
}
} else {
resp, err := client.Index().Index(indexName).Id(docId).BodyJson(item).Do(context.Background())
if err != nil {
fmt.Println("新增失败:", err.Error())
return err
}
if resp.Status == 0 && resp.Result == "created" {
fmt.Println("新增成功" + docId)
return nil
} else {
fmt.Println("AddData", resp.Status, resp.Result)
}
}
return
}
// AnalyzeResp 分词接口返回结构体
type AnalyzeResp struct {
Tokens []struct {
EndOffset int64 `json:"end_offset"`
Position int64 `json:"position"`
StartOffset int64 `json:"start_offset"`
Token string `json:"token"`
Type string `json:"type"`
} `json:"tokens"`
}
// Analyze 根据输入的文字获取分词后的文字
func Analyze(content string) (contentList []string, err error) {
defer func() {
if err != nil {
fmt.Println("Analyze Err:", err.Error())
}
}()
client := utils.EsClient
queryMap := map[string]string{
"text": content,
"analyzer": "ik_max_word",
}
res, err := client.PerformRequest(
context.Background(),
elastic.PerformRequestOptions{
Method: "GET",
Path: "/_analyze",
Body: queryMap,
Stream: false,
},
)
if res.StatusCode == 200 {
var analyzeResp AnalyzeResp
tmpErr := json.Unmarshal(res.Body, &analyzeResp)
if tmpErr != nil {
err = errors.New("返回数据转结构体失败:" + tmpErr.Error())
return
}
for _, v := range analyzeResp.Tokens {
contentList = append(contentList, v.Token)
}
} else {
err = errors.New("分词失败,返回code异常:" + strconv.Itoa(res.StatusCode))
}
return
}
// EsAddOrEditChartInfoData 新增/修改es中的图表数据
func EsAddOrEditChartInfoData(indexName, docId string, item *data_manage.ChartInfo) (err error) {
defer func() {
if err != nil {
fmt.Println("EsAddOrEditData Err:", err.Error())
}
}()
client := utils.EsClient
resp, err := client.Index().Index(indexName).Id(docId).BodyJson(item).Do(context.Background())
if err != nil {
fmt.Println("新增失败:", err.Error())
return err
}
fmt.Println(resp)
if resp.Status == 0 {
fmt.Println("新增成功", resp.Result)
err = nil
} else {
fmt.Println("AddData", resp.Status, resp.Result)
}
return
}
// EsDeleteDataV2 删除es中的数据
func EsDeleteDataV2(indexName, docId string) (err error) {
defer func() {
if err != nil {
fmt.Println("EsDeleteEdbInfoData Err:", err.Error())
}
}()
client := utils.EsClient
resp, err := client.Delete().Index(indexName).Id(docId).Do(context.Background())
fmt.Println(resp)
if err != nil {
return
}
if resp.Status == 0 {
fmt.Println("删除成功")
} else {
fmt.Println("AddData", resp.Status, resp.Result)
}
return
}
// SearchChartInfoData 查询es中的图表数据
func SearchChartInfoData(indexName, keywordStr string, showSysId int, sourceList []int, noPermissionChartIdList []int, from, size int) (list []*data_manage.ChartInfo, total int64, err error) {
list = make([]*data_manage.ChartInfo, 0)
defer func() {
if err != nil {
fmt.Println("EsAddOrEditData Err:", err.Error())
}
}()
client := utils.EsClient
//queryString := elastic.NewQueryStringQuery(keywordStr)
//boolQueryJson, err := json.Marshal(queryString)
//if err != nil {
// fmt.Println("boolQueryJson err:", err)
//} else {
// fmt.Println("boolQueryJson ", string(boolQueryJson))
//}
highlight := elastic.NewHighlight()
highlight = highlight.Fields(elastic.NewHighlighterField("ChartName"))
highlight = highlight.PreTags("").PostTags("")
mustMap := make([]interface{}, 0)
mustNotMap := make([]interface{}, 0)
//指标来源
if showSysId > 0 {
mustMap = append(mustMap, map[string]interface{}{
"term": map[string]interface{}{
"SysUserId": showSysId,
//"Frequency.keyword": "月度",
},
})
}
mustMap = append(mustMap, map[string]interface{}{
"terms": map[string]interface{}{
"Source": sourceList,
},
})
//关键字匹配
//shouldMap := map[string]interface{}{
// "should": []interface{}{
// map[string]interface{}{
// "match": map[string]interface{}{
// "ChartName": keywordStr,
// //"Frequency.keyword": "月度",
// },
// },
// // 因为关键词被分了,所以需要用下面的语句来让他 整个词 查询,从而加重整词的权重
// map[string]interface{}{
// "match": map[string]interface{}{
// "ChartName": map[string]interface{}{
// "query": keywordStr,
// "operator": "and",
// },
// //"Frequency.keyword": "月度",
// },
// },
// map[string]interface{}{
// "match": map[string]interface{}{
// "ChartNameEn": keywordStr,
// //"Frequency.keyword": "月度",
// },
// },
// // 因为关键词被分了,所以需要用下面的语句来让他 整个词 查询,从而加重整词的权重
// map[string]interface{}{
// "match": map[string]interface{}{
// "ChartNameEn": map[string]interface{}{
// "query": keywordStr,
// "operator": "and",
// },
// //"Frequency.keyword": "月度",
// },
// },
// },
//}
// 默认使用中文名字字段去匹配
keywordNameKey := `ChartName`
// 如果没有中文,则使用英文名称字段去匹配
if !utils.ContainsChinese(keywordStr) {
keywordNameKey = `ChartNameEn`
}
shouldMap := map[string]interface{}{
"should": []interface{}{
map[string]interface{}{
"match": map[string]interface{}{
keywordNameKey: keywordStr,
//"Frequency.keyword": "月度",
},
},
// 因为关键词被分了,所以需要用下面的语句来让他 整个词 查询,从而加重整词的权重
map[string]interface{}{
"match": map[string]interface{}{
keywordNameKey: map[string]interface{}{
"query": keywordStr,
"operator": "and",
},
//"Frequency.keyword": "月度",
},
},
},
}
mustMap = append(mustMap, map[string]interface{}{
"bool": shouldMap,
})
// noPermissionEdbInfoIdList 无权限指标id
if len(noPermissionChartIdList) > 0 {
mustNotMap = append(mustNotMap, map[string]interface{}{
"terms": map[string]interface{}{
"ChartInfoId": noPermissionChartIdList,
//"Frequency.keyword": "月度",
},
})
}
queryMap := map[string]interface{}{
"query": map[string]interface{}{
"bool": map[string]interface{}{
"must": mustMap,
"must_not": mustNotMap,
//"should": shouldMap,
},
},
}
//根据条件数量统计
requestTotalHits := client.Count(indexName).BodyJson(queryMap)
total, err = requestTotalHits.Do(context.Background())
if err != nil {
return
}
// 分页查询
queryMap["from"] = from
queryMap["size"] = size
jsonBytes, _ := json.Marshal(queryMap)
fmt.Println(string(jsonBytes))
request := client.Search(indexName).Highlight(highlight).Source(queryMap) // sets the JSON request
//requestJson, err := json.Marshal(request)
//if err != nil {
// fmt.Println("requestJson err:", err)
//}
//fmt.Println("requestJson ", string(requestJson))
searchMap := make(map[string]string)
searchResp, err := request.Do(context.Background())
if err != nil {
return
}
fmt.Println(searchResp)
fmt.Println(searchResp.Status)
if searchResp.Status != 0 {
return
}
if searchResp.Hits != nil {
for _, v := range searchResp.Hits.Hits {
if _, ok := searchMap[v.Id]; !ok {
itemJson, tmpErr := v.Source.MarshalJSON()
if tmpErr != nil {
err = tmpErr
fmt.Println("movieJson err:", err)
return
}
chartInfoItem := new(data_manage.ChartInfo)
tmpErr = json.Unmarshal(itemJson, &chartInfoItem)
if err != nil {
fmt.Println("json.Unmarshal chartInfoJson err:", err)
err = tmpErr
return
}
if len(v.Highlight["ChartName"]) > 0 {
chartInfoItem.ChartName = v.Highlight["ChartName"][0]
}
list = append(list, chartInfoItem)
searchMap[v.Id] = v.Id
}
}
}
//for _, v := range result {
// fmt.Println(v)
//}
return
}
// EsAddOrEditDataInterface 新增/修改es中的数据
func EsAddOrEditDataInterface(indexName, docId string, item interface{}) (err error) {
defer func() {
if err != nil {
fmt.Println("EsAddOrEditData Err:", err.Error())
}
}()
client := utils.EsClient
resp, err := client.Index().Index(indexName).Id(docId).BodyJson(item).Do(context.Background())
if err != nil {
fmt.Println("新增失败:", err.Error())
return err
}
fmt.Println(resp)
if resp.Status == 0 {
fmt.Println("新增成功", resp.Result)
err = nil
} else {
fmt.Println("AddData", resp.Status, resp.Result)
}
return
}
// SearchMyChartInfoData 查询es中的我的图表数据
func SearchMyChartInfoData(indexName, keywordStr string, adminId int, noPermissionChartIdList []int, from, size int) (list []*data_manage.MyChartList, total int64, err error) {
list = make([]*data_manage.MyChartList, 0)
defer func() {
if err != nil {
fmt.Println("EsAddOrEditData Err:", err.Error())
}
}()
client := utils.EsClient
//queryString := elastic.NewQueryStringQuery(keywordStr)
//boolQueryJson, err := json.Marshal(queryString)
//if err != nil {
// fmt.Println("boolQueryJson err:", err)
//} else {
// fmt.Println("boolQueryJson ", string(boolQueryJson))
//}
highlight := elastic.NewHighlight()
highlight = highlight.Fields(elastic.NewHighlighterField("ChartName"))
highlight = highlight.PreTags("").PostTags("")
mustMap := make([]interface{}, 0)
mustNotMap := make([]interface{}, 0)
//指标来源
if adminId > 0 {
mustMap = append(mustMap, map[string]interface{}{
"term": map[string]interface{}{
"AdminId": adminId,
//"Frequency.keyword": "月度",
},
})
}
//关键字匹配
//shouldMap := map[string]interface{}{
// "should": []interface{}{
// map[string]interface{}{
// "match": map[string]interface{}{
// "ChartName": keywordStr,
// //"Frequency.keyword": "月度",
// },
// },
// // 因为关键词被分了,所以需要用下面的语句来让他 整个词 查询,从而加重整词的权重
// map[string]interface{}{
// "match": map[string]interface{}{
// "ChartName": map[string]interface{}{
// "query": keywordStr,
// "operator": "and",
// },
// //"Frequency.keyword": "月度",
// },
// },
// map[string]interface{}{
// "match": map[string]interface{}{
// "ChartNameEn": keywordStr,
// //"Frequency.keyword": "月度",
// },
// },
// // 因为关键词被分了,所以需要用下面的语句来让他 整个词 查询,从而加重整词的权重
// map[string]interface{}{
// "match": map[string]interface{}{
// "ChartNameEn": map[string]interface{}{
// "query": keywordStr,
// "operator": "and",
// },
// //"Frequency.keyword": "月度",
// },
// },
// },
//}
// 默认使用中文名字字段去匹配
keywordNameKey := `ChartName`
// 如果没有中文,则使用英文名称字段去匹配
if !utils.ContainsChinese(keywordStr) {
keywordNameKey = `ChartNameEn`
}
shouldMap := map[string]interface{}{
"should": []interface{}{
map[string]interface{}{
"match": map[string]interface{}{
keywordNameKey: keywordStr,
//"Frequency.keyword": "月度",
},
},
// 因为关键词被分了,所以需要用下面的语句来让他 整个词 查询,从而加重整词的权重
map[string]interface{}{
"match": map[string]interface{}{
keywordNameKey: map[string]interface{}{
"query": keywordStr,
"operator": "and",
},
//"Frequency.keyword": "月度",
},
},
},
}
mustMap = append(mustMap, map[string]interface{}{
"bool": shouldMap,
})
// noPermissionEdbInfoIdList 无权限指标id
if len(noPermissionChartIdList) > 0 {
mustNotMap = append(mustNotMap, map[string]interface{}{
"terms": map[string]interface{}{
"ChartInfoId": noPermissionChartIdList,
//"Frequency.keyword": "月度",
},
})
}
queryMap := map[string]interface{}{
"query": map[string]interface{}{
"bool": map[string]interface{}{
"must": mustMap,
"must_not": mustNotMap,
//"should": shouldMap,
},
},
}
//根据条件数量统计
requestTotalHits := client.Count(indexName).BodyJson(queryMap)
total, err = requestTotalHits.Do(context.Background())
if err != nil {
return
}
// 分页查询
queryMap["from"] = from
queryMap["size"] = size
jsonBytes, _ := json.Marshal(queryMap)
fmt.Println(string(jsonBytes))
request := client.Search(indexName).Highlight(highlight).Source(queryMap) // sets the JSON request
//requestJson, err := json.Marshal(request)
//if err != nil {
// fmt.Println("requestJson err:", err)
//}
//fmt.Println("requestJson ", string(requestJson))
searchMap := make(map[string]string)
searchResp, err := request.Do(context.Background())
if err != nil {
return
}
fmt.Println(searchResp)
fmt.Println(searchResp.Status)
if searchResp.Status != 0 {
return
}
if searchResp.Hits != nil {
for _, v := range searchResp.Hits.Hits {
if _, ok := searchMap[v.Id]; !ok {
itemJson, tmpErr := v.Source.MarshalJSON()
if tmpErr != nil {
err = tmpErr
fmt.Println("movieJson err:", err)
return
}
chartInfoItem := new(data_manage.MyChartList)
tmpErr = json.Unmarshal(itemJson, &chartInfoItem)
if err != nil {
fmt.Println("json.Unmarshal chartInfoJson err:", err)
err = tmpErr
return
}
if len(v.Highlight["ChartName"]) > 0 {
chartInfoItem.ChartName = v.Highlight["ChartName"][0]
}
list = append(list, chartInfoItem)
searchMap[v.Id] = v.Id
}
}
}
//for _, v := range result {
// fmt.Println(v)
//}
return
}