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);
                    }
                }
            }
        }
    }
}