source: vlo/branches/vlo-2.13-param/vlo_web_app/src/main/java/eu/clarin/cmdi/vlo/dao/AutoCompleteDao.java @ 2641

package eu.clarin.cmdi.vlo.dao;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.apache.solr.client.solrj.SolrQuery;
import org.apache.solr.client.solrj.response.QueryResponse;
import org.apache.solr.client.solrj.response.SpellCheckResponse.Suggestion;

/**
 * DAO that delivers suggestions for incomplete terms (autocomplete function)
 * 
 * @author Thomas Eckart
 *
 */
public class AutoCompleteDao extends SolrDao {
        
        /**
         * Returns list of suggestions for incomplete input (used for autocomplete function)
         * @param input user input
         * @return list of suggestions
         */
        public List<String> getChoices(String input) {
                List<String> choices = new ArrayList<String>();
                
                SolrQuery query = new SolrQuery();
                query.setQuery(input.toLowerCase());
                query.setQueryType("/suggest");
                QueryResponse response = fireQuery(query);
                if(response.getSpellCheckResponse() != null) {
                        List<Suggestion> suggestions = response.getSpellCheckResponse().getSuggestions();
                        if(suggestions.size() > 0) {
                                Iterator<String> iter = suggestions.get(0).getAlternatives().iterator();
                                while(iter.hasNext()) {
                                        choices.add(iter.next());
                                }
                        }
                }
                
                return choices;
        }
}