springsolr4spring-data-solr

Spring Data Solr How to get suggestions


I am beginner in Solr with Spring. I am using Solr 4.6 and Spring 3.x

I have successfully configured Solr to my Spring application and I am able to search the address using below code.

@Query("text:*?0*")
public List<AddressDocument> findAll(String searchText);

My Schema is

<fields>
    <field name="id" type="long" indexed="true" stored="true" required="true" multiValued="false" />
    <field name="name" type="string" indexed="false" stored="true" required="true" multiValued="false"/>
    <field name="number" type="string" indexed="false" stored="true" required="true" multiValued="false"/>
    <field name="address" type="text_general" indexed="false" stored="true" required="true" multiValued="false"/>
    <field name="city" type="string" indexed="false" stored="true" multiValued="false"/>
    <field name="state" type="string" indexed="false" stored="true" multiValued="false"/>
    <field name="zipcode" type="string" indexed="false" stored="true" multiValued="false"/>
    <field name="country" type="string" indexed="false" stored="true" multiValued="false"/>
    <field name="latlng" type="string" indexed="false" stored="true" multiValued="false"/>
    <field name="text" type="text_general" indexed="true" stored="false" multiValued="true"/>
    <field name="_version_" type="long" indexed="true" stored="true"/>
</fields>

<!-- Configure unique key -->
<uniqueKey>id</uniqueKey>
<copyField source="name" dest="text"/>
<copyField source="number" dest="text"/>
<copyField source="address" dest="text"/>
<copyField source="city" dest="text"/>
<copyField source="state" dest="text"/>
<copyField source="zipcode" dest="text"/>
<copyField source="country" dest="text"/>

Now I have integrated the Spellchecker and when I enter any incorrect spelling then it shows me following response in log.

For example the correct city name is Fargo if I enter Frgo then it shows me below output.

2014-01-30 04:09:10,903 DEBUG [http-8080-2] (DirectSolrSpellChecker.java:179) - getSuggestions: [frgo]
2014-01-30 04:09:10,940 DEBUG [http-8080-2] (SpellCheckCollator.java:180) - Collation: text:*fargo* will return 3 hits.
2014-01-30 04:09:10,941  INFO [http-8080-2] (SolrCore.java:1864) - [customer_site_address] webapp=null path=/select params={q=text%3A*frgo*&start=0&rows=0} hits=0 status=0 QTime=43 
2014-01-30 04:09:10,944 DEBUG [http-8080-2] (SolrTemplate.java:326) - Executing query 'q=text%3A*frgo*&start=0&rows=1' against solr.
2014-01-30 04:09:10,951 DEBUG [http-8080-2] (DirectSolrSpellChecker.java:179) - getSuggestions: [frgo]
2014-01-30 04:09:10,959 DEBUG [http-8080-2] (SpellCheckCollator.java:180) - Collation: text:*fargo* will return 3 hits.
2014-01-30 04:09:10,960  INFO [http-8080-2] (SolrCore.java:1864) - [customer_site_address] webapp=null path=/select params={q=text%3A*frgo*&start=0&rows=1} hits=0 status=0 QTime=15 
2014-01-30 04:09:10,961 DEBUG [http-8080-2] (AbstractMessageConverterMethodProcessor.java:150) - Written [[]] as "application/json;charset=UTF-8" using [org.springframework.http.converter.json.MappingJacksonHttpMessageConverter@1fb032d]

As you can see in log it suggest the correct word but I have no idea how can I get this suggestions using @Query or any another annotation.

I did lots of google, but didn't found any perfect solution.


Solution

  • Spellcheck result can be gathered using SolrTemplage.execute.

    final String query = "foo"; //any value to be used in solr query
    
    SpellcheckedPage<ExampleSolrBean> page = solrTemplate.execute(new SolrCallback<SpellcheckedPage<ExampleSolrBean>>() {
    
        @Override
        public SpellcheckedPage<ExampleSolrBean> doInSolr(SolrServer solrServer) throws SolrServerException, IOException {
            SolrQuery q = new SolrQuery("name:"+query);
    
            ModifiableSolrParams params = new ModifiableSolrParams() {
                {
                    add("spellcheck.build", "true");
                    add("spellcheck.q", query);
                    add("spellcheck", "true");
                }
            };
            q.add(CommonParams.QT, "/spell");
            q.add(params);
    
            QueryResponse response = solrServer.query(q);
    
            //init page with search result
            SpellcheckedPage<ExampleSolrBean> page = new SpellcheckedPage<ExampleSolrBean>(solrTemplate.getConverter().read(response.getResults(), ExampleSolrBean.class));
    
            //add spellcheck result
            SpellCheckResponse scr = response.getSpellCheckResponse();
            for (Suggestion suggestion : scr.getSuggestions()) {
                page.addSuggestions(suggestion.getToken(), suggestion.getAlternatives());
            }
    
            return page;
        }
    });
    
    class SpellcheckedPage<T> extends SolrResultPage<T> {
    
        public SpellcheckedPage(List<T> content) {
            super(content);
        }
    
        private Map<String, List<String>> suggestions = new LinkedHashMap<String, List<String>>();
    
        public void addSuggestions(String term, List<String> suggestions) {
            this.suggestions.put(term, suggestions);
        }
    
        public Collection<String> getSuggestions(String term) {
            return this.suggestions.get(term);
        }
    
        public Collection<String> getSuggestions() {
            List<String> allSuggestions = new ArrayList<String>();
            for (List<String> suggestions : this.suggestions.values()) {
                allSuggestions.addAll(suggestions);
            }
            return allSuggestions;
        }
    }