org.alfresco.solr.component.spellcheck.AlfrescoSpellCheckBackCompatComponent.java Source code

Introduction

Here is the source code for org.alfresco.solr.component.spellcheck.AlfrescoSpellCheckBackCompatComponent.java

Source

/*
 * Copyright (C) 2005-2014 Alfresco Software Limited.
 *
 * This file is part of Alfresco
 *
 * Alfresco is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Alfresco is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with Alfresco. If not, see <http://www.gnu.org/licenses/>.
 */

package org.alfresco.solr.component.spellcheck;

import java.util.HashMap;
import java.util.Map;

import org.apache.solr.common.SolrException;
import org.apache.solr.common.params.ShardParams;
import org.apache.solr.common.util.NamedList;
import org.apache.solr.handler.component.*;

/**
 * @author Joel Bernstein
 * @since 5.2
 */

public class AlfrescoSpellCheckBackCompatComponent extends SearchComponent {
    public void prepare(ResponseBuilder responseBuilder) {

    }

    public void process(ResponseBuilder responseBuilder) {
        if (!responseBuilder.req.getParams().getBool(SpellCheckComponent.COMPONENT_NAME, false)) {
            return;
        }
        NamedList response = responseBuilder.rsp.getValues();

        NamedList spellcheck = (NamedList) response.get("spellcheck");
        if (spellcheck == null) {
            return;
        }

        NamedList collations = (NamedList) spellcheck.get("collations");
        NamedList suggest = (NamedList) spellcheck.get("suggest");

        if (collations == null && suggest == null) {
            return;
        }

        NamedList collationList = collations != null ? collations : suggest;

        NamedList spellCheckExtras = new NamedList();

        for (int i = 0; i < collationList.size(); i++) {
            if ("collation".equals(collationList.getName(i))) {
                NamedList collation = (NamedList) collationList.getVal(i);
                String collationQuery = (String) collation.get("collationQuery");
                String collationQueryString = (String) collation.get("collationQueryString");
                spellCheckExtras.add(collationQuery, collationQueryString);
            }
        }

        response.add("spellcheck-extras", spellCheckExtras);
    }

    public String getDescription() {
        return null;
    }

    public void finishStage(ResponseBuilder rb) {
        if (!rb.req.getParams().getBool(SpellCheckComponent.COMPONENT_NAME, false)
                || rb.stage != ResponseBuilder.STAGE_GET_FIELDS)
            return;

        Map extras = new HashMap();
        for (ShardRequest sreq : rb.finished) {
            for (ShardResponse srsp : sreq.responses) {
                NamedList nl = null;

                try {
                    nl = (NamedList) srsp.getSolrResponse().getResponse().get("spellcheck-extras");
                } catch (Exception e) {
                    if (rb.req.getParams().getBool(ShardParams.SHARDS_TOLERANT, false)) {
                        continue; // looks like a shard did not return anything
                    }
                    throw new SolrException(SolrException.ErrorCode.SERVER_ERROR,
                            "Unable to read spelling info for shard: " + srsp.getShard(), e);
                }

                if (nl != null) {
                    collectExtras(nl, extras);
                }
            }
        }

        if (extras.size() == 0) {
            return;
        }

        NamedList response = rb.rsp.getValues();
        NamedList spellcheck = (NamedList) response.get("spellcheck");

        if (spellcheck == null) {
            return;
        }

        NamedList collations = (NamedList) spellcheck.get("collations");
        NamedList suggestions = (NamedList) spellcheck.get("suggestions");

        if (collations != null) {
            //Fix up the collationQueryString in Solr 6
            for (int i = 0; i < collations.size(); i++) {
                if ("collation".equals(collations.getName(i))) {
                    NamedList collation = (NamedList) collations.getVal(i);
                    String collationQuery = (String) collation.get("collationQuery");
                    String collationQueryString = (String) extras.get(collationQuery);
                    collation.add("collationQueryString", collationQueryString);
                }
            }
            //Add the collations to the suggestions to support the Solr 4 format
            suggestions.addAll(collations);
        } else {
            //Fix up the collationQueryString Solr4
            for (int i = 0; i < suggestions.size(); i++) {
                if ("collation".equals(suggestions.getName(i))) {
                    NamedList collation = (NamedList) suggestions.getVal(i);
                    String collationQuery = (String) collation.get("collationQuery");
                    String collationQueryString = (String) extras.get(collationQuery);
                    collation.add("collationQueryString", collationQueryString);
                }
            }
        }
    }

    private void collectExtras(NamedList spellcheckExtras, Map map) {
        for (int i = 0; i < spellcheckExtras.size(); i++) {
            map.put(spellcheckExtras.getName(i), spellcheckExtras.getVal(i));
        }
    }
}