solrJ(bean)的使用

package solr;


import java.io.IOException;
import java.util.List;
import java.util.Map;


import org.apache.solr.client.solrj.SolrClient;
import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.SolrQuery.ORDER;
import org.apache.solr.client.solrj.SolrServerException;
import org.apache.solr.client.solrj.impl.HttpSolrClient;
import org.apache.solr.client.solrj.response.FacetField;
import org.apache.solr.client.solrj.response.Group;
import org.apache.solr.client.solrj.response.GroupCommand;
import org.apache.solr.client.solrj.response.GroupResponse;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.client.solrj.response.FacetField.Count;
import org.apache.solr.common.SolrDocument;
import org.apache.solr.common.SolrDocumentList;
import org.apache.solr.common.params.GroupParams;
import org.junit.Test;


public class TestFoods {


    static String urlString = "http://localhost:8080/solr/core2";


    static SolrClient solr;
    static {
        solr = new HttpSolrClient(urlString);
    }


    @Test
    public void write() throws IOException, SolrServerException {
        Foods foods = new Foods();
        foods.setId("2");
        foods.setFoodId(1);
        foods.setFoodName("蝦仁炒蛋");
        foods.setPrice(10);
        foods.setTypeName("粵菜");
        //添加到索引庫
        solr.addBean(foods);
        solr.commit();
        solr.close();
    }


    @Test
    public void read() throws SolrServerException, IOException {
        SolrQuery sq=new SolrQuery();  
        sq.setQuery("*:*");  
        //過濾
        sq.setFilterQueries("foodName:炒蛋");
        //排序
        sq.setSort("foodName",ORDER.desc);
        //分頁
        sq.setStart(0);
        sq.setRows(2);
        QueryResponse query = solr.query(sq);
        //獲取查詢的數據
        List<Foods> beans = query.getBeans(Foods.class);
        for (Foods foods : beans) {
                System.out.println(foods.getFoodName()+":"+foods.getTypeName());
        }
    }
    @Test
    public void readHighlighting() throws SolrServerException, IOException {
        SolrQuery sq=new SolrQuery();  
        sq.setQuery("foodName:炒蛋");  
        //排序
        sq.setSort("foodName",ORDER.desc);
        //分頁
        sq.setStart(0);
        sq.setRows(2);
        //高亮
        //是否高亮
        sq.setHighlight(true);
        //指定高亮的條件
        sq.addHighlightField("foodName");
        sq.setHighlightSimplePre("<font color=red>");
        sq.setHighlightSimplePost("</font>");
      //對查詢的條件進行綁定
        QueryResponse query = solr.query(sq);
        //獲取查詢的數據
        List<Foods> beans = query.getBeans(Foods.class);
        //獲取高亮的數據
        Map<String, Map<String, List<String>>> highlighting = query.getHighlighting();
        for (Foods foods : beans) {
            Map<String, List<String>> map = highlighting.get(foods.getId());
            //獲取高亮的條件獲取數據
            List<String> list = map.get("foodName");
            String hlString=list.get(0);
            System.out.println(hlString);
        }
    }
    
    @Test
    public void deleteById() throws SolrServerException, IOException {
        // 根據id來刪除數據
        solr.deleteById("1");
        solr.commit();
        solr.close();
    }


    @Test
    public void deleteByCondition() throws SolrServerException, IOException {
        // 根據條件來刪除數據
        solr.deleteByQuery("foodName:青椒");
        solr.commit();
        solr.close();
    }


    @Test
    public void facet() throws SolrServerException, IOException {
        SolrQuery solrQuery = new SolrQuery("foodName:炒蛋");
        solrQuery.setFacet(true); // 開啓分類
        solrQuery.setRows(0);
        solrQuery.addFacetField("typeName"); // 按字段分類 相同的歸於一類
        solrQuery.setFacetLimit(10);// 限制facet返回的數量
        solrQuery.setFacetMissing(false);// 不統計null的值
        solrQuery.setFacetMinCount(1);// 設置返回的數據中每個分組的數據最小值，比如設置爲1，則統計數量最小爲1，不然不顯示
        // solrQuery.setFacetSort("count asc"); //根據 count 數量 升序和降序 也可以根據索引
        // 默認是降序
        QueryResponse query = solr.query(solrQuery); // 對查詢的條件進行綁定
        // System.out.println("查詢時間：" + query.getQTime());
        List<FacetField> facets = query.getFacetFields();// 返回的facet列表
        for (FacetField facet : facets) {
            System.out.println(facet.getName());// 獲取分組的條件--typeName
            List<Count> counts = facet.getValues();
            for (Count count : counts) {
                // 打印分組的結果和數量
                System.out.println(count.getName() + ":" + count.getCount());
            }
        }
        solr.close();
    }


    @Test
    public void GroupFieldQuery() throws Exception {
        SolrQuery solrQuery = new SolrQuery("foodName:炒蛋");
        // 開啓group功能
        solrQuery.setParam(GroupParams.GROUP, true);
        // 指定group條件
        solrQuery.setParam(GroupParams.GROUP_FIELD, "typeName");
        // solrQuery.setParam("group.ngroups", true); --開啓分組統計
        // 設置獲取的數據數量,默認只獲取分組中的第一條
        solrQuery.setParam(GroupParams.GROUP_LIMIT, "5");
        // 分頁
        solrQuery.setStart(0);
        solrQuery.setRows(5);
        QueryResponse query = solr.query(solrQuery);
        GroupResponse groupResponse = query.getGroupResponse();
        List<GroupCommand> groupList = groupResponse.getValues();
        for (GroupCommand groupCommand : groupList) {
            System.out.println(groupCommand.getName()); // 獲取group的條件
            List<Group> groups = groupCommand.getValues();
            for (Group group : groups) {
                
                SolrDocumentList results = group.getResult();
                // System.out.println(group.getGroupValue()+"-------"+results.size());
                // 獲取group的結果和數量
                System.out.println(group.getGroupValue() + "====" + group.getResult().getNumFound());
                for (SolrDocument doc : results) {
                    // 獲取group後的數據內容，可以獲取到document中的所有屬性(field)
                    System.out.println(doc.getFieldValue("foodName") + ":" + doc.getFieldValue("typeName") + ":" + doc.getFieldValue("price"));
                }
            }
        }
        solr.close();
    }
}