|
@@ -3,671 +3,14 @@ package elastic
|
|
|
import (
|
|
|
"context"
|
|
|
"encoding/json"
|
|
|
- "errors"
|
|
|
"eta/eta_forum_hub/models"
|
|
|
"eta/eta_forum_hub/utils"
|
|
|
"fmt"
|
|
|
"github.com/olivere/elastic/v7"
|
|
|
- "strconv"
|
|
|
)
|
|
|
|
|
|
-// 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 *models.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 []int) (total int64, list []*models.EdbInfoList, err error) {
|
|
|
- list = make([]*models.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("<font color='red'>").PostTags("</font>")
|
|
|
-
|
|
|
- //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": "月度",
|
|
|
- },
|
|
|
- })
|
|
|
- }
|
|
|
-
|
|
|
- // 指标类型:普通指标、预测指标(小于0 代表不区分指标是普通还是预测)
|
|
|
- if edbInfoType >= 0 {
|
|
|
- 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": "月度",
|
|
|
- },
|
|
|
- },
|
|
|
- map[string]interface{}{
|
|
|
- "match": map[string]interface{}{
|
|
|
- "EdbNameEn": 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 []*models.EdbInfoList, err error) {
|
|
|
- list = make([]*models.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("<font color='red'>").PostTags("</font>")
|
|
|
-
|
|
|
- //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(models.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 []int, from, size int) (total int64, list []*models.EdbInfoList, err error) {
|
|
|
- list = make([]*models.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("<font color='red'>").PostTags("</font>")
|
|
|
-
|
|
|
- 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{"日度", "周度", "月度"},
|
|
|
- },
|
|
|
- })
|
|
|
-
|
|
|
- //关键字匹配
|
|
|
- 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": "月度",
|
|
|
- },
|
|
|
- },
|
|
|
- },
|
|
|
- }
|
|
|
-
|
|
|
- // noPermissionEdbInfoIdList 无权限指标id
|
|
|
- if len(noPermissionEdbInfoIdList) > 0 {
|
|
|
- mustNotMap = append(mustNotMap, map[string]interface{}{
|
|
|
- "terms": map[string]interface{}{
|
|
|
- "EdbInfoId": noPermissionEdbInfoIdList,
|
|
|
- //"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 []*models.EdbInfoList, err error) {
|
|
|
- list = make([]*models.EdbInfoList, 0)
|
|
|
+func EsAddOrEditEdbInfoData(indexName, docId string, item *models.EdbInfoList) (err error) {
|
|
|
defer func() {
|
|
|
if err != nil {
|
|
|
fmt.Println("EsAddOrEditData Err:", err.Error())
|
|
@@ -675,109 +18,18 @@ func searchEdbInfoData(indexName string, mustMap, mustNotMap []interface{}, shou
|
|
|
}()
|
|
|
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("<font color='red'>").PostTags("</font>")
|
|
|
-
|
|
|
- //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())
|
|
|
+ resp, err := client.Index().Index(indexName).Id(docId).BodyJson(item).Do(context.Background())
|
|
|
if err != nil {
|
|
|
- return
|
|
|
- }
|
|
|
- fmt.Println(searchResp)
|
|
|
- fmt.Println(searchResp.Status)
|
|
|
- if searchResp.Status != 0 {
|
|
|
- return
|
|
|
+ fmt.Println("新增失败:", err.Error())
|
|
|
+ return err
|
|
|
}
|
|
|
- //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(models.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
|
|
|
- }
|
|
|
- }
|
|
|
+ fmt.Println(resp)
|
|
|
+ if resp.Status == 0 {
|
|
|
+ fmt.Println("新增成功", resp.Result)
|
|
|
+ err = nil
|
|
|
+ } else {
|
|
|
+ fmt.Println("AddData", resp.Status, resp.Result)
|
|
|
}
|
|
|
-
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -803,55 +55,6 @@ func EsDeleteEdbInfoData(indexName, docId string) (err error) {
|
|
|
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 *models.ChartInfo) (err error) {
|
|
|
defer func() {
|
|
@@ -899,7 +102,7 @@ func EsDeleteDataV2(indexName, docId string) (err error) {
|
|
|
}
|
|
|
|
|
|
// SearchChartInfoData 查询es中的图表数据
|
|
|
-func SearchChartInfoData(indexName, keywordStr string, showSysId int, sourceList []int, noPermissionChartIdList []int, from, size int) (list []*models.ChartInfo, total int64, err error) {
|
|
|
+func SearchChartInfoData(indexName, keywordStr string, showSysId []int, sourceList []int, permissionClassifyIdList []int, from, size int) (list []*models.ChartInfo, total int64, err error) {
|
|
|
list = make([]*models.ChartInfo, 0)
|
|
|
defer func() {
|
|
|
if err != nil {
|
|
@@ -924,9 +127,9 @@ func SearchChartInfoData(indexName, keywordStr string, showSysId int, sourceList
|
|
|
mustNotMap := make([]interface{}, 0)
|
|
|
|
|
|
//指标来源
|
|
|
- if showSysId > 0 {
|
|
|
+ if len(showSysId) > 0 {
|
|
|
mustMap = append(mustMap, map[string]interface{}{
|
|
|
- "term": map[string]interface{}{
|
|
|
+ "terms": map[string]interface{}{
|
|
|
"SysUserId": showSysId,
|
|
|
//"Frequency.keyword": "月度",
|
|
|
},
|
|
@@ -937,36 +140,71 @@ func SearchChartInfoData(indexName, keywordStr string, showSysId int, sourceList
|
|
|
"Source": sourceList,
|
|
|
},
|
|
|
})
|
|
|
-
|
|
|
+ // permissionClassifyIdList 只在特定的分类ID下的图表
|
|
|
+ if len(permissionClassifyIdList) > 0 {
|
|
|
+ mustMap = append(mustMap, map[string]interface{}{
|
|
|
+ "terms": map[string]interface{}{
|
|
|
+ "ChartClassifyId": permissionClassifyIdList,
|
|
|
+ //"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{}{
|
|
|
- "ChartName": keywordStr,
|
|
|
- //"Frequency.keyword": "月度",
|
|
|
- },
|
|
|
- },
|
|
|
- // 因为关键词被分了,所以需要用下面的语句来让他 整个词 查询,从而加重整词的权重
|
|
|
- map[string]interface{}{
|
|
|
- "match": map[string]interface{}{
|
|
|
- "ChartName": map[string]interface{}{
|
|
|
- "query": keywordStr,
|
|
|
- "operator": "and",
|
|
|
- },
|
|
|
- //"Frequency.keyword": "月度",
|
|
|
- },
|
|
|
- },
|
|
|
- map[string]interface{}{
|
|
|
+ /*map[string]interface{}{
|
|
|
"match": map[string]interface{}{
|
|
|
- "ChartNameEn": keywordStr,
|
|
|
+ keywordNameKey: keywordStr,
|
|
|
//"Frequency.keyword": "月度",
|
|
|
},
|
|
|
- },
|
|
|
+ },*/
|
|
|
// 因为关键词被分了,所以需要用下面的语句来让他 整个词 查询,从而加重整词的权重
|
|
|
map[string]interface{}{
|
|
|
"match": map[string]interface{}{
|
|
|
- "ChartNameEn": map[string]interface{}{
|
|
|
+ keywordNameKey: map[string]interface{}{
|
|
|
"query": keywordStr,
|
|
|
"operator": "and",
|
|
|
},
|
|
@@ -975,20 +213,11 @@ func SearchChartInfoData(indexName, keywordStr string, showSysId int, sourceList
|
|
|
},
|
|
|
},
|
|
|
}
|
|
|
+
|
|
|
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{}{
|
|
@@ -1061,227 +290,3 @@ func SearchChartInfoData(indexName, keywordStr string, showSysId int, sourceList
|
|
|
//}
|
|
|
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
|
|
|
-}
|
|
|
-
|
|
|
-// SearchEdbInfoDataByAdminId 查询es中的指标数据
|
|
|
-func SearchEdbInfoDataByAdminId(indexName, keywordStr string, from, size, filterSource, source int, edbInfoType uint8, frequency string, adminId int) (total int64, list []*models.EdbInfoList, err error) {
|
|
|
- list = make([]*models.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("<font color='red'>").PostTags("</font>")
|
|
|
-
|
|
|
- //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": "月度",
|
|
|
- },
|
|
|
- },
|
|
|
- },
|
|
|
- }
|
|
|
- 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)
|
|
|
-}
|