/**
* The implementation for getPackets() function. This is a blocking API.
*
* @param sessionid
* @return A list of packets associated with the session
*/
@Override
public List<ByteBuffer> getPackets(String sessionid) throws org.apache.thrift.TException {
List<ByteBuffer> packets = new ArrayList<ByteBuffer>();
int count = 0;
while (!msgQueues.containsKey(sessionid) && count++ < 100) {
log.debug("Queue for session {} doesn't exist yet.", sessionid);
try {
Thread.sleep(100); // Wait 100 ms to check again.
} catch (InterruptedException e) {
log.error(e.toString());
}
}
if (count < 100) {
SessionQueue pQueue = msgQueues.get(sessionid);
BlockingQueue<ByteBuffer> queue = pQueue.getQueue();
// Block if queue is empty
try {
packets.add(queue.take());
queue.drainTo(packets);
} catch (InterruptedException e) {
log.error(e.toString());
}
}
return packets;
}
/**
* The implementation for terminateSession() function. It removes the session to queue
* association.
*
* @param sessionid
* @throws TException
*/
@Override
public void terminateSession(String sessionid) throws org.apache.thrift.TException {
if (!msgQueues.containsKey(sessionid)) {
return;
}
SessionQueue pQueue = msgQueues.get(sessionid);
log.debug("terminateSession: SessionId: " + sessionid + "\n");
String data = "FilterTimeout";
ByteBuffer bb = ByteBuffer.wrap(data.getBytes());
BlockingQueue<ByteBuffer> queue = pQueue.getQueue();
if (queue != null) {
if (!queue.offer(bb)) {
log.error("Failed to queue message for session: " + sessionid);
}
msgQueues.remove(sessionid);
} else {
log.error("queue for session {} is null", sessionid);
}
}
/**
* The implementation for pushMessageSync() function.
*
* @param msg
* @return 1 for success, 0 for failure
* @throws TException
*/
@Override
public int pushMessageSync(Message msg) throws org.apache.thrift.TException {
if (msg == null) {
log.error("Error, pushMessageSync: empty message ");
return 0;
}
List<String> sessionids = msg.getSessionIDs();
for (String sid : sessionids) {
SessionQueue pQueue = null;
if (!msgQueues.containsKey(sid)) {
pQueue = new SessionQueue();
msgQueues.put(sid, pQueue);
} else {
pQueue = msgQueues.get(sid);
}
log.debug(
"pushMessageSync: SessionId: " + sid + " Receive a message, " + msg.toString() + "\n");
ByteBuffer bb = ByteBuffer.wrap(msg.getPacket().getData());
// ByteBuffer dst = ByteBuffer.wrap(msg.getPacket().toString().getBytes());
BlockingQueue<ByteBuffer> queue = pQueue.getQueue();
if (queue != null) {
if (!queue.offer(bb)) {
log.error("Failed to queue message for session: " + sid);
} else {
log.debug("insert a message to session: " + sid);
}
} else {
log.error("queue for session {} is null", sid);
}
}
return 1;
}