package com.liucheng.solr;

import java.io.IOException;

import java.util.List;

import org.apache.solr.client.solrj.SolrQuery;

import org.apache.solr.client.solrj.SolrServerException;

import org.apache.solr.client.solrj.impl.HttpSolrClient;

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.common.params.GroupParams;

public class SolrGroup {

    public static void searchByGroup(){

        HttpSolrClient server = solrServer.getServer();

        SolrQuery query = new SolrQuery("brand_s:海尔儿");

        query.setParam(GroupParams.GROUP, true);//表示查询时使用group机制

        query.setParam(GroupParams.GROUP_FIELD, "brand_s");//设置group查询针对的域

        //设置每个组最多返回记录数，可用于数据采集，若只需要数量，可设置为0

        query.setParam(GroupParams.GROUP_LIMIT, "5");

        //设置返回的行数

        query.setRows(10);

        try {

            QueryResponse queryResponse = server.query(query);

            if(queryResponse!=null){

                GroupResponse groupResponse = queryResponse.getGroupResponse();

                if(groupResponse != null){

                    List<GroupCommand> groupList = groupResponse.getValues();

                    for(GroupCommand groupCommand : groupList){

                        List<Group> groups = groupCommand.getValues();

                        for(Group group : groups){

                            System.out.println(group.getResult().get(0).getFieldValue("subMajor_s"));

                            System.out.println("group查询。。。"+group.getGroupValue()+"数量为："

                                    + group.getResult().getNumFound());

                        }

                    }

                }

            }

        } catch (SolrServerException e) {

            e.printStackTrace();

        } catch (IOException e) {

            e.printStackTrace();

        }

    }

    public static void main(String[] args) {

        searchByGroup();

    }

}