初始化v1.2

2025-11-24 16:19:09 +08:00
parent c0f51ef007
commit de49404116
918 changed files with 4236 additions and 4239 deletions
@@ -0,0 +1,94 @@
+package com.tongran.mtragent.consumer;
+
+import com.alibaba.fastjson.JSON;
+import com.tongran.mtragent.domain.DeviceMessage;
+import com.tongran.mtragent.enums.MessageCodeEnum;
+import com.tongran.mtragent.handler.MessageHandler;
+import com.tongran.mtragent.producer.ConsumeException;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyContext;
+import org.apache.rocketmq.client.consumer.listener.ConsumeConcurrentlyStatus;
+import org.apache.rocketmq.client.consumer.listener.MessageListenerConcurrently;
+import org.apache.rocketmq.common.message.MessageExt;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+import org.springframework.util.CollectionUtils;
+
+import java.io.UnsupportedEncodingException;
+import java.util.List;
+
+/**
+ * 消息监听
+ */
+@Slf4j
+@Component
+public class RocketMsgListener implements MessageListenerConcurrently {
+    @Autowired
+    private MessageHandler messageHandler;
+
+    /**
+     * 消费消息
+     * @param list    msgs.size() >= 1
+     *                DefaultMQPushConsumer.consumeMessageBatchMaxSize=1，you can modify here
+     *                这里只设置为1，当设置为多个时，list中只要有一条消息消费失败，就会整体重试
+     * @param consumeConcurrentlyContext 上下文信息
+     * @return 消费状态  成功（CONSUME_SUCCESS）或者 重试 (RECONSUME_LATER)
+     */
+    @Override
+    public ConsumeConcurrentlyStatus consumeMessage(List<MessageExt> list, ConsumeConcurrentlyContext consumeConcurrentlyContext) {
+       try{
+           //消息不等于空情况
+           if (!CollectionUtils.isEmpty(list)) {
+               //获取topic
+               for (MessageExt messageExt : list) {
+                   // 解析消息内容
+                   String body = new String(messageExt.getBody());
+                   log.info("接受到的消息为：{}", body);
+                   String tags = messageExt.getTags();
+                   String topic = messageExt.getTopic();
+                   String msgId = messageExt.getMsgId();
+                   String keys = messageExt.getKeys();
+                   int reConsume = messageExt.getReconsumeTimes();
+                   // 消息已经重试了3次，如果不需要再次消费，则返回成功
+                   if (reConsume == 3) {
+                       // TODO 补偿信息
+                       log.error("消息消费三次失败，消息内容：{}", body);
+                       return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;//根据业务返回是否正常
+                   }
+                   if(MessageCodeEnum.TR_MTRAGENT_UP.getCode().equals(topic)){
+                       // 拿到信息
+                       DeviceMessage message = JSON.parseObject(body, DeviceMessage.class);
+                       // 处理消息
+                       messageHandler.handleMessage(message);
+                       return ConsumeConcurrentlyStatus.CONSUME_SUCCESS;//业务处理成功
+                   }
+                   // 根据不同的topic处理不同的业务 这里以订单消息为例子
+               }
+           }
+           // 消息消费失败
+           //broker会根据设置的messageDelayLevel发起重试，默认16次
+           return ConsumeConcurrentlyStatus.RECONSUME_LATER;
+       } catch (Exception e) {
+           // 调用 handleException 方法处理异常并返回处理结果
+           return handleException(e);
+       }
+    }
+
+    /**
+     * 异常处理
+     *
+     * @param e 捕获的异常
+     * @return 消息消费结果
+     */
+    private static ConsumeConcurrentlyStatus handleException(final Exception e) {
+        Class exceptionClass = e.getClass();
+        if (exceptionClass.equals(UnsupportedEncodingException.class)) {
+            log.error(e.getMessage());
+        } else if (exceptionClass.equals(ConsumeException.class)) {
+            log.error(e.getMessage());
+        } else{
+            log.error(e.getMessage());
+        }
+        return ConsumeConcurrentlyStatus.RECONSUME_LATER;
+    }
+}
@@ -0,0 +1,30 @@
+package com.tongran.mtragent.consumer;
+
+import org.apache.rocketmq.client.producer.LocalTransactionState;
+import org.apache.rocketmq.client.producer.TransactionListener;
+import org.apache.rocketmq.common.message.Message;
+import org.apache.rocketmq.common.message.MessageExt;
+
+/**
+ * 事物消息监听
+ */
+public class RocketMsgTransactionListenerImpl implements TransactionListener {
+
+    @Override
+    public LocalTransactionState executeLocalTransaction(Message msg, Object arg) {
+        // 在这里执行本地事务，比如数据库操作等
+        // 如果本地事务执行成功，返回 COMMIT_MESSAGE
+        // 如果本地事务执行失败，返回 ROLLBACK_MESSAGE
+        // 如果本地事务执行中，可以返回 UNKNOW，RocketMQ 将会检查事务状态，并根据状态处理消息
+        return LocalTransactionState.COMMIT_MESSAGE; // 根据实际情况返回对应的状态
+    }
+
+    @Override
+    public LocalTransactionState checkLocalTransaction(MessageExt msg) {
+        // 检查本地事务状态，如果本地事务执行成功，返回 COMMIT_MESSAGE
+        // 如果本地事务执行失败，返回 ROLLBACK_MESSAGE
+        // 如果本地事务仍在执行中，返回 UNKNOW，RocketMQ 将会继续检查事务状态
+        return LocalTransactionState.COMMIT_MESSAGE; // 根据实际情况返回对应的状态
+    }
+
+}