快捷導(dǎo)航

springboot整合solr的方法詳解

更新時(shí)間：2019年08月19日 16:06:43 作者：Leon_W

這篇文章主要介紹了springboot整合solr的方法詳解,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下

這一篇寫一下springboot整合solr，代碼已經(jīng)上傳到github，傳送門。

1、新建core并配置schema

solr create -c "book_core" ，配置分詞器并且field類型定義為分詞器類型。

<fieldType name="ik_word" class="solr.TextField">
   <analyzer type="index">
     <tokenizer class="org.wltea.analyzer.lucene.IKTokenizerFactory" useSmart="false" conf="ik.conf"/>
     <filter class="solr.LowerCaseFilterFactory"/>
   </analyzer>
   <analyzer type="query">
     <tokenizer class="org.wltea.analyzer.lucene.IKTokenizerFactory" useSmart="true" conf="ik.conf"/>
     <filter class="solr.LowerCaseFilterFactory"/>
   </analyzer>
  </fieldType>

   <!-- 自定義添加filed -->
  <field name="description" type="ik_word" indexed="true" stored="true"/>

配置完畢后啟動(dòng)solr。

2、相關(guān)配置

配置maven

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-data-solr</artifactId>
</dependency>

配置application.properties相關(guān)solr的內(nèi)容

# solr配置
spring.data.solr.host=http://localhost:8983/solr/book_core

如果想同時(shí)使用好幾個(gè)core，通過(guò)HttpSolrClient solrClient = new HttpSolrClient(url);構(gòu)造出對(duì)應(yīng)core的solrClient進(jìn)行相關(guān)操作。

3、solr增刪改查

3.1、新增實(shí)體

@SolrDocument(solrCoreName = "book_core")
public class Book {

  @Id
  @Field
  private String id;

  @Field
  private String description;

  public String getId() {
    return id;
  }

  public void setId(String id) {
    this.id = id;
  }

  public String getDescription() {
    return description;
  }

  public void setDescription(String description) {
    this.description = description;
  }
}

在這里要說(shuō)明一下實(shí)體類上的注解@field和@SolrDocument 是為了QueryResponse.getBeans()能轉(zhuǎn)化成相關(guān)實(shí)體類，不然無(wú)法識(shí)別。

3.2增刪改

@Autowired
  SolrClient solrClient;

  @Override
  public void add(Book book) {
    SolrInputDocument document = new SolrInputDocument();
    document.setField("id",book.getId());
    document.setField("description",book.getDescription());
    try {
      solrClient.add(document);
      solrClient.commit();
    } catch (SolrServerException e) {
      e.printStackTrace();
    } catch (IOException e) {
      e.printStackTrace();
    }
  }

  @Override
  public void delete(String query) {
    try {
      solrClient.deleteByQuery(query);
      solrClient.commit();
    } catch (SolrServerException e) {
      e.printStackTrace();
    } catch (IOException e) {
      e.printStackTrace();
    }
  }

  @Override
  public Book update(Book book) {
    try {
      solrClient.addBean(book);
      solrClient.commit();
    } catch (IOException e) {
      e.printStackTrace();
    } catch (SolrServerException e) {
      e.printStackTrace();
    }
    return book;
  }

增加刪除修改都差不多，增加和修改可以通過(guò)add（SolrInputDocument）和addBean，根據(jù)主鍵是否存在來(lái)判斷是新增還是修改，刪除可以通過(guò)主鍵或者條件進(jìn)行刪除。增刪改之后一定要記得commit!

主鍵是在schema中配置的，默認(rèn)有 id字段做主鍵，也可以自定義主鍵，在schema中配置。

通過(guò)add方法加了幾個(gè)demo后，可以在solr管理頁(yè)面的查詢出查看添加的內(nèi)容

3.3查詢

查詢則通過(guò)SolrQuery入?yún)⑦M(jìn)行查詢

@Override
  public List<Book> queryAll() {
    List<Book> bookList = new ArrayList<Book>();
    SolrQuery solrQuery = new SolrQuery();
    solrQuery.setQuery("*:*");
    try {
      QueryResponse queryResponse = solrClient.query(solrQuery);
      if (queryResponse != null){
        bookList = queryResponse.getBeans(Book.class);
      }
    } catch (SolrServerException e) {
      e.printStackTrace();
    } catch (IOException e) {
      e.printStackTrace();
    }
    return bookList;

同時(shí)我寫了一個(gè)jsp頁(yè)面來(lái)展示查詢結(jié)果。

到此，增刪改查的基本操作都已經(jīng)實(shí)現(xiàn)了，更多高級(jí)用法可以參考官方api文檔。

4、高亮

一開(kāi)始在嘗試使用高亮?xí)r發(fā)現(xiàn)無(wú)法起作用，排查后發(fā)現(xiàn)高亮是在QueryResponse中的另一個(gè)結(jié)果集的。

QueryResponse.getHighlighting()

取出來(lái)的字段值為自定義前綴后綴拼接具體字段值。

@Override
  public List<Book> query(String query) {
    List<Book> bookList = new ArrayList<Book>();
    SolrQuery solrQuery = new SolrQuery();
    //設(shè)置默認(rèn)搜索的域
    solrQuery.set("df", "description");
    solrQuery.setQuery(query);
    //高亮顯示
    solrQuery.setHighlight(true);
    //設(shè)置高亮顯示的域
    solrQuery.addHighlightField("description");
    //高亮顯示前綴
    solrQuery.setHighlightSimplePre("<font color='red'>");
    //后綴
    solrQuery.setHighlightSimplePost("</font>");
    try {
      QueryResponse queryResponse = solrClient.query(solrQuery);
      if (queryResponse == null){
        return null;
      }
      SolrDocumentList solrDocumentList = queryResponse.getResults();
      if (solrDocumentList.isEmpty()){
        return null;
      }
      //獲取高亮
      Map<String, Map<String, List<String>>> map = queryResponse.getHighlighting();
      for (SolrDocument solrDocument : solrDocumentList){
        Book book;
        List<String> list = map.get(solrDocument.get("id")).get("description");
        if (!CollectionUtils.isEmpty(list)){
          solrDocument.setField("description",list.get(0));
        }
        String bookStr = JSONUtil.toJSON(solrDocument);
        book = JSON.parseObject(bookStr,Book.class);
        bookList.add(book);
      }
    } catch (SolrServerException e) {
      e.printStackTrace();
    } catch (IOException e) {
      e.printStackTrace();
    }
    return bookList;
  }

頁(yè)面展示如下：