Skip to content

Commit fba96e8

Browse files
authored
fix(example) fix example: RocketMQSourceExample1 (#100)
Co-authored-by: 维章 <[email protected]>
1 parent d0d88fb commit fba96e8

File tree

1 file changed

+21
-5
lines changed
  • rocketmq-streams-channel-rocketmq/src/main/java/org/apache/rocketmq/streams/source

1 file changed

+21
-5
lines changed

rocketmq-streams-channel-rocketmq/src/main/java/org/apache/rocketmq/streams/source/RocketMQSource.java

Lines changed: 21 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -287,13 +287,17 @@ protected void setOffsetStore(DefaultMQPushConsumer consumer) {
287287
}
288288
MQClientInstance mQClientFactory = MQClientManager.getInstance().getAndCreateMQClientInstance(defaultMQPushConsumer.getDefaultMQPushConsumer());
289289
RemoteBrokerOffsetStore offsetStore = new RemoteBrokerOffsetStore(mQClientFactory, NamespaceUtil.wrapNamespace(consumer.getNamespace(), consumer.getConsumerGroup())) {
290-
290+
Set<MessageQueue> firstComing = new HashSet<>();
291291
@Override
292292
public void removeOffset(MessageQueue mq) {
293-
Set<String> splitIds = new HashSet<>();
294-
splitIds.add(new RocketMQMessageQueue(mq).getQueueId());
295-
removeSplit(splitIds);
296-
super.removeOffset(mq);
293+
if (!firstComing.contains(mq)){
294+
firstComing.add(mq);
295+
} else {
296+
Set<String> splitIds = new HashSet<>();
297+
splitIds.add(new RocketMQMessageQueue(mq).getQueueId());
298+
removeSplit(splitIds);
299+
super.removeOffset(mq);
300+
}
297301
}
298302

299303
@Override
@@ -398,4 +402,16 @@ public String getConsumerOffset() {
398402
public void setConsumerOffset(String consumerOffset) {
399403
this.consumerOffset = consumerOffset;
400404
}
405+
406+
public String getStrategyName() {
407+
return strategyName;
408+
}
409+
410+
public void setStrategyName(String strategyName) {
411+
this.strategyName = strategyName;
412+
}
413+
414+
public void setConsumer(DefaultMQPushConsumer consumer) {
415+
this.consumer = consumer;
416+
}
401417
}

0 commit comments

Comments
 (0)