|
|
|
@ -110,14 +110,15 @@ public class RocketMQMessageSource extends AbstractMessageSource<Object>
|
|
|
|
|
this.running = true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private MessageQueue acquireCurrentMessageQueue(String topic,int queueId, String brokerName) {
|
|
|
|
|
Collection<MessageQueue> messageQueueSet = messageQueuesForTopic.get(topic);
|
|
|
|
|
if(CollectionUtils.isEmpty(messageQueueSet)){
|
|
|
|
|
private MessageQueue acquireCurrentMessageQueue(String topic, int queueId,
|
|
|
|
|
String brokerName) {
|
|
|
|
|
Collection<MessageQueue> messageQueueSet = messageQueuesForTopic.get(topic);
|
|
|
|
|
if (CollectionUtils.isEmpty(messageQueueSet)) {
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
for (MessageQueue messageQueue : messageQueueSet) {
|
|
|
|
|
if (messageQueue.getQueueId() == queueId &&
|
|
|
|
|
ObjectUtils.nullSafeEquals(brokerName, messageQueue.getBrokerName())) {
|
|
|
|
|
if (messageQueue.getQueueId() == queueId && ObjectUtils
|
|
|
|
|
.nullSafeEquals(brokerName, messageQueue.getBrokerName())) {
|
|
|
|
|
return messageQueue;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
@ -154,8 +155,8 @@ public class RocketMQMessageSource extends AbstractMessageSource<Object>
|
|
|
|
|
if (null == messageExt) {
|
|
|
|
|
return null;
|
|
|
|
|
}
|
|
|
|
|
MessageQueue messageQueue = this.acquireCurrentMessageQueue(messageExt.getTopic(), messageExt.getQueueId(),
|
|
|
|
|
messageExt.getBrokerName());
|
|
|
|
|
MessageQueue messageQueue = this.acquireCurrentMessageQueue(messageExt.getTopic(),
|
|
|
|
|
messageExt.getQueueId(), messageExt.getBrokerName());
|
|
|
|
|
if (messageQueue == null) {
|
|
|
|
|
throw new IllegalArgumentException(
|
|
|
|
|
"The message queue is not in assigned list");
|
|
|
|
|