use of org.apache.qpid.amqp_1_0.type.Binary in project qpid-broker-j by apache.
the class SendingLinkEndpoint method receiveFlow.
@Override
public void receiveFlow(final Flow flow) {
UnsignedInteger receiverDeliveryCount = flow.getDeliveryCount();
UnsignedInteger receiverLinkCredit = flow.getLinkCredit();
setDrain(flow.getDrain());
Map<Symbol, Object> properties = flow.getProperties();
if (properties != null) {
final Binary transactionId = (Binary) properties.get(Symbol.valueOf("txn-id"));
if (transactionId != null) {
try {
getSession().getTransaction(transactionId);
} catch (UnknownTransactionException e) {
close(new Error(TransactionError.UNKNOWN_ID, e.getMessage()));
return;
}
}
_transactionId = transactionId;
}
if (receiverDeliveryCount == null) {
setLinkCredit(receiverLinkCredit);
} else {
// 2.6.7 Flow Control : link_credit_snd := delivery_count_rcv + link_credit_rcv - delivery_count_snd
UnsignedInteger limit = receiverDeliveryCount.add(receiverLinkCredit);
if (limit.compareTo(getDeliveryCount().unsignedIntegerValue()) <= 0) {
setLinkCredit(UnsignedInteger.valueOf(0));
} else {
setLinkCredit(limit.subtract(getDeliveryCount().unsignedIntegerValue()));
}
}
// send flow when echo=true or drain=true but link credit is zero
boolean sendFlow = Boolean.TRUE.equals(flow.getEcho()) || (Boolean.TRUE.equals(flow.getDrain()) && getLinkCredit().equals(UnsignedInteger.ZERO));
flowStateChanged();
if (sendFlow) {
sendFlow();
}
}
use of org.apache.qpid.amqp_1_0.type.Binary in project qpid-broker-j by apache.
the class StandardReceivingLinkEndpoint method receiveDelivery.
@Override
protected Error receiveDelivery(Delivery delivery) {
ReceiverSettleMode transferReceiverSettleMode = delivery.getReceiverSettleMode();
if (delivery.getResume()) {
DeliveryState deliveryState = _unsettled.get(delivery.getDeliveryTag());
if (deliveryState instanceof Outcome) {
boolean settled = shouldReceiverSettleFirst(transferReceiverSettleMode);
updateDisposition(delivery.getDeliveryTag(), deliveryState, settled);
return null;
} else {
// TODO: QPID-7845: create message ?
}
} else {
ServerMessage<?> serverMessage;
UnsignedInteger messageFormat = delivery.getMessageFormat();
DeliveryState xfrState = delivery.getState();
MessageFormat format = MessageFormatRegistry.getFormat(messageFormat.intValue());
if (format != null) {
try (QpidByteBuffer payload = delivery.getPayload()) {
serverMessage = format.createMessage(payload, getAddressSpace().getMessageStore(), getSession().getConnection().getReference());
} catch (AmqpErrorRuntimeException e) {
return e.getCause().getError();
}
} else {
final Error err = new Error();
err.setCondition(AmqpError.NOT_IMPLEMENTED);
err.setDescription("Unknown message format: " + messageFormat);
return err;
}
MessageReference<?> reference = serverMessage.newReference();
try {
Binary transactionId = null;
if (xfrState != null) {
if (xfrState instanceof TransactionalState) {
transactionId = ((TransactionalState) xfrState).getTxnId();
}
}
final ServerTransaction transaction;
boolean setRollbackOnly = true;
if (transactionId != null) {
try {
transaction = getSession().getTransaction(transactionId);
} catch (UnknownTransactionException e) {
return new Error(TransactionError.UNKNOWN_ID, String.format("transaction-id '%s' is unknown.", transactionId));
}
if (!(transaction instanceof AutoCommitTransaction)) {
transaction.addPostTransactionAction(new ServerTransaction.Action() {
@Override
public void postCommit() {
updateDisposition(delivery.getDeliveryTag(), null, true);
}
@Override
public void onRollback() {
updateDisposition(delivery.getDeliveryTag(), null, true);
}
});
}
} else {
transaction = new AsyncAutoCommitTransaction(getAddressSpace().getMessageStore(), this);
}
try {
Session_1_0 session = getSession();
session.getAMQPConnection().checkAuthorizedMessagePrincipal(serverMessage.getMessageHeader().getUserId());
Outcome outcome;
if (serverMessage.isPersistent() && !getAddressSpace().getMessageStore().isPersistent()) {
final Error preconditionFailedError = new Error(AmqpError.PRECONDITION_FAILED, "Non-durable message store cannot accept durable message.");
if (_rejectedOutcomeSupportedBySource) {
final Rejected rejected = new Rejected();
rejected.setError(preconditionFailedError);
outcome = rejected;
} else {
// TODO - disposition not updated for the non-transaction case
return preconditionFailedError;
}
} else {
try {
getReceivingDestination().send(serverMessage, transaction, session.getSecurityToken());
outcome = ACCEPTED;
} catch (UnroutableMessageException e) {
final Error error = new Error();
error.setCondition(e.getErrorCondition());
error.setDescription(e.getMessage());
String targetAddress = getTarget().getAddress();
if (targetAddress == null || "".equals(targetAddress.trim())) {
error.setInfo(Collections.singletonMap(DELIVERY_TAG, delivery.getDeliveryTag()));
}
if (!_rejectedOutcomeSupportedBySource || (delivery.isSettled() && !(transaction instanceof LocalTransaction))) {
return error;
} else {
if (delivery.isSettled() && transaction instanceof LocalTransaction) {
((LocalTransaction) transaction).setRollbackOnly();
}
Rejected rejected = new Rejected();
rejected.setError(error);
outcome = rejected;
}
}
}
Outcome sourceDefaultOutcome = getSource().getDefaultOutcome();
boolean defaultOutcome = sourceDefaultOutcome != null && sourceDefaultOutcome.getSymbol().equals(outcome.getSymbol());
DeliveryState resultantState;
if (transactionId == null) {
resultantState = defaultOutcome ? null : outcome;
} else {
TransactionalState transactionalState = new TransactionalState();
transactionalState.setOutcome(defaultOutcome ? null : outcome);
transactionalState.setTxnId(transactionId);
resultantState = transactionalState;
}
boolean settled = shouldReceiverSettleFirst(transferReceiverSettleMode);
if (transaction instanceof AsyncAutoCommitTransaction) {
_pendingDispositions.add(new PendingDispositionHolder(delivery.getDeliveryTag(), resultantState, settled));
} else {
getSession().receivedComplete();
updateDisposition(delivery.getDeliveryTag(), resultantState, settled);
}
getSession().getAMQPConnection().registerMessageReceived(serverMessage.getSize());
if (transactionId != null) {
getSession().getAMQPConnection().registerTransactedMessageReceived();
}
setRollbackOnly = false;
} catch (AccessControlException e) {
final Error err = new Error();
err.setCondition(AmqpError.NOT_ALLOWED);
err.setDescription(e.getMessage());
return err;
} finally {
if (setRollbackOnly && transaction instanceof LocalTransaction) {
((LocalTransaction) transaction).setRollbackOnly();
}
}
} finally {
reference.release();
}
}
return null;
}
use of org.apache.qpid.amqp_1_0.type.Binary in project qpid-broker-j by apache.
the class MessageConverter_v1_0_to_InternalTest method doTestDataWithAnnotation.
private void doTestDataWithAnnotation(final byte[] data, final MessageAnnotations messageAnnotations, final String mimeType, final String expectedMimeType) {
final Data value = new Data(new Binary(data));
Message_1_0 sourceMessage;
if (mimeType != null) {
Properties properties = new Properties();
properties.setContentType(Symbol.valueOf(mimeType));
sourceMessage = createTestMessage(properties, messageAnnotations, value.createEncodingRetainingSection());
} else {
sourceMessage = createTestMessage(messageAnnotations, value.createEncodingRetainingSection());
}
final InternalMessage convertedMessage = _converter.convert(sourceMessage, mock(NamedAddressSpace.class));
assertEquals("Unexpected mime type", expectedMimeType, convertedMessage.getMessageHeader().getMimeType());
assertArrayEquals("Unexpected content", data, ((byte[]) convertedMessage.getMessageBody()));
}
use of org.apache.qpid.amqp_1_0.type.Binary in project storm by apache.
the class EventHubReceiverMock method receive.
@Override
public EventData receive(long timeoutInMilliseconds) {
if (isPaused) {
return null;
}
currentOffset++;
List<Section> body = new ArrayList<Section>();
//the body of the message is "message" + currentOffset, e.g. "message123"
body.add(new Data(new Binary(("message" + currentOffset).getBytes())));
Message m = new Message(body);
MessageId mid = new MessageId(partitionId, "" + currentOffset, currentOffset);
EventData ed = new EventData(m, mid);
return ed;
}
use of org.apache.qpid.amqp_1_0.type.Binary in project qpid-broker-j by apache.
the class MessageConverter_1_0_to_v0_8Test method testDataWithContentTypeAmqpList.
public void testDataWithContentTypeAmqpList() throws Exception {
List<Object> originalMap = Collections.singletonList("testValue");
byte[] bytes = new ListToAmqpListConverter().toMimeContent(originalMap);
final Data value = new Data(new Binary(bytes));
Properties properties = new Properties();
properties.setContentType(Symbol.valueOf("amqp/list"));
Message_1_0 sourceMessage = createTestMessage(properties, value.createEncodingRetainingSection());
final AMQMessage convertedMessage = _converter.convert(sourceMessage, mock(NamedAddressSpace.class));
assertEquals("Unexpected mime type", "amqp/list", convertedMessage.getMessageHeader().getMimeType());
final QpidByteBuffer content = convertedMessage.getContent(0, (int) convertedMessage.getSize());
assertArrayEquals("Unexpected content", bytes, getBytes(content));
}
Aggregations