/**
* 1. Get all the fragment buffers 2. When all are received -> Assemble them into one big buffer
* 3. Read headers and byte buffer from big buffer 4. Set headers and buffer in msg 5. Pass msg up
* the stack
*/
private void unfragment(Message msg, FragHeader hdr) {
Address sender = msg.getSrc();
Message assembled_msg = null;
ConcurrentMap<Long, FragEntry> frag_table = fragment_list.get(sender);
if (frag_table == null) {
frag_table = Util.createConcurrentMap(16, .075f, 16);
ConcurrentMap<Long, FragEntry> tmp = fragment_list.putIfAbsent(sender, frag_table);
if (tmp != null) // value was already present
frag_table = tmp;
}
num_received_frags.incrementAndGet();
FragEntry entry = frag_table.get(hdr.id);
if (entry == null) {
entry = new FragEntry(hdr.num_frags);
FragEntry tmp = frag_table.putIfAbsent(hdr.id, entry);
if (tmp != null) entry = tmp;
}
entry.lock();
try {
entry.set(hdr.frag_id, msg);
if (entry.isComplete()) {
assembled_msg = entry.assembleMessage();
frag_table.remove(hdr.id);
}
} finally {
entry.unlock();
}
// assembled_msg=frag_table.add(hdr.id, hdr.frag_id, hdr.num_frags, msg);
if (assembled_msg != null) {
try {
if (log.isTraceEnabled()) log.trace("assembled_msg is " + assembled_msg);
assembled_msg.setSrc(sender); // needed ? YES, because fragments have a null src !!
num_received_msgs.incrementAndGet();
up_prot.up(new Event(Event.MSG, assembled_msg));
} catch (Exception e) {
if (log.isErrorEnabled()) log.error("unfragmentation failed", e);
}
}
}
