solrj+spellcheck

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("&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("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("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"); //根据solrconfig中的设置
		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("™");
			}

		}

	}

}