Solrj example

package com.doculibre.constellio.servlets;

import java.net.MalformedURLException;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

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.CommonsHttpSolrServer;
import org.apache.solr.client.solrj.impl.XMLResponseParser;
import org.apache.solr.client.solrj.response.FacetField;
import org.apache.solr.client.solrj.response.FacetField.Count;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.common.SolrDocument;
import org.apache.solr.common.SolrDocumentList;
import org.apache.solr.common.params.ModifiableSolrParams;
import org.apache.solr.common.params.SolrParams;
import org.apache.solr.common.util.NamedList;
import org.apache.solr.servlet.SolrRequestParsers;

/**
 * A simple main to illustrate how to execute a request using SolrJ
 * 
  */
public  class SolrJExampleMain {

     private  static  final String myServer = "http://localhost:8080/constellio/app";

     private  static  final String myCollection = "test";

     //  Can be set to 'on', 'off' or 'constellio' to include Constellio's facets
     private  static  final String facet = "constellio";

     //  q=...
     private  static  final String query = "open source";

     private  static  final  int start = 0;
     private  static  final  int nbDocuments = 11;

     public  static  void main(String[] args)  throws MalformedURLException,
            SolrServerException {

         //  Prepare the SolrServer. Right now, the default SolrJ's ResponseParser
         //  isn't supported by Constellio.
        CommonsHttpSolrServer server =  new CommonsHttpSolrServer(myServer);
        server.setParser( new XMLResponseParser());

         //  Do the same query three times using three different method
        System.out
                .println("= = = = = = = = = = = = = = = = = First way to execute a query = = = = = = = = = = = = = = = = =");
        print(doFirstQuery(server));
        System.out
                .println("= = = = = = = = = = = = = = = = = Second way to execute a query = = = = = = = = = = = = = = = = =");
        print(doSecondQuery(server));
        System.out
                .println("= = = = = = = = = = = = = = = = = Third way to execute query = = = = = = = = = = = = = = = = =");
        print(doThirdQuery(server));
        System.out
                .println("= = = = = = = = = = = = = = = = = Using SpellChecker = = = = = = = = = = = = = = = = =");
        print(spellCheck(server, "opn sorce source"));
    }

     /**
     * Do the query using a StringBuffer
      */
     public  static QueryResponse doFirstQuery(SolrServer server)
             throws SolrServerException {
        StringBuffer request =  new StringBuffer();
        request.append("collectionName=" + myCollection);
        request.append("&username=" + "admin");
        request.append("&password=" + "password");
        request.append("&facet=" + facet);
        request.append("&q=" + query);
        request.append("&start=" + start);
        request.append("&rows=" + nbDocuments);
        SolrParams solrParams = SolrRequestParsers.parseQueryString(request
                .toString());

         return server.query(solrParams);
    }

     /**
     * Do the query using a ModifiableSolrParams
      */
     public  static QueryResponse doSecondQuery(SolrServer server)
             throws SolrServerException {
        ModifiableSolrParams solrParams =  new ModifiableSolrParams();
        solrParams.set("collectionName", myCollection);
        solrParams.set("username", "admin");
        solrParams.set("password", "password");
        solrParams.set("facet", facet);
        solrParams.set("q", query);
        solrParams.set("start", start);
        solrParams.set("rows", nbDocuments);
         return server.query(solrParams);
    }

     /**
     * Do the query using a SolrQuery
      */
     public  static QueryResponse doThirdQuery(SolrServer server)
             throws SolrServerException {
        SolrQuery solrQuery =  new SolrQuery();
        solrQuery.setQuery(query);
        solrQuery.set("collectionName", myCollection);
        solrQuery.set("username", "admin");
        solrQuery.set("password", "password");
        solrQuery.set("facet", facet);
        solrQuery.setStart(start);
        solrQuery.setRows(nbDocuments);
         return server.query(solrQuery);
    }

     /**
     * Do the query using a SolrQuery
      */
     public  static QueryResponse spellCheck(SolrServer server, String badQuery)
             throws SolrServerException {
        SolrQuery solrQuery =  new SolrQuery();
        solrQuery.setQuery(badQuery);
        solrQuery.set("collectionName", myCollection);

         //  qt=spellcheck || qt=spellchecker
        solrQuery.setQueryType("spellcheck");
         return server.query(solrQuery);
    }

     /**
     * Print documents and facets
     * 
     *  @param  response
      */
    @SuppressWarnings("unchecked")
     public  static  void print(QueryResponse response) {
        SolrDocumentList docs = response.getResults();
         if (docs !=  null) {
            System.out.println(docs.getNumFound() + " documents found, "
                    + docs.size() + " returned : ");
             for ( int i = 0; i < docs.size(); i++) {
                SolrDocument doc = docs.get(i);
                System.out.println("\t" + doc.toString());
            }
        }

        List<FacetField> fieldFacets = response.getFacetFields();
         if (fieldFacets !=  null && fieldFacets.isEmpty()) {
            System.out.println("\nField Facets : ");
             for (FacetField fieldFacet : fieldFacets) {
                System.out.print("\t" + fieldFacet.getName() + " :\t");
                 if (fieldFacet.getValueCount() > 0) {
                     for (Count count : fieldFacet.getValues()) {
                        System.out.print(count.getName() + "["
                                + count.getCount() + "]\t");
                    }
                }
                System.out.println("");
            }
        }

        Map<String, Integer> queryFacets = response.getFacetQuery();
         if (queryFacets !=  null && !queryFacets.isEmpty()) {
            System.out.println("\nQuery facets : ");
             for (String queryFacet : queryFacets.keySet()) {
                System.out.println("\t" + queryFacet + "\t["
                        + queryFacets.get(queryFacet) + "]");
            }
            System.out.println("");
        }

        NamedList<NamedList<Object>> spellCheckResponse = (NamedList<NamedList<Object>>) response
                .getResponse().get("spellcheck");

         if (spellCheckResponse !=  null) {
            Iterator<Entry<String, NamedList<Object>>> wordsIterator = spellCheckResponse
                    .iterator();

             while (wordsIterator.hasNext()) {
                Entry<String, NamedList<Object>> entry = wordsIterator.next();
                String word = entry.getKey();
                NamedList<Object> spellCheckWordResponse = entry.getValue();
                 boolean correct = spellCheckWordResponse.get("frequency")
                        .equals(1);
                System.out.println("Word: " + word + ",\tCorrect?: " + correct);
                NamedList<Integer> suggestions = (NamedList<Integer>) spellCheckWordResponse
                        .get("suggestions");
                 if (suggestions !=  null && suggestions.size() > 0) {
                    System.out.println("Suggestions : ");
                    Iterator<Entry<String, Integer>> suggestionsIterator = suggestions
                            .iterator();
                     while (suggestionsIterator.hasNext()) {
                        System.out.println("\t"
                                + suggestionsIterator.next().getKey());
                    }

                }
                System.out.println("");
            }

        }

    }

}