@Override
protected void setUp() throws Exception {
sq = new SimpleTimer(true);
q = sq;
oldService = ErrorService.getErrorCallback();
serviceStub = new ErrorCallbackStub();
ErrorService.setErrorCallback(serviceStub);
}
/** Times out old responders. */
protected void timeout(long now) {
List<Responder> responders = null;
synchronized (RESPONDERS) {
Responder next = null;
for (int i = RESPONDERS.size() - 1; i >= 0; i--) {
next = RESPONDERS.get(i);
if (next.dead <= now) {
REQUESTED.remove(next.urn);
TIMEOUTS.add(next.urn);
if (responders == null) responders = new ArrayList<Responder>(2);
responders.add(next);
RESPONDERS.remove(i);
next = null;
} else {
break;
}
}
}
// Now call outside of lock.
if (responders != null) {
for (int i = 0; i < responders.size(); i++) {
Responder next = responders.get(i);
if (LOG.isDebugEnabled())
LOG.debug("Timing out responder: " + next + " for URN: " + next.urn);
try {
next.observer.handleResponse(next.urn, null);
} catch (Throwable t) {
ErrorService.error(t, "Content ContentResponseData Error");
}
}
}
}
/** Adds XML to the responses in a QueryReply. */
public static boolean addXMLToResponses(
QueryReply qr, LimeXMLDocumentHelper limeXMLDocumentHelper) {
// get xml collection string, then get dis-aggregated docs, then
// in loop
// you can match up metadata to responses
String xmlCollectionString = "";
try {
LOG.trace("Trying to do uncompress XML.....");
byte[] xmlCompressed = qr.getXMLBytes();
if (xmlCompressed.length > 1) {
byte[] xmlUncompressed = LimeXMLUtils.uncompress(xmlCompressed);
xmlCollectionString = new String(xmlUncompressed, "UTF-8");
}
} catch (UnsupportedEncodingException use) {
// b/c this should never happen, we will show and error
// if it ever does for some reason.
// we won't throw a BadPacketException here but we will show it.
// the uee will effect the xml part of the reply but we could
// still show the reply so there shouldn't be any ill effect if
// xmlCollectionString is ""
ErrorService.error(use);
} catch (IOException ignored) {
}
// valid response, no XML in EQHD.
if (xmlCollectionString.equals("")) return true;
Response[] responses;
int responsesLength;
try {
responses = qr.getResultsArray();
responsesLength = responses.length;
} catch (BadPacketException bpe) {
LOG.trace("Unable to get responses", bpe);
return false;
}
if (LOG.isDebugEnabled()) LOG.debug("xmlCollectionString = " + xmlCollectionString);
List<LimeXMLDocument[]> allDocsArray =
limeXMLDocumentHelper.getDocuments(xmlCollectionString, responsesLength);
for (int i = 0; i < responsesLength; i++) {
Response response = responses[i];
LimeXMLDocument[] metaDocs;
for (int schema = 0; schema < allDocsArray.size(); schema++) {
metaDocs = allDocsArray.get(schema);
// If there are no documents in this schema, try another.
if (metaDocs == null) continue;
// If this schema had a document for this response, use it.
if (metaDocs[i] != null) {
response.setDocument(metaDocs[i]);
break; // we only need one, so break out.
}
}
}
return true;
}
/** Deflates (compresses) the data. */
public static byte[] deflate(byte[] data) {
OutputStream dos = null;
Deflater def = null;
try {
def = new Deflater();
ByteArrayOutputStream baos = new ByteArrayOutputStream();
dos = new DeflaterOutputStream(baos, def);
dos.write(data, 0, data.length);
dos.close(); // flushes bytes
return baos.toByteArray();
} catch (IOException impossible) {
ErrorService.error(impossible);
return null;
} finally {
close(dos);
close(def);
}
}
@Override
protected void tearDown() throws Exception {
ErrorService.setErrorCallback(oldService);
assertEquals(0, serviceStub.getExceptionCount());
}
