package zr.com.solr.utils;

import java.io.IOException;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import javax.management.Query;

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

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

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

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

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

import org.apache.solr.client.solrj.response.FacetField.Count;

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.UpdateResponse;

import org.apache.solr.common.SolrDocument;

import org.apache.solr.common.SolrDocumentList;

import org.apache.solr.common.SolrInputDocument;

import org.apache.solr.common.params.CommonParams;

import org.apache.solr.common.params.GroupParams;

import org.apache.solr.common.params.ModifiableSolrParams;

import org.junit.Test;

/**

 *

 * @author lf

 *

 */

public class SolrUtils {

    public static void main(String[] args) {

        try {

//            SolrUtils.addData();

//            SolrUtils.queryData();

//            System.out.println("===================");

//            SolrUtils.weightQuery();

//            SolrUtils.deleteByID("0");

//            SolrUtils.queryData();

//            SolrUtils.countGroupQuery();

            SolrUtils.groupQuery();

        } catch (Exception e) {

            e.printStackTrace();

        }

    }

      private static HttpSolrServer server = null;

      static{

          server = new HttpSolrServer("http://localhost:28080");

      }

     /**

      * 添加数据

      */

      public  static void addData() throws SolrServerException, IOException{

          for (int i = 500; i < 600; i++) {

            SolrInputDocument doc = new SolrInputDocument();

            doc.addField("name", "wr");

            doc.addField("id", ""+i);

            doc.addField("age", i);

            doc.addField("team", "Team-15");

            doc.addField("hoby", "swimming");

            doc.addField("content", "www"+i);

            server.add(doc);

        }

          server.commit();

          System.out.println("server:"+server);

      }

      /**

       * 查询

       */

      public static void queryData() throws SolrServerException{

          /**

           * 第一种方式查找

           */

          /*ModifiableSolrParams params = new ModifiableSolrParams();

          // 设置查询的条件

          params.set("q", "name:www");

          // 设置获取的字段

          params.set("fl", "name age hoby id content");

          // 过滤查询（年龄在5到8之间）

//          params.set("fq", "age:[5 TO 8]");

          // 不包含（首尾）的范围查询

//          params.set("fq", "age:{5 TO 8}");

          // 设置查询的起始位置

          params.set("start", "0");

//          params.set("wt", "json");

          QueryResponse response = server.query(params);

          SolrDocumentList result = response.getResults();

          System.out.println("数量："+result.size());

          for (int i = 0; i < result.size(); i++) {

            System.out.println(result.get(i));

          }*/

          /**

           * 第二种方式查找

           */

          SolrQuery query = new SolrQuery();

          // 查询条件

          String queryStr = "name:*";

          // 添加查询条件

          query.setQuery(queryStr);

          // 设置起始值

          query.setStart(0);

          // 设置每一页显示的数量

          query.setRows(500);

          // 执行查询

          QueryResponse response = server.query(query);

          SolrDocumentList list = response.getResults();

          System.out.println("数量："+list.size());

          for (int i = 0; i < list.size(); i++) {

            System.out.println(list.get(i));

          }

      }

      /**

       * 权重查找

     * @throws SolrServerException

       */

      public static void weightQuery() throws SolrServerException{

          SolrQuery query = new SolrQuery();

          /*********权重查找**********/

          /* qf对默认查询增加权重比值，

           * 比如:fieldOne^1.9 fieldTwo fieldThree^5 ，

           * 值越大权重越大

          */

          String queryStr = "name:www^2 OR content:www*^15";

          query.setQuery(queryStr);

          query.setStart(0);

          query.setRows(10);

          QueryResponse response = server.query(query);

          SolrDocumentList list = response.getResults();

          System.out.println("数量："+list.size());

          for (int i = 0; i < list.size(); i++) {

            System.out.println(list.get(i));

          }

      }

      /**

       * 删除

       * @throws SolrServerException

       * @throws IOException

       */

      public static void deleteByID(String id) throws SolrServerException, IOException{

          UpdateResponse uResponse = server.deleteById(id);

          server.commit();

      }

      /**

       *  分组统计

       *

       */

    public static void countGroupQuery() throws SolrServerException{

        SolrQuery query = new SolrQuery();

        query.setQuery("name:*");

        // 是否按照每组数量高低排序

        query.setIncludeScore(false);

        // 是否分组查询

        query.setFacet(true);

        //设置返回结果条数，如果你时分组查询，你就设置为0

        query.setRows(0);

        // 限制每次返回的结果数

        query.setFacetLimit(5);

        // 添加分组的字段

        query.addFacetField("name");

        QueryResponse response = server.query(query);

        //返回分组的结果（这里要和addFacetField相对应）

        List<Count> returnList = response.getFacetField("name").getValues();

        System.out.println("数量:"+returnList.size());

        for (Count count : returnList) {

            if (count.getCount() > 0) {

                // 打印每组的名字和数量

                System.out.println(count.getName()+","+count.getCount());

            }

        }

    }

    /**

     * 分组查询

     * @throws SolrServerException

     */

    public static void groupQuery() throws SolrServerException{

        SolrQuery query = new SolrQuery();

        query.setQuery("schoolName:*");

        query.setParam("group", true);

        query.setParam("group.field", "schoolName");

        query.setParam("group.limit", "5");

        query.setParam("group.ngroups", true);

        QueryResponse response = server.query(query);

        GroupResponse gResponse = response.getGroupResponse();

        List<GroupCommand> commands = gResponse.getValues();

        if (commands != null) {

            for (GroupCommand groupCommand : commands) {

                System.out.println("总分组的个数："+groupCommand.getNGroups().longValue());

                for (Group group : groupCommand.getValues()) {

                    System.out.println("-----group:"+group);

                    SolrDocumentList list = group.getResult();

                    for (SolrDocument solrDocument : list) {

                        String id = solrDocument.get("id").toString();

                        String schoolName = solrDocument.get("schoolName").toString();

                        String courseName = solrDocument.get("courseName").toString();

                        System.out.println("id:"+id+",schoolName:"+schoolName+",courseName:"+courseName);

                    }

                }

            }

        }

    }

}