Spring Data Solr How to get suggestions

Question

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.

Answer 1

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