Examples with BytesMessage io.openmessaging.BytesMessage used on opensource projects

Example 6 with BytesMessage

use of io.openmessaging.BytesMessage in project rocketmq by apache.

the class SequenceProducerImplTest method testRollback.

@Test
public void testRollback() {
    when(rocketmqProducer.getMaxMessageSize()).thenReturn(1024);
    final BytesMessage message = producer.createBytesMessageToTopic("HELLO_TOPIC", new byte[] { 'a' });
    producer.send(message);
    producer.rollback();
    // Commit nothing.
    producer.commit();
    assertThat(message.headers().getString(MessageHeader.MESSAGE_ID)).isEqualTo(null);
}

Example 7 with BytesMessage

use of io.openmessaging.BytesMessage in project rocketmq-rocketmq-all-4.1.0-incubating by lirenzuo.

the class SequenceProducerImplTest method testSend_WithCommit.

@Test
public void testSend_WithCommit() throws InterruptedException, RemotingException, MQClientException, MQBrokerException {
    SendResult sendResult = new SendResult();
    sendResult.setMsgId("TestMsgID");
    sendResult.setSendStatus(SendStatus.SEND_OK);
    when(rocketmqProducer.send(ArgumentMatchers.<Message>anyList())).thenReturn(sendResult);
    when(rocketmqProducer.getMaxMessageSize()).thenReturn(1024);
    final BytesMessage message = producer.createBytesMessageToTopic("HELLO_TOPIC", new byte[] { 'a' });
    producer.send(message);
    producer.commit();
    assertThat(message.headers().getString(MessageHeader.MESSAGE_ID)).isEqualTo("TestMsgID");
}

Example 8 with BytesMessage

use of io.openmessaging.BytesMessage in project rocketmq-rocketmq-all-4.1.0-incubating by lirenzuo.

the class AbstractOMSProducer method createBytesMessageToTopic.

@Override
public BytesMessage createBytesMessageToTopic(final String topic, final byte[] body) {
    BytesMessage bytesMessage = new BytesMessageImpl();
    bytesMessage.setBody(body);
    bytesMessage.headers().put(MessageHeader.TOPIC, topic);
    return bytesMessage;
}

Example 9 with BytesMessage

use of io.openmessaging.BytesMessage in project rocketmq-rocketmq-all-4.1.0-incubating by lirenzuo.

the class SequenceProducerImpl method commit.

@Override
public synchronized void commit() {
    List<Message> messages = new ArrayList<>();
    msgCacheQueue.drainTo(messages);
    List<org.apache.rocketmq.common.message.Message> rmqMessages = new ArrayList<>();
    for (Message message : messages) {
        rmqMessages.add(OMSUtil.msgConvert((BytesMessage) message));
    }
    if (rmqMessages.size() == 0) {
        return;
    }
    try {
        SendResult sendResult = this.rocketmqProducer.send(rmqMessages);
        String[] msgIdArray = sendResult.getMsgId().split(",");
        for (int i = 0; i < messages.size(); i++) {
            Message message = messages.get(i);
            message.headers().put(MessageHeader.MESSAGE_ID, msgIdArray[i]);
        }
    } catch (Exception e) {
        throw checkProducerException("", "", e);
    }
}

Example 10 with BytesMessage

use of io.openmessaging.BytesMessage in project rocketmq-rocketmq-all-4.1.0-incubating by lirenzuo.

the class OMSUtil method msgConvert.

public static org.apache.rocketmq.common.message.Message msgConvert(BytesMessage omsMessage) {
    org.apache.rocketmq.common.message.Message rmqMessage = new org.apache.rocketmq.common.message.Message();
    rmqMessage.setBody(omsMessage.getBody());
    KeyValue headers = omsMessage.headers();
    KeyValue properties = omsMessage.properties();
    // All destinations in RocketMQ use Topic
    if (headers.containsKey(MessageHeader.TOPIC)) {
        rmqMessage.setTopic(headers.getString(MessageHeader.TOPIC));
        rmqMessage.putUserProperty(NonStandardKeys.MESSAGE_DESTINATION, "TOPIC");
    } else {
        rmqMessage.setTopic(headers.getString(MessageHeader.QUEUE));
        rmqMessage.putUserProperty(NonStandardKeys.MESSAGE_DESTINATION, "QUEUE");
    }
    for (String key : properties.keySet()) {
        MessageAccessor.putProperty(rmqMessage, key, properties.getString(key));
    }
    // Headers has a high priority
    for (String key : headers.keySet()) {
        MessageAccessor.putProperty(rmqMessage, key, headers.getString(key));
    }
    return rmqMessage;
}