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(indexName, keywordStr string, from, size, filterSource, source int, edbInfoType int8, frequency string, noPermissionEdbInfoIdList, noPermissionEdbClassifyIdList []int, edbType 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()) } }() highlight := elastic.NewHighlight() highlight = highlight.Fields(elastic.NewHighlighterField("EdbCode"), elastic.NewHighlighterField("EdbName")) highlight = highlight.PreTags("").PostTags("") //var source map[string]interface{} //source := map[string]interface{}{ // "query": map[string]interface{}{ // "match_all": map[string]interface{}{}, // }, //} mustMap := make([]interface{}, 0) mustNotMap := make([]interface{}, 0) //source := map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // "fields": []string{"EdbCode", "EdbName"}, // }, // }, // }, // }, //} switch filterSource { case 2: //source = map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // }, // }, // "filter": []interface{}{ // map[string]interface{}{ // "term": map[string]interface{}{ // "Frequency.keyword": "月度", // }, // }}, // }, // }, //} mustMap = []interface{}{ map[string]interface{}{ "term": map[string]interface{}{ "Frequency.keyword": "月度", //"Frequency.keyword": "月度", }, }, } case 3: //source = map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // }, // }, // "must_not": []interface{}{ // map[string]interface{}{ // "match": map[string]interface{}{ // "Frequency.keyword": "日度", // }, // }}, // }, // }, //} ////注释掉,所有频度都可以变频 2022-08-31 14:31:28 //mustNotMap = []interface{}{ // map[string]interface{}{ // "match": map[string]interface{}{ // "Frequency.keyword": "日度", // //"Frequency.keyword": "月度", // }, // }, //} case 4: //source = map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // }, // }, // "filter": []interface{}{ // map[string]interface{}{ // "term": map[string]interface{}{ // "EdbType": 1, // }, // }}, // }, // }, //} mustMap = []interface{}{ map[string]interface{}{ "term": map[string]interface{}{ "EdbType": 1, }, }, } case 5: mustMap = []interface{}{ map[string]interface{}{ "term": map[string]interface{}{ "Source": 6, }, }, } case 6: mustNotMap = []interface{}{ 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": "月度", }, }) } // 指标类型:普通指标、预测指标(小于0 代表不区分指标是普通还是预测) if edbInfoType >= 0 { mustMap = append(mustMap, map[string]interface{}{ "term": map[string]interface{}{ "EdbInfoType": edbInfoType, }, }) } // 指标类型:0-基础+计算;1-基础指标;2-计算指标 if edbType > 0 { mustMap = append(mustMap, map[string]interface{}{ "term": map[string]interface{}{ "EdbType": edbType, }, }) } //普通指标 //mustMap = append(mustMap, map[string]interface{}{ // "term": map[string]interface{}{ // "EdbInfoType": 0, // //"Frequency.keyword": "月度", // }, //}) //关键字匹配 //shouldMap := map[string]interface{}{ // "should": []interface{}{ // map[string]interface{}{ // "match": map[string]interface{}{ // "EdbCode": keywordStr, // //"Frequency.keyword": "月度", // }, // }, // map[string]interface{}{ // "match": map[string]interface{}{ // "EdbName": keywordStr, // //"Frequency.keyword": "月度", // }, // }, // map[string]interface{}{ // "match": map[string]interface{}{ // "EdbNameEn": keywordStr, // //"Frequency.keyword": "月度", // }, // }, // }, //} // 默认使用中文名字字段去匹配 keywordNameKey := `EdbName` // 如果没有中文,则使用英文名称字段去匹配 if !utils.ContainsChinese(keywordStr) { keywordNameKey = `EdbNameEn` } 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": "月度", }, }, }, } //mustMap = append(mustMap, map[string]interface{}{ // "bool": shouldMap, //}) return searchEdbInfoData(indexName, mustMap, mustNotMap, shouldMap, from, size) } func SearchEdbInfoDataBak(indexName, keywordStr string, from, size, filterSource, source int, frequency string) (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()) } for _, v := range list { v.ConvertToResp() } }() 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":{}}`) //var source map[string]interface{} //source := map[string]interface{}{ // "query": map[string]interface{}{ // "match_all": map[string]interface{}{}, // }, //} mustMap := make([]interface{}, 0) mustNotMap := make([]interface{}, 0) //source := map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // "fields": []string{"EdbCode", "EdbName"}, // }, // }, // }, // }, //} switch filterSource { case 2: //source = map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // }, // }, // "filter": []interface{}{ // map[string]interface{}{ // "term": map[string]interface{}{ // "Frequency.keyword": "月度", // }, // }}, // }, // }, //} mustMap = []interface{}{ map[string]interface{}{ "term": map[string]interface{}{ "Frequency.keyword": "月度", //"Frequency.keyword": "月度", }, }, } case 3: //source = map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // }, // }, // "must_not": []interface{}{ // map[string]interface{}{ // "match": map[string]interface{}{ // "Frequency.keyword": "日度", // }, // }}, // }, // }, //} ////注释掉,所有频度都可以变频 2022-08-31 14:31:28 //mustNotMap = []interface{}{ // map[string]interface{}{ // "match": map[string]interface{}{ // "Frequency.keyword": "日度", // //"Frequency.keyword": "月度", // }, // }, //} case 4: //source = map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // }, // }, // "filter": []interface{}{ // map[string]interface{}{ // "term": map[string]interface{}{ // "EdbType": 1, // }, // }}, // }, // }, //} mustMap = []interface{}{ map[string]interface{}{ "term": map[string]interface{}{ "EdbType": 1, }, }, } case 5: mustMap = []interface{}{ map[string]interface{}{ "term": map[string]interface{}{ "Source": 6, }, }, } case 6: mustNotMap = []interface{}{ 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": "月度", }, }) } //普通指标 //mustMap = append(mustMap, map[string]interface{}{ // "term": map[string]interface{}{ // "EdbInfoType": 0, // //"Frequency.keyword": "月度", // }, //}) //关键字匹配 shouldMap := map[string]interface{}{ "should": []interface{}{ map[string]interface{}{ "match": map[string]interface{}{ "EdbCode": keywordStr, //"Frequency.keyword": "月度", }, }, map[string]interface{}{ "match": map[string]interface{}{ "EdbName": keywordStr, //"Frequency.keyword": "月度", }, }, // 因为关键词被分了,所以需要用下面的语句来让他 整个词 查询,从而加重整词的权重 map[string]interface{}{ "match": map[string]interface{}{ "EdbCode": map[string]interface{}{ "query": keywordStr, "operator": "and", }, //"Frequency.keyword": "月度", }, }, // 因为关键词被分了,所以需要用下面的语句来让他 整个词 查询,从而加重整词的权重 map[string]interface{}{ "match": map[string]interface{}{ "EdbName": map[string]interface{}{ "query": keywordStr, "operator": "and", }, //"Frequency.keyword": "月度", }, }, }, } 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 jsonBytes, _ := json.Marshal(queryMap) fmt.Println(string(jsonBytes)) //queryStr := fmt.Sprintf(`{"query":{"bool":{"must":{"query_string":{"query":"%s","fields":["EdbCode","EdbName"]}}}}}`, keywordStr) //switch filterSource { //case 2: // queryStr = fmt.Sprintf(`{"query":{"bool":{"must":{"query_string":{"query":"%s","fields":["EdbCode","EdbName"]}},"filter":{"term":{"Frequency.keyword":"%s"}}}}}`, keywordStr, "月度") //case 3: // queryStr = fmt.Sprintf(`{"query":{"bool":{"must":{"query_string":{"query":"%s","fields":["EdbCode","EdbName"]}},"must_not":[{"match":{"Frequency.keyword":"%s"}}]}}}`, keywordStr, "日度") //case 4: // queryStr = fmt.Sprintf(`{"query":{"bool":{"must":{"query_string":{"query":"%s","fields":["EdbCode","EdbName"]}},"must_not":[{"match":{"EdbType":1}}]}}}`, keywordStr) //} //queryString := elastic.RawStringQuery(queryStr) //fmt.Println("queryString:", queryString) //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 } 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 } } } //for _, v := range result { // fmt.Println(v) //} return } // SearchAddPredictEdbInfoData 查询允许添加预测指标的数据 func SearchAddPredictEdbInfoData(indexName, keywordStr string, noPermissionEdbInfoIdList, noPermissionEdbClassifyIdList []int, from, size, edbType 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()) } for _, v := range list { v.ConvertToResp() } }() highlight := elastic.NewHighlight() highlight = highlight.Fields(elastic.NewHighlighterField("EdbCode"), elastic.NewHighlighterField("EdbName")) highlight = highlight.PreTags("").PostTags("") mustMap := make([]interface{}, 0) mustNotMap := make([]interface{}, 0) mustNotMap = []interface{}{ //map[string]interface{}{ // "terms": map[string]interface{}{ // "Frequency.keyword": []string{"日度", "周度", "月度"}, // }, // //"match": map[string]interface{}{ // // "Frequency": []string{"日度", "周度", "月度"}, // // //"Frequency.keyword": []string{"日度", "周度", "月度"}, // //}, //}, } // 指标类型:普通指标、预算指标 mustMap = append(mustMap, map[string]interface{}{ "term": map[string]interface{}{ "EdbInfoType": 0, }, }) mustMap = append(mustMap, map[string]interface{}{ "terms": map[string]interface{}{ "Frequency.keyword": []string{"日度", "周度", "月度"}, }, }) // 指标类型: 1-基础指标; 2-计算指标 if edbType > 0 { mustMap = append(mustMap, map[string]interface{}{ "term": map[string]interface{}{ "EdbType": edbType, }, }) } //关键字匹配 //shouldMap := map[string]interface{}{ // "should": []interface{}{ // map[string]interface{}{ // "match": map[string]interface{}{ // "EdbCode": keywordStr, // //"Frequency.keyword": "月度", // }, // }, // map[string]interface{}{ // "match": map[string]interface{}{ // "EdbName": keywordStr, // //"Frequency.keyword": "月度", // }, // }, // map[string]interface{}{ // "match": map[string]interface{}{ // "EdbNameEn": keywordStr, // //"Frequency.keyword": "月度", // }, // }, // }, //} // 默认使用中文名字字段去匹配 keywordNameKey := `EdbName` // 如果没有中文,则使用英文名称字段去匹配 if !utils.ContainsChinese(keywordStr) { keywordNameKey = `EdbNameEn` } 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": "月度", }, }, }, } // noPermissionEdbInfoIdList 无权限指标id if len(noPermissionEdbInfoIdList) > 0 { mustNotMap = append(mustNotMap, map[string]interface{}{ "terms": map[string]interface{}{ "EdbInfoId": noPermissionEdbInfoIdList, //"Frequency.keyword": "月度", }, }) } // noPermissionEdbClassifyIdList 无权限指标分类id if len(noPermissionEdbClassifyIdList) > 0 { mustNotMap = append(mustNotMap, map[string]interface{}{ "terms": map[string]interface{}{ "ClassifyId": noPermissionEdbClassifyIdList, //"Frequency.keyword": "月度", }, }) } return searchEdbInfoData(indexName, mustMap, mustNotMap, shouldMap, from, size) } // searchEdbInfoData 查询es中的指标数据 func searchEdbInfoData(indexName string, mustMap, mustNotMap []interface{}, shouldMap map[string]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":{}}`) //关键字匹配 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 jsonBytes, _ := json.Marshal(queryMap) fmt.Println(string(jsonBytes)) //queryStr := fmt.Sprintf(`{"query":{"bool":{"must":{"query_string":{"query":"%s","fields":["EdbCode","EdbName"]}}}}}`, keywordStr) //switch filterSource { //case 2: // queryStr = fmt.Sprintf(`{"query":{"bool":{"must":{"query_string":{"query":"%s","fields":["EdbCode","EdbName"]}},"filter":{"term":{"Frequency.keyword":"%s"}}}}}`, keywordStr, "月度") //case 3: // queryStr = fmt.Sprintf(`{"query":{"bool":{"must":{"query_string":{"query":"%s","fields":["EdbCode","EdbName"]}},"must_not":[{"match":{"Frequency.keyword":"%s"}}]}}}`, keywordStr, "日度") //case 4: // queryStr = fmt.Sprintf(`{"query":{"bool":{"must":{"query_string":{"query":"%s","fields":["EdbCode","EdbName"]}},"must_not":[{"match":{"EdbType":1}}]}}}`, keywordStr) //} //queryString := elastic.RawStringQuery(queryStr) //fmt.Println("queryString:", queryString) //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 } // SearchEdbInfoDataByAdminId 查询es中的指标数据 func SearchEdbInfoDataByAdminId(indexName, keywordStr string, from, size, filterSource, source int, edbInfoType uint8, frequency string, adminId 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()) } for _, v := range list { v.ConvertToResp() } }() highlight := elastic.NewHighlight() highlight = highlight.Fields(elastic.NewHighlighterField("EdbCode"), elastic.NewHighlighterField("EdbName")) highlight = highlight.PreTags("").PostTags("") //var source map[string]interface{} //source := map[string]interface{}{ // "query": map[string]interface{}{ // "match_all": map[string]interface{}{}, // }, //} mustMap := make([]interface{}, 0) mustNotMap := make([]interface{}, 0) //source := map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // "fields": []string{"EdbCode", "EdbName"}, // }, // }, // }, // }, //} switch filterSource { case 2: //source = map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // }, // }, // "filter": []interface{}{ // map[string]interface{}{ // "term": map[string]interface{}{ // "Frequency.keyword": "月度", // }, // }}, // }, // }, //} mustMap = []interface{}{ map[string]interface{}{ "term": map[string]interface{}{ "Frequency.keyword": "月度", //"Frequency.keyword": "月度", }, }, } case 3: //source = map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // }, // }, // "must_not": []interface{}{ // map[string]interface{}{ // "match": map[string]interface{}{ // "Frequency.keyword": "日度", // }, // }}, // }, // }, //} ////注释掉,所有频度都可以变频 2022-08-31 14:31:28 //mustNotMap = []interface{}{ // map[string]interface{}{ // "match": map[string]interface{}{ // "Frequency.keyword": "日度", // //"Frequency.keyword": "月度", // }, // }, //} case 4: //source = map[string]interface{}{ // "query": map[string]interface{}{ // "bool": map[string]interface{}{ // "must": map[string]interface{}{ // "query_string": map[string]interface{}{ // "query": keywordStr, // }, // }, // "filter": []interface{}{ // map[string]interface{}{ // "term": map[string]interface{}{ // "EdbType": 1, // }, // }}, // }, // }, //} mustMap = []interface{}{ map[string]interface{}{ "term": map[string]interface{}{ "EdbType": 1, }, }, } case 5: mustMap = []interface{}{ map[string]interface{}{ "term": map[string]interface{}{ "Source": 6, }, }, } case 6: mustNotMap = []interface{}{ 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": "月度", }, }) } // 指标类型:普通指标、预算指标 mustMap = append(mustMap, map[string]interface{}{ "term": map[string]interface{}{ "EdbInfoType": edbInfoType, }, }) //普通指标 //mustMap = append(mustMap, map[string]interface{}{ // "term": map[string]interface{}{ // "EdbInfoType": 0, // //"Frequency.keyword": "月度", // }, //}) //关键字匹配 //shouldMap := map[string]interface{}{ // "should": []interface{}{ // map[string]interface{}{ // "match": map[string]interface{}{ // "EdbCode": keywordStr, // //"Frequency.keyword": "月度", // }, // }, // map[string]interface{}{ // "match": map[string]interface{}{ // "EdbName": keywordStr, // //"Frequency.keyword": "月度", // }, // }, // }, //} // 默认使用中文名字字段去匹配 keywordNameKey := `EdbName` // 如果没有中文,则使用英文名称字段去匹配 if !utils.ContainsChinese(keywordStr) { keywordNameKey = `EdbNameEn` } 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": "月度", }, }, }, } mustMap = append(mustMap, map[string]interface{}{ "bool": shouldMap, }) //创建人 if adminId > 0 { mustMap = append(mustMap, map[string]interface{}{ "term": map[string]interface{}{ "SysUserId": adminId, }, }) } return searchEdbInfoData(indexName, mustMap, mustNotMap, shouldMap, from, size) }