@Test public void testNaNInf() throws IOException { SolrQueryRequest req = req("dummy"); SolrQueryResponse rsp = new SolrQueryResponse(); QueryResponseWriter w = new PythonResponseWriter(); StringWriter buf = new StringWriter(); rsp.add("data1", Float.NaN); rsp.add("data2", Double.NEGATIVE_INFINITY); rsp.add("data3", Float.POSITIVE_INFINITY); w.write(buf, req, rsp); assertEquals( buf.toString(), "{'data1':float('NaN'),'data2':-float('Inf'),'data3':float('Inf')}"); w = new RubyResponseWriter(); buf = new StringWriter(); w.write(buf, req, rsp); assertEquals(buf.toString(), "{'data1'=>(0.0/0.0),'data2'=>-(1.0/0.0),'data3'=>(1.0/0.0)}"); w = new JSONResponseWriter(); buf = new StringWriter(); w.write(buf, req, rsp); assertEquals( buf.toString(), "{\"data1\":\"NaN\",\"data2\":\"-Infinity\",\"data3\":\"Infinity\"}"); req.close(); }
void doPerf(String writerName, SolrQueryRequest req, int encIter, int decIter) throws Exception { SolrQueryResponse rsp = getResponse(req); QueryResponseWriter w = h.getCore().getQueryResponseWriter(writerName); ByteArrayOutputStream out = null; System.gc(); long start = System.currentTimeMillis(); for (int i = 0; i < encIter; i++) { if (w instanceof BinaryQueryResponseWriter) { BinaryQueryResponseWriter binWriter = (BinaryQueryResponseWriter) w; out = new ByteArrayOutputStream(); binWriter.write(out, req, rsp); out.close(); } else { out = new ByteArrayOutputStream(); // to be fair, from my previous tests, much of the performance will be sucked up // by java's UTF-8 encoding/decoding, not the actual writing Writer writer = new OutputStreamWriter(out, "UTF-8"); w.write(writer, req, rsp); writer.close(); } } long encodeTime = Math.max(System.currentTimeMillis() - start, 1); byte[] arr = out.toByteArray(); start = System.currentTimeMillis(); writerName = writerName.intern(); for (int i = 0; i < decIter; i++) { ResponseParser rp = null; if (writerName == "xml") { rp = new XMLResponseParser(); } else if (writerName == "javabin") { rp = new BinaryResponseParser(); } else { break; } ByteArrayInputStream in = new ByteArrayInputStream(arr); rp.processResponse(in, "UTF-8"); } long decodeTime = Math.max(System.currentTimeMillis() - start, 1); log.info( "writer " + writerName + ", size=" + out.size() + ", encodeRate=" + (encodeTime == 1 ? "N/A" : "" + (encIter * 1000L / encodeTime)) + ", decodeRate=" + (decodeTime == 1 ? "N/A" : "" + (decIter * 1000L / decodeTime))); req.close(); SolrRequestInfo.clearRequestInfo(); }
public void testJSON() throws IOException { SolrQueryRequest req = req("wt", "json", "json.nl", "arrarr"); SolrQueryResponse rsp = new SolrQueryResponse(); JSONResponseWriter w = new JSONResponseWriter(); StringWriter buf = new StringWriter(); NamedList nl = new NamedList(); nl.add("data1", "hello"); nl.add(null, 42); rsp.add("nl", nl); w.write(buf, req, rsp); assertEquals(buf.toString(), "{\"nl\":[[\"data1\",\"hello\"],[null,42]]}"); req.close(); }