Examples with ShardResponse org.apache.solr.handler.component.ShardResponse used on opensource projects

Example 1 with ShardResponse

use of org.apache.solr.handler.component.ShardResponse in project lucene-solr by apache.

the class StoredFieldsShardResponseProcessor method process.

/**
   * {@inheritDoc}
   */
@Override
public void process(ResponseBuilder rb, ShardRequest shardRequest) {
    boolean returnScores = (rb.getFieldFlags() & SolrIndexSearcher.GET_SCORES) != 0;
    ShardResponse srsp = shardRequest.responses.get(0);
    SolrDocumentList docs = (SolrDocumentList) srsp.getSolrResponse().getResponse().get("response");
    String uniqueIdFieldName = rb.req.getSchema().getUniqueKeyField().getName();
    for (SolrDocument doc : docs) {
        Object id = doc.getFieldValue(uniqueIdFieldName).toString();
        ShardDoc shardDoc = rb.resultIds.get(id);
        FieldDoc fieldDoc = (FieldDoc) shardDoc;
        if (shardDoc != null) {
            if (returnScores && !Float.isNaN(fieldDoc.score)) {
                doc.setField("score", fieldDoc.score);
            }
            rb.retrievedDocuments.put(id, doc);
        }
    }
}

Example 2 with ShardResponse

use of org.apache.solr.handler.component.ShardResponse in project lucene-solr by apache.

the class ExactStatsCache method mergeToGlobalStats.

@Override
public void mergeToGlobalStats(SolrQueryRequest req, List<ShardResponse> responses) {
    Set<Object> allTerms = new HashSet<>();
    for (ShardResponse r : responses) {
        LOG.debug("Merging to global stats, shard={}, response={}", r.getShard(), r.getSolrResponse().getResponse());
        String shard = r.getShard();
        SolrResponse res = r.getSolrResponse();
        NamedList<Object> nl = res.getResponse();
        // TODO: nl == null if not all shards respond (no server hosting shard)
        String termStatsString = (String) nl.get(TERM_STATS_KEY);
        if (termStatsString != null) {
            addToPerShardTermStats(req, shard, termStatsString);
        }
        List<Object> terms = nl.getAll(TERMS_KEY);
        allTerms.addAll(terms);
        String colStatsString = (String) nl.get(COL_STATS_KEY);
        Map<String, CollectionStats> colStats = StatsUtil.colStatsMapFromString(colStatsString);
        if (colStats != null) {
            addToPerShardColStats(req, shard, colStats);
        }
    }
    if (allTerms.size() > 0) {
        req.getContext().put(TERMS_KEY, Lists.newArrayList(allTerms));
    }
    if (LOG.isDebugEnabled())
        printStats(req);
}

Example 3 with ShardResponse

use of org.apache.solr.handler.component.ShardResponse in project lucene-solr by apache.

the class FacetQueryMerger method handleResponses.

@Override
public void handleResponses(ResponseBuilder rb, ShardRequest sreq) {
    FacetComponentState facetState = getFacetComponentState(rb);
    if (facetState == null)
        return;
    for (ShardResponse shardRsp : sreq.responses) {
        SolrResponse rsp = shardRsp.getSolrResponse();
        NamedList<Object> top = rsp.getResponse();
        // shards.tolerant=true will cause this to happen on exceptions/errors
        if (top == null)
            continue;
        Object facet = top.get("facets");
        if (facet == null)
            continue;
        if (facetState.merger == null) {
            facetState.merger = facetState.facetRequest.createFacetMerger(facet);
            facetState.mcontext = new FacetMerger.Context(sreq.responses.size());
        }
        if ((sreq.purpose & PURPOSE_REFINE_JSON_FACETS) != 0) {
            System.err.println("REFINE FACET RESULT FROM SHARD = " + facet);
            // call merge again with a diff flag set on the context???
            //        throw new SolrException(SolrException.ErrorCode.BAD_REQUEST, "WORK IN PROGRESS, MERGING FACET REFINEMENT NOT SUPPORTED YET!");
            facetState.mcontext.root = facet;
            // TODO: roll newShard into setShard?
            facetState.mcontext.setShard(shardRsp.getShard());
            facetState.merger.merge(facet, facetState.mcontext);
            return;
        }
        facetState.mcontext.root = facet;
        facetState.mcontext.newShard(shardRsp.getShard());
        facetState.merger.merge(facet, facetState.mcontext);
    }
}

Example 4 with ShardResponse

use of org.apache.solr.handler.component.ShardResponse in project lucene-solr by apache.

the class PeerSync method alreadyInSync.

/**
   * Check if we are already in sync. Simple fingerprint comparison should do
   */
private boolean alreadyInSync() {
    for (String replica : replicas) {
        requestFingerprint(replica);
    }
    for (; ; ) {
        ShardResponse srsp = shardHandler.takeCompletedOrError();
        if (srsp == null)
            break;
        Object replicaFingerprint = srsp.getSolrResponse().getResponse().get("fingerprint");
        if (replicaFingerprint == null) {
            log.warn("Replica did not return a fingerprint - possibly an older Solr version");
            continue;
        }
        try {
            IndexFingerprint otherFingerprint = IndexFingerprint.fromObject(replicaFingerprint);
            IndexFingerprint ourFingerprint = IndexFingerprint.getFingerprint(core, Long.MAX_VALUE);
            if (IndexFingerprint.compare(otherFingerprint, ourFingerprint) == 0) {
                log.info("We are already in sync. No need to do a PeerSync ");
                return true;
            }
        } catch (IOException e) {
            log.warn("Could not cofirm if we are already in sync. Continue with PeerSync");
        }
    }
    return false;
}

Example 5 with ShardResponse

use of org.apache.solr.handler.component.ShardResponse in project lucene-solr by apache.

the class OverseerCollectionMessageHandler method processResponses.

void processResponses(NamedList results, ShardHandler shardHandler, boolean abortOnError, String msgOnError, String asyncId, Map<String, String> requestMap, Set<String> okayExceptions) {
    //Processes all shard responses
    ShardResponse srsp;
    do {
        srsp = shardHandler.takeCompletedOrError();
        if (srsp != null) {
            processResponse(results, srsp, okayExceptions);
            Throwable exception = srsp.getException();
            if (abortOnError && exception != null) {
                // drain pending requests
                while (srsp != null) {
                    srsp = shardHandler.takeCompletedOrError();
                }
                throw new SolrException(ErrorCode.SERVER_ERROR, msgOnError, exception);
            }
        }
    } while (srsp != null);
    //If request is async wait for the core admin to complete before returning
    if (asyncId != null) {
        waitForAsyncCallsToComplete(requestMap, results);
        requestMap.clear();
    }
}