Java BinaryResponseWriter类代码示例

import org.apache.solr.response.BinaryResponseWriter; //导入依赖的package包/类
@Override
public void handleRequestBody(SolrQueryRequest req, SolrQueryResponse rsp) throws Exception {
  NamedList<NamedList<NamedList<Object>>> cats = getMBeanInfo(req);
  if(req.getParams().getBool("diff", false)) {
    ContentStream body = null;
    try {
      body = req.getContentStreams().iterator().next();
    }
    catch(Exception ex) {
      throw new SolrException(ErrorCode.BAD_REQUEST, "missing content-stream for diff");
    }
    String content = IOUtils.toString(body.getReader());
    
    NamedList<NamedList<NamedList<Object>>> ref = fromXML(content);
    
    
    // Normalize the output 
    SolrQueryResponse wrap = new SolrQueryResponse();
    wrap.add("solr-mbeans", cats);
    cats = (NamedList<NamedList<NamedList<Object>>>)
        BinaryResponseWriter.getParsedResponse(req, wrap).get("solr-mbeans");
    
    // Get rid of irrelevant things
    ref = normalize(ref);
    cats = normalize(cats);
    
    // Only the changes
    boolean showAll = req.getParams().getBool("all", false);
    rsp.add("solr-mbeans", getDiff(ref,cats, showAll));
  }
  else {
    rsp.add("solr-mbeans", cats);
  }
  rsp.setHttpCaching(false); // never cache, no matter what init config looks like
}
 

import org.apache.solr.response.BinaryResponseWriter; //导入依赖的package包/类
@Test
public void testGroupingSimpleFormatArrayIndexOutOfBoundsExceptionWithJavaBin() throws Exception {
  assertU(add(doc("id", "1", "nullfirst", "1")));
  assertU(add(doc("id", "2", "nullfirst", "1")));
  assertU(add(doc("id", "3", "nullfirst", "2")));
  assertU(add(doc("id", "4", "nullfirst", "2")));
  assertU(add(doc("id", "5", "nullfirst", "2")));
  assertU(add(doc("id", "6", "nullfirst", "3")));
  assertU(commit());

  SolrQueryRequest request =
      req("q", "*:*","group", "true", "group.field", "nullfirst", "group.main", "true", "wt", "javabin", "start", "4", "rows", "10");

  SolrQueryResponse response = new SolrQueryResponse();
  ByteArrayOutputStream out = new ByteArrayOutputStream();
  try {
    SolrRequestInfo.setRequestInfo(new SolrRequestInfo(request, response));
    String handlerName = request.getParams().get(CommonParams.QT);
    h.getCore().execute(h.getCore().getRequestHandler(handlerName), request, response);
    BinaryResponseWriter responseWriter = new BinaryResponseWriter();
    responseWriter.write(out, request, response);
  } finally {
    request.close();
    SolrRequestInfo.clearRequestInfo();
  }

  assertEquals(6, ((ResultContext) response.getValues().get("response")).docs.matches());
  new BinaryResponseParser().processResponse(new ByteArrayInputStream(out.toByteArray()), "");
  out.close();
}
 

import org.apache.solr.response.BinaryResponseWriter; //导入依赖的package包/类
/**
 * Parse the solr response to named list (need to create solrj query
 * respond).
 * 
 * @param req
 *            The request.
 * @param rsp
 *            The response.
 * @return The named list.
 */
public NamedList<Object> getParsedResponse(SolrQueryRequest req, SolrQueryResponse rsp) {
	try {
		BinaryResponseWriter writer = new BinaryResponseWriter();
		ByteArrayOutputStream bos = new ByteArrayOutputStream();
		writer.write(bos, req, rsp);
		BinaryResponseParser parser = new BinaryResponseParser();
		return parser.processResponse(new ByteArrayInputStream(bos.toByteArray()), "UTF-8");
	} catch (Exception ex) {
		throw new RuntimeException(ex);
	}
}
 

import org.apache.solr.response.BinaryResponseWriter; //导入依赖的package包/类
@Override
public void processAdd(AddUpdateCommand cmd) throws IOException {
    SolrInputDocument updateDoc = cmd.getSolrInputDocument();
    String id = (String) updateDoc.getField(ID_FIELD).getValue();
    String query = "+" + ID_FIELD + ":" + escape(id);
    final SolrIndexSearcher searcher = cmd.getReq().getSearcher();
    LocalSolrQueryRequest newReq = new LocalSolrQueryRequest(cmd.getReq().getCore(), query, "", 0, 1, REQ_ARGS) {
            @Override
            public SolrIndexSearcher getSearcher() { return searcher; }
            @Override
            public void close() { }
        };
    SolrQueryResponse rsp = new SolrQueryResponse();
    cmd.getReq().getCore().execute(cmd.getReq().getCore().getRequestHandler(REQUEST_HANDLER), newReq, rsp);
    QueryResponse qRsp = new QueryResponse();
    qRsp.setResponse(BinaryResponseWriter.getParsedResponse(newReq, rsp));
    if (qRsp.getResults().size() > 0) {
        SolrDocument existingDoc = qRsp.getResults().get(0);
        String existingContent = (String) existingDoc.getFieldValue(CONTENT_FIELD);
        Collection existingArcnames = existingDoc.getFieldValues(ARCNAME_FIELD);

        if (existingArcnames.contains((String)updateDoc.getFieldValue(ARCNAME_FIELD))) {
            /* abort if already indexed */
            return;
        } else {
            List<String> removedFields = new ArrayList<String>();
            /* fields that should be the same */
            for (String fieldName : updateDoc.getFieldNames()) {
                if (!fieldName.equals(DATE_FIELD) &&
                    !fieldName.equals(JOB_FIELD) &&
                    !fieldName.equals(ARCNAME_FIELD) &&
                    !fieldName.equals(ID_FIELD) &&
                    !fieldName.equals(CONTENT_FIELD)) {
                    removedFields.add(fieldName);
                }
            }

            for (String removedField : removedFields) {
                updateDoc.removeField(removedField);
            }

            for (String f : UPDATE_FIELD_ARR) {
                final String val = (String) updateDoc.getFieldValue(f);
                updateDoc.setField(f, new HashMap(){{ put("add", val); }});
            }

            /* in case we had empty content before */
            if (existingContent == null || existingContent == "") {
                final String updateDocContent = (String) updateDoc.getFieldValue(CONTENT_FIELD);
                updateDoc.setField(CONTENT_FIELD, new HashMap(){{ put("set", updateDocContent); }});
            } else {
                /* otherwise just keep the old content */
                updateDoc.removeField(CONTENT_FIELD);
            }
        }
    }
    super.processAdd(cmd);
}
 

import org.apache.solr.response.BinaryResponseWriter; //导入依赖的package包/类
/**
 * Returns a response object equivalent to what you get from the XML/JSON/javabin parser. Documents
 * become SolrDocuments, DocList becomes SolrDocumentList etc.
 * 
 * @deprecated use {@link BinaryResponseWriter#getParsedResponse(SolrQueryRequest, SolrQueryResponse)}
 */
@Deprecated
public NamedList<Object> getParsedResponse( SolrQueryRequest req, SolrQueryResponse rsp )
{
  return BinaryResponseWriter.getParsedResponse(req, rsp);
}