Search in sources :

Example 1 with ActiveMQBytesMessage

use of org.apache.activemq.command.ActiveMQBytesMessage in project nifi by apache.

the class TestJmsConsumer method testMap2FlowFileBytesMessage.

/**
 * Test BytesMessage to FlowFile conversion
 *
 * @throws java.lang.Exception ex
 */
@Test
public void testMap2FlowFileBytesMessage() throws Exception {
    TestRunner runner = TestRunners.newTestRunner(GetJMSQueue.class);
    BytesMessage bytesMessage = new ActiveMQBytesMessage();
    String sourceString = "Apache NiFi is an easy to use, powerful, and reliable system to process and distribute data.!";
    byte[] payload = sourceString.getBytes("UTF-8");
    bytesMessage.writeBytes(payload);
    bytesMessage.reset();
    ProcessContext context = runner.getProcessContext();
    ProcessSession session = runner.getProcessSessionFactory().createSession();
    ProcessorInitializationContext pic = new MockProcessorInitializationContext(runner.getProcessor(), (MockProcessContext) runner.getProcessContext());
    JmsProcessingSummary summary = JmsConsumer.map2FlowFile(context, session, bytesMessage, true, pic.getLogger());
    assertEquals("BytesMessage content length should equal to FlowFile content size", payload.length, summary.getLastFlowFile().getSize());
    final byte[] buffer = new byte[payload.length];
    runner.clearTransferState();
    session.read(summary.getLastFlowFile(), new InputStreamCallback() {

        @Override
        public void process(InputStream in) throws IOException {
            StreamUtils.fillBuffer(in, buffer, false);
        }
    });
    String contentString = new String(buffer, "UTF-8");
    assertEquals("", sourceString, contentString);
}
Also used : ProcessSession(org.apache.nifi.processor.ProcessSession) TestRunner(org.apache.nifi.util.TestRunner) InputStream(java.io.InputStream) MockProcessorInitializationContext(org.apache.nifi.util.MockProcessorInitializationContext) ActiveMQBytesMessage(org.apache.activemq.command.ActiveMQBytesMessage) BytesMessage(javax.jms.BytesMessage) JmsProcessingSummary(org.apache.nifi.processors.standard.util.JmsProcessingSummary) IOException(java.io.IOException) ProcessContext(org.apache.nifi.processor.ProcessContext) MockProcessContext(org.apache.nifi.util.MockProcessContext) MockProcessorInitializationContext(org.apache.nifi.util.MockProcessorInitializationContext) ProcessorInitializationContext(org.apache.nifi.processor.ProcessorInitializationContext) ActiveMQBytesMessage(org.apache.activemq.command.ActiveMQBytesMessage) InputStreamCallback(org.apache.nifi.processor.io.InputStreamCallback) Test(org.junit.Test)

Example 2 with ActiveMQBytesMessage

use of org.apache.activemq.command.ActiveMQBytesMessage in project activemq-artemis by apache.

the class OpenWireMessageConverter method toAMQMessage.

private static ActiveMQMessage toAMQMessage(MessageReference reference, ICoreMessage coreMessage, WireFormat marshaller, AMQConsumer consumer) throws IOException {
    final ActiveMQMessage amqMsg;
    final byte coreType = coreMessage.getType();
    final Boolean compressProp = (Boolean) coreMessage.getObjectProperty(AMQ_MSG_COMPRESSED);
    final boolean isCompressed = compressProp == null ? false : compressProp.booleanValue();
    final byte[] bytes;
    final ActiveMQBuffer buffer = coreMessage.getDataBuffer();
    buffer.resetReaderIndex();
    switch(coreType) {
        case org.apache.activemq.artemis.api.core.Message.BYTES_TYPE:
            amqMsg = new ActiveMQBytesMessage();
            bytes = toAMQMessageBytesType(buffer, isCompressed);
            break;
        case org.apache.activemq.artemis.api.core.Message.MAP_TYPE:
            amqMsg = new ActiveMQMapMessage();
            bytes = toAMQMessageMapType(buffer, isCompressed);
            break;
        case org.apache.activemq.artemis.api.core.Message.OBJECT_TYPE:
            amqMsg = new ActiveMQObjectMessage();
            bytes = toAMQMessageObjectType(buffer, isCompressed);
            break;
        case org.apache.activemq.artemis.api.core.Message.STREAM_TYPE:
            amqMsg = new ActiveMQStreamMessage();
            bytes = toAMQMessageStreamType(buffer, isCompressed);
            break;
        case org.apache.activemq.artemis.api.core.Message.TEXT_TYPE:
            amqMsg = new ActiveMQTextMessage();
            bytes = toAMQMessageTextType(buffer, isCompressed);
            break;
        case org.apache.activemq.artemis.api.core.Message.DEFAULT_TYPE:
            amqMsg = new ActiveMQMessage();
            bytes = toAMQMessageDefaultType(buffer, isCompressed);
            break;
        default:
            throw new IllegalStateException("Unknown message type: " + coreMessage.getType());
    }
    final String type = coreMessage.getStringProperty(JMS_TYPE_PROPERTY);
    if (type != null) {
        amqMsg.setJMSType(type);
    }
    amqMsg.setPersistent(coreMessage.isDurable());
    amqMsg.setExpiration(coreMessage.getExpiration());
    amqMsg.setPriority(coreMessage.getPriority());
    amqMsg.setTimestamp(coreMessage.getTimestamp());
    Long brokerInTime = (Long) coreMessage.getObjectProperty(AMQ_MSG_BROKER_IN_TIME);
    if (brokerInTime == null) {
        brokerInTime = 0L;
    }
    amqMsg.setBrokerInTime(brokerInTime);
    amqMsg.setCompressed(isCompressed);
    // we need check null because messages may come from other clients
    // and those amq specific attribute may not be set.
    Long arrival = (Long) coreMessage.getObjectProperty(AMQ_MSG_ARRIVAL);
    if (arrival == null) {
        // messages from other sources (like core client) may not set this prop
        arrival = 0L;
    }
    amqMsg.setArrival(arrival);
    final String brokerPath = (String) coreMessage.getObjectProperty(AMQ_MSG_BROKER_PATH);
    if (brokerPath != null && !brokerPath.isEmpty()) {
        setAMQMsgBrokerPath(amqMsg, brokerPath);
    }
    final String clusterPath = (String) coreMessage.getObjectProperty(AMQ_MSG_CLUSTER);
    if (clusterPath != null && !clusterPath.isEmpty()) {
        setAMQMsgClusterPath(amqMsg, clusterPath);
    }
    Integer commandId = (Integer) coreMessage.getObjectProperty(AMQ_MSG_COMMAND_ID);
    if (commandId == null) {
        commandId = -1;
    }
    amqMsg.setCommandId(commandId);
    final SimpleString corrId = (SimpleString) coreMessage.getObjectProperty(JMS_CORRELATION_ID_PROPERTY);
    if (corrId != null) {
        amqMsg.setCorrelationId(corrId.toString());
    }
    final byte[] dsBytes = (byte[]) coreMessage.getObjectProperty(AMQ_MSG_DATASTRUCTURE);
    if (dsBytes != null) {
        setAMQMsgDataStructure(amqMsg, marshaller, dsBytes);
    }
    final ActiveMQDestination actualDestination = consumer.getOpenwireDestination();
    amqMsg.setDestination(OpenWireUtil.toAMQAddress(coreMessage, actualDestination));
    final Object value = coreMessage.getGroupID();
    if (value != null) {
        String groupId = value.toString();
        amqMsg.setGroupID(groupId);
    }
    Integer groupSequence = (Integer) coreMessage.getObjectProperty(AMQ_MSG_GROUP_SEQUENCE);
    if (groupSequence == null) {
        groupSequence = -1;
    }
    amqMsg.setGroupSequence(groupSequence);
    final MessageId mid;
    final byte[] midBytes = (byte[]) coreMessage.getObjectProperty(AMQ_MSG_MESSAGE_ID);
    if (midBytes != null) {
        ByteSequence midSeq = new ByteSequence(midBytes);
        mid = (MessageId) marshaller.unmarshal(midSeq);
    } else {
        mid = new MessageId(UUIDGenerator.getInstance().generateStringUUID() + ":-1");
    }
    amqMsg.setMessageId(mid);
    final byte[] origDestBytes = (byte[]) coreMessage.getObjectProperty(AMQ_MSG_ORIG_DESTINATION);
    if (origDestBytes != null) {
        setAMQMsgOriginalDestination(amqMsg, marshaller, origDestBytes);
    }
    final byte[] origTxIdBytes = (byte[]) coreMessage.getObjectProperty(AMQ_MSG_ORIG_TXID);
    if (origTxIdBytes != null) {
        setAMQMsgOriginalTransactionId(amqMsg, marshaller, origTxIdBytes);
    }
    final byte[] producerIdBytes = (byte[]) coreMessage.getObjectProperty(AMQ_MSG_PRODUCER_ID);
    if (producerIdBytes != null) {
        ProducerId producerId = (ProducerId) marshaller.unmarshal(new ByteSequence(producerIdBytes));
        amqMsg.setProducerId(producerId);
    }
    final byte[] marshalledBytes = (byte[]) coreMessage.getObjectProperty(AMQ_MSG_MARSHALL_PROP);
    if (marshalledBytes != null) {
        amqMsg.setMarshalledProperties(new ByteSequence(marshalledBytes));
    }
    amqMsg.setRedeliveryCounter(reference.getDeliveryCount() - 1);
    final byte[] replyToBytes = (byte[]) coreMessage.getObjectProperty(AMQ_MSG_REPLY_TO);
    if (replyToBytes != null) {
        setAMQMsgReplyTo(amqMsg, marshaller, replyToBytes);
    }
    final String userId = (String) coreMessage.getObjectProperty(AMQ_MSG_USER_ID);
    if (userId != null) {
        amqMsg.setUserID(userId);
    }
    final Boolean isDroppable = (Boolean) coreMessage.getObjectProperty(AMQ_MSG_DROPPABLE);
    if (isDroppable != null) {
        amqMsg.setDroppable(isDroppable);
    }
    final SimpleString dlqCause = (SimpleString) coreMessage.getObjectProperty(AMQ_MSG_DLQ_DELIVERY_FAILURE_CAUSE_PROPERTY);
    if (dlqCause != null) {
        setAMQMsgDlqDeliveryFailureCause(amqMsg, dlqCause);
    }
    final SimpleString lastValueProperty = coreMessage.getLastValueProperty();
    if (lastValueProperty != null) {
        setAMQMsgHdrLastValueName(amqMsg, lastValueProperty);
    }
    final Set<SimpleString> props = coreMessage.getPropertyNames();
    if (props != null) {
        setAMQMsgObjectProperties(amqMsg, coreMessage, props, consumer);
    }
    if (bytes != null) {
        ByteSequence content = new ByteSequence(bytes);
        amqMsg.setContent(content);
    }
    return amqMsg;
}
Also used : ActiveMQMapMessage(org.apache.activemq.command.ActiveMQMapMessage) ProducerId(org.apache.activemq.command.ProducerId) ActiveMQObjectMessage(org.apache.activemq.command.ActiveMQObjectMessage) SimpleString(org.apache.activemq.artemis.api.core.SimpleString) ActiveMQStreamMessage(org.apache.activemq.command.ActiveMQStreamMessage) SimpleString(org.apache.activemq.artemis.api.core.SimpleString) ActiveMQMessage(org.apache.activemq.command.ActiveMQMessage) ActiveMQTextMessage(org.apache.activemq.command.ActiveMQTextMessage) ActiveMQDestination(org.apache.activemq.command.ActiveMQDestination) ActiveMQBytesMessage(org.apache.activemq.command.ActiveMQBytesMessage) ByteSequence(org.apache.activemq.util.ByteSequence) ActiveMQBuffer(org.apache.activemq.artemis.api.core.ActiveMQBuffer) MessageId(org.apache.activemq.command.MessageId)

Example 3 with ActiveMQBytesMessage

use of org.apache.activemq.command.ActiveMQBytesMessage in project activemq-artemis by apache.

the class MessageTransformationTest method testTransformMessage.

/**
 * Tests transforming messages into ActiveMQ's message implementation.
 */
public void testTransformMessage() throws Exception {
    assertTrue("Transforming a BytesMessage message into an ActiveMQBytesMessage", ActiveMQMessageTransformation.transformMessage(new ActiveMQBytesMessage(), null) instanceof ActiveMQBytesMessage);
    assertTrue("Transforming a MapMessage message to an ActiveMQMapMessage", ActiveMQMessageTransformation.transformMessage(new ActiveMQMapMessage(), null) instanceof ActiveMQMapMessage);
    assertTrue("Transforming an ObjectMessage message to an ActiveMQObjectMessage", ActiveMQMessageTransformation.transformMessage(new ActiveMQObjectMessage(), null) instanceof ActiveMQObjectMessage);
    assertTrue("Transforming a StreamMessage message to an ActiveMQStreamMessage", ActiveMQMessageTransformation.transformMessage(new ActiveMQStreamMessage(), null) instanceof ActiveMQStreamMessage);
    assertTrue("Transforming a TextMessage message to an ActiveMQTextMessage", ActiveMQMessageTransformation.transformMessage(new ActiveMQTextMessage(), null) instanceof ActiveMQTextMessage);
    assertTrue("Transforming an ActiveMQMessage message to an ActiveMQMessage", ActiveMQMessageTransformation.transformMessage(new ActiveMQMessage(), null) instanceof ActiveMQMessage);
}
Also used : ActiveMQMapMessage(org.apache.activemq.command.ActiveMQMapMessage) ActiveMQBytesMessage(org.apache.activemq.command.ActiveMQBytesMessage) ActiveMQObjectMessage(org.apache.activemq.command.ActiveMQObjectMessage) ActiveMQStreamMessage(org.apache.activemq.command.ActiveMQStreamMessage) ActiveMQTextMessage(org.apache.activemq.command.ActiveMQTextMessage) ActiveMQMessage(org.apache.activemq.command.ActiveMQMessage)

Example 4 with ActiveMQBytesMessage

use of org.apache.activemq.command.ActiveMQBytesMessage in project activemq-artemis by apache.

the class ActiveMQBytesMessageTest method createObject.

@Override
public Object createObject() throws Exception {
    ActiveMQBytesMessage info = new ActiveMQBytesMessage();
    populateObject(info);
    return info;
}
Also used : ActiveMQBytesMessage(org.apache.activemq.command.ActiveMQBytesMessage)

Example 5 with ActiveMQBytesMessage

use of org.apache.activemq.command.ActiveMQBytesMessage in project activemq-artemis by apache.

the class OpenTypeSupportTest method testBrowseByteMessageFails.

@Test
public void testBrowseByteMessageFails() throws Exception {
    ActiveMQBytesMessage bm = new ActiveMQBytesMessage();
    bm.writeBytes("123456".getBytes());
    Object result = OpenTypeSupport.convert(bm);
    LOG.info("result : " + result);
}
Also used : ActiveMQBytesMessage(org.apache.activemq.command.ActiveMQBytesMessage) Test(org.junit.Test)

Aggregations

ActiveMQBytesMessage (org.apache.activemq.command.ActiveMQBytesMessage)9 ActiveMQMapMessage (org.apache.activemq.command.ActiveMQMapMessage)3 ActiveMQObjectMessage (org.apache.activemq.command.ActiveMQObjectMessage)3 ActiveMQStreamMessage (org.apache.activemq.command.ActiveMQStreamMessage)3 ActiveMQTextMessage (org.apache.activemq.command.ActiveMQTextMessage)3 Test (org.junit.Test)3 BytesMessage (javax.jms.BytesMessage)2 ActiveMQMessage (org.apache.activemq.command.ActiveMQMessage)2 IOException (java.io.IOException)1 InputStream (java.io.InputStream)1 MapMessage (javax.jms.MapMessage)1 Message (javax.jms.Message)1 MessageConsumer (javax.jms.MessageConsumer)1 MessageProducer (javax.jms.MessageProducer)1 StreamMessage (javax.jms.StreamMessage)1 ActiveMQBuffer (org.apache.activemq.artemis.api.core.ActiveMQBuffer)1 SimpleString (org.apache.activemq.artemis.api.core.SimpleString)1 ActiveMQDestination (org.apache.activemq.command.ActiveMQDestination)1 MessageId (org.apache.activemq.command.MessageId)1 ProducerId (org.apache.activemq.command.ProducerId)1