@Override
public void onLinkFlow(Event event) {
Sender snd = (Sender) event.getLink();
if (snd.getCredit() > 0 && snd.getLocalState() != EndpointState.CLOSED) {
Message message = Proton.message();
++count;
message.setBody(new AmqpValue("message-" + count));
byte[] msgData = new byte[1024];
int length;
while (true) {
try {
length = message.encode(msgData, 0, msgData.length);
break;
} catch (BufferOverflowException e) {
msgData = new byte[msgData.length * 2];
}
}
byte[] tag = String.valueOf(count).getBytes();
Delivery dlv = snd.delivery(tag);
snd.send(msgData, 0, length);
dlv.settle();
snd.advance();
if (count == numMsgs) {
snd.close();
snd.getSession().close();
snd.getSession().getConnection().close();
result = true;
}
}
}
@Override
public void onDelivery(Event event) {
Receiver recv = (Receiver) event.getLink();
Delivery delivery = recv.current();
if (delivery.isReadable() && !delivery.isPartial()) {
int size = delivery.pending();
byte[] buffer = new byte[size];
int read = recv.recv(buffer, 0, buffer.length);
recv.advance();
Message msg = Proton.message();
msg.decode(buffer, 0, read);
++count;
String msgBody = ((AmqpValue) msg.getBody()).getValue().toString();
String expected = "message-" + count;
if (!expected.equals(msgBody)) {
throw new RuntimeException(
"Received message body '" + msgBody + "', expected: '" + expected + "'");
}
if (count == numMsgs) {
recv.close();
recv.getSession().close();
recv.getSession().getConnection().close();
acceptor.close();
}
}
}
