commit ba0bc52673f90eb17dc41c815f90c2c729f74942
author Jiangnan Jia <jiajiangnan.hz@foxmail.com> Mon Feb 07 17:14:06 2022 +0800
committer GitHub <noreply@github.com> Mon Feb 07 17:14:06 2022 +0800
tree 0cb475bd04fc21e3fc9dce1791a05b03383ce9c2
parent 91d7287b1067af77f6d579026170fda44ae1af82

[ISSUE #411] Support DELAY and WAIT properties in RocketMQHeaders.java, which can convert Spring-Message to Rocket-Message conveniently #

rocketmq-spring-boot/src/main/java/org/apache/rocketmq/spring/support/RocketMQHeaders.java

diff --git a/rocketmq-spring-boot/src/main/java/org/apache/rocketmq/spring/support/RocketMQHeaders.java b/rocketmq-spring-boot/src/main/java/org/apache/rocketmq/spring/support/RocketMQHeaders.java
index 7afefbd..8c49809 100644
--- a/rocketmq-spring-boot/src/main/java/org/apache/rocketmq/spring/support/RocketMQHeaders.java
+++ b/rocketmq-spring-boot/src/main/java/org/apache/rocketmq/spring/support/RocketMQHeaders.java
@@ -31,4 +31,6 @@
     public static final String QUEUE_ID = "QUEUE_ID";
     public static final String SYS_FLAG = "SYS_FLAG";
     public static final String TRANSACTION_ID = "TRANSACTION_ID";
+    public static final String DELAY = "DELAY";
+    public static final String WAIT = "WAIT";
 }

rocketmq-spring-boot/src/main/java/org/apache/rocketmq/spring/support/RocketMQUtil.java

diff --git a/rocketmq-spring-boot/src/main/java/org/apache/rocketmq/spring/support/RocketMQUtil.java b/rocketmq-spring-boot/src/main/java/org/apache/rocketmq/spring/support/RocketMQUtil.java
index a5b3b1b..e598363 100644
--- a/rocketmq-spring-boot/src/main/java/org/apache/rocketmq/spring/support/RocketMQUtil.java
+++ b/rocketmq-spring-boot/src/main/java/org/apache/rocketmq/spring/support/RocketMQUtil.java
@@ -105,7 +105,9 @@
                 setHeader(toRocketHeaderKey(RocketMQHeaders.FLAG), message.getFlag()).
                 setHeader(toRocketHeaderKey(RocketMQHeaders.QUEUE_ID), message.getQueueId()).
                 setHeader(toRocketHeaderKey(RocketMQHeaders.SYS_FLAG), message.getSysFlag()).
-                setHeader(toRocketHeaderKey(RocketMQHeaders.TRANSACTION_ID), message.getTransactionId());
+                setHeader(toRocketHeaderKey(RocketMQHeaders.TRANSACTION_ID), message.getTransactionId()).
+                setHeader(toRocketHeaderKey(RocketMQHeaders.DELAY), message.getDelayTimeLevel()).
+                setHeader(toRocketHeaderKey(RocketMQHeaders.WAIT), message.isWaitStoreMsgOK());
         addUserProperties(message.getProperties(), messageBuilder);
         return messageBuilder.build();
     }