ES7學習筆記（十二）高亮 和 搜索建議

ES當中大部分的內容都已經學習完了，今天呢算是對前面內容的查漏補缺，把ES中非常實用的功能整理一下，在以后的項目開發中，這些功能肯定是對你的項目加分的，我們來看看吧。

高亮

高亮在搜索功能中是十分重要的，我們希望搜索的內容在搜索結果中重點突出，讓用戶聚焦在搜索的內容上。我們看看在ES當中是怎么實現高亮的，我們還用之前的索引ik_index，前面的章節，我們搜索過香蕉好吃，但是返回的結果中並沒有高亮，那么想要在搜索結果中，對香蕉好吃高亮該怎么辦呢？我們看看，

POST /ik_index/_search
{
  "query": {
    "bool": {
      "must": {
        "match": {
          "desc": "香蕉好吃"
        }
      }
    }
  },
  "highlight": {
    "fields": {
      "desc": {}
    }
  }
}

我們重點看一下請求體中的highlight部分，這部分就是對返回結果高亮的設置，fields字段中，指定哪些字段需要高亮，我們指定了desc字段，執行一下，看看結果吧。

{
    "took": 73,
    "timed_out": false,
    "_shards": { "total": 1,"successful": 1,"skipped": 0,"failed": 0},
    "hits": {
        "total": {
            "value": 5,
            "relation": "eq"
        },
        "max_score": 1.3948275,
        "hits": [
            {
            "_index": "ik_index",
            "_type": "_doc",
            "_id": "2",
            "_score": 1.3948275,
            "_source": {
                "id": 1,
                "title": "香蕉",
                "desc": "香蕉真好吃"
                },
            "highlight": {
                "desc": [
                        "<em>香蕉</em>真<em>好吃</em>"
                    ]
                }
            }
        ……

我們看到在返回的結果中，增加了highlight，highlight里有我們指定的高亮字段desc，它的值是<em>香蕉</em>真<em>好吃</em>，其中“香蕉”和“好吃”字段在<em>標簽中，前端的小伙伴就可以針對這個<em>標簽寫樣式了。我們再看看程序當中怎么設置高亮，繼續使用上一節中的搜索的程序，

    public void searchIndex() throws IOException {
        SearchRequest searchRequest = new SearchRequest("ik_index");
        SearchSourceBuilder ssb = new SearchSourceBuilder();
        QueryBuilder qb = new MatchQueryBuilder("desc","香蕉好吃");
        ssb.query(qb);
        
        HighlightBuilder highlightBuilder = new HighlightBuilder();
        highlightBuilder.field("desc");

        ssb.highlighter(highlightBuilder);

        searchRequest.source(ssb);
        SearchResponse response = client.search(searchRequest, RequestOptions.DEFAULT);

        SearchHit[] hits = response.getHits().getHits();
        for (SearchHit hit : hits) {
            String record = hit.getSourceAsString();
            HighlightField highlightField = hit.getHighlightFields().get("desc");

            for (Text fragment : highlightField.getFragments()) {
                System.out.println(fragment.string());
            }
        }
    }

我們重點關注一下HighlightBuilder，我們在發送請求前，創建HighlightBuilder，並指定高亮字段為desc。搜索結束后，我們取結果，從hit當中取出高亮字段desc，然后打印出fragment，運行一下，看看結果吧，

<em>香蕉</em>真<em>好吃</em>
<em>香蕉</em>真<em>好吃</em>
橘子真<em>好吃</em>
桃子真<em>好吃</em>
蘋果真<em>好吃</em>

完全符合預期，“香蕉好吃”被分詞后，在搜索結果中都增加了<em>標簽，我們可不可以自定義高亮標簽呢？當然是可以的，我們稍微改一下程序就可以了，

HighlightBuilder highlightBuilder = new HighlightBuilder();
highlightBuilder.field("desc");

highlightBuilder.preTags("<b>");
highlightBuilder.postTags("</b>");

ssb.highlighter(highlightBuilder);

在HighlightBuilder中，使用preTags添加起始標簽，指定為<b>，用postTags添加閉合標簽，指定為</b>，再運行一下，看看結果，

<b>香蕉</b>真<b>好吃</b>
<b>香蕉</b>真<b>好吃</b>
橘子真<b>好吃</b>
桃子真<b>好吃</b>
蘋果真<b>好吃</b>

結果完全正確，用<b>替換了<em>，是不是很靈活。接下來我們再看看搜索建議。

搜索建議

“搜索建議”這個功能也是相當實用的，當我們在搜索框中輸入某個字時，與這個字的相關搜索內容就會羅列在下面，我們選擇其中一個搜索就可以了，省去了敲其他字的時間。我們看看ES中是怎么實現“搜索建議”的。

如果要在ES中使用“搜索建議”功能，是需要特殊設置的，要設置一個類型為completion的字段，由於之前的索引中已經有了數據，再添加字段是會報錯的，索引我們新建一個索引，

PUT /my_suggester
{
    "settings":{
        "analysis":{
            "analyzer":{
                "default":{
                    "type":"ik_max_word"
                }
            }
        }
    },
    "mappings":{
        "dynamic_date_formats": [
             "MM/dd/yyyy",
             "yyyy/MM/dd HH:mm:ss",
             "yyyy-MM-dd",
             "yyyy-MM-dd HH:mm:ss"
         ],
        "properties":{
            "suggest":{
                "type":"completion"
            }
        }
    }
}

這已經成了我們新建索引的一個標配了，指定分詞器為ik中文分詞，動態字段的時間映射格式，以及搜索建議字段，注意suggest字段的類型為completion。我們再添加字段的時候，就要為suggest字段添加值了，如下：

POST /my_suggester/_doc
{
    "title":"天氣",
    "desc":"今天天氣不錯",
    "suggest": {
       "input": "天氣"
    }
}

POST /my_suggester/_doc
{
    "title":"天空",
    "desc":"藍藍的天空，白白的雲",
    "suggest": {
       "input": "天空"
    }
}

我們向索引中添加了兩條數據，大家需要額外注意的是suggest字段的賦值方法，要使用input，我們看一下數據，

suggest字段並沒有像其他字段那樣展示出來，說明它和其他字段是不一樣的。現在我們如果只輸入一個“天”字，看看搜索建議能不能給出提示，如下：

POST /my_suggester/_search
{
  "suggest": {
    "s-test": {
      "prefix": "天",
      "completion": {
        "field": "suggest"
      }
    }
  }
}

在請求體中，suggest就是“搜索建議”的標識，s-test是自定義的一個名稱，prefix是前綴，也就是我們輸入的“天”字，completion指定搜索建議的字段，我們看看查詢的結果，

……
"suggest": {
    "s-test": [
        {
            "text": "天",
            "offset": 0,
            "length": 1,
            "options": [{
                "text": "天氣",
                "_index": "my_suggester",
                "_type": "_doc",
                "_id": "QtgAWnIBOZNtuLQtJgpt",
                "_score": 1,
                "_source": { "title": "天氣","desc": "今天天氣不錯","suggest": { "input": "天氣"}}
        	}
        	,
            {
                "text": "天空",
                "_index": "my_suggester",
                "_type": "_doc",
                "_id": "T9gAWnIBOZNtuLQtWQoX",
                "_score": 1,
                "_source": { "title": "天空","desc": "藍藍的天空，白白的雲","suggest": { "input": "天空"}}
            }
        ]
        }
    ]
}

在s-test.options里，包含了兩條記錄，text字段就是我們寫的建議字段，后面_source里還包含對應的數據，下面我們再看看程序里怎么使用“搜索建議”，

public void searchSuggest(String prefix) throws IOException {
    SearchRequest searchRequest = new SearchRequest("my_suggester");
    SearchSourceBuilder ssb = new SearchSourceBuilder();

    CompletionSuggestionBuilder suggest = SuggestBuilders
        .completionSuggestion("suggest")
        .prefix(prefix);

    SuggestBuilder suggestBuilder = new SuggestBuilder();
    suggestBuilder.addSuggestion("s-test",suggest);

    ssb.suggest(suggestBuilder);

    searchRequest.source(ssb);
    SearchResponse response = client.search(searchRequest, RequestOptions.DEFAULT);

    CompletionSuggestion suggestion = response.getSuggest().getSuggestion("s-test");
    for (CompletionSuggestion.Entry.Option option : suggestion.getOptions()) {
        System.out.println(option.getText().string());
    }

}

@Test
public void searchSuggest() throws IOException {
    eService.searchSuggest("天");
}

我們創建了CompletionSuggestionBuilder，通過方法completionSuggestion指定“搜索建議”字段suggest，並且指定前綴為方法傳入的prefix，我們在測試的時候傳入"天"字。然后，我們自定義“搜索建議”的名字為s-test，傳入前面構造好的suggest。

發送請求后，在響應中獲取前面自定義的s-test，然后循環options，取出text字段，這就是搜索建議的字段，我們運行一下，看看結果，

天氣
天空

完全符合預期，這樣用戶在搜索的時候，就會給出提示信息了。

好了，今天這兩個ES的知識點就全部OK了~ 大家有問題在評論區留言。