diff --git a/message/message-boot/src/main/java/com/njcn/message/consumer/FrontDataConsumer.java b/message/message-boot/src/main/java/com/njcn/message/consumer/FrontDataConsumer.java index 61f8988..b1c22f8 100644 --- a/message/message-boot/src/main/java/com/njcn/message/consumer/FrontDataConsumer.java +++ b/message/message-boot/src/main/java/com/njcn/message/consumer/FrontDataConsumer.java @@ -16,8 +16,10 @@ import lombok.extern.slf4j.Slf4j; import org.apache.rocketmq.spring.annotation.RocketMQMessageListener; import org.apache.rocketmq.spring.core.RocketMQListener; import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.beans.factory.annotation.Value; import org.springframework.stereotype.Component; +import javax.annotation.PostConstruct; import javax.annotation.Resource; import java.util.ArrayList; import java.util.List; @@ -45,7 +47,8 @@ public class FrontDataConsumer extends EnhanceConsumerMessageHandler messageList = new ArrayList<>(1); - + private List messageList = new ArrayList<>(); + @PostConstruct + public void validateConfig() { + if (consumerSize == null) { + throw new IllegalStateException("rocketmq.consumer_size 未配置!"); + } + this.messageList = new ArrayList<>(consumerSize); + } @Override public void onMessage(String baseMessage) { MessageDataDTO messageDataDTO = JSONObject.parseObject(baseMessage,MessageDataDTO.class); @@ -92,7 +101,7 @@ public class FrontDataConsumer extends EnhanceConsumerMessageHandler= 1) { + if (messageList.size() >= consumerSize) { saveToDatabase(); } } @@ -159,7 +168,7 @@ public class FrontDataConsumer extends EnhanceConsumerMessageHandler