Skip to content

Commit 06e22b4

Browse files
landonchan90landonchan90
and
landonchan90
authored
[ISSUE #7765] Fix unit test testEstimateLag
Co-authored-by: landonchan90 <[email protected]>
1 parent f70862e commit 06e22b4

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

test/src/test/java/org/apache/rocketmq/test/offset/LagCalculationIT.java

+8-1
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
import org.apache.rocketmq.broker.filter.ExpressionMessageFilter;
2727
import org.apache.rocketmq.client.consumer.MessageSelector;
2828
import org.apache.rocketmq.common.Pair;
29+
import org.apache.rocketmq.common.filter.ExpressionType;
2930
import org.apache.rocketmq.common.message.Message;
3031
import org.apache.rocketmq.common.message.MessageQueue;
3132
import org.apache.rocketmq.logging.org.slf4j.Logger;
@@ -171,6 +172,13 @@ public void testEstimateLag() throws Exception {
171172
RMQSqlConsumer sqlConsumer = ConsumerFactory.getRMQSqlConsumer(NAMESRV_ADDR, initConsumerGroup(), topic, selector, sqlListener);
172173
RMQBlockListener tagListener = new RMQBlockListener(true);
173174
RMQNormalConsumer tagConsumer = getConsumer(NAMESRV_ADDR, topic, tag, tagListener);
175+
176+
//init subscriptionData & consumerFilterData for sql
177+
SubscriptionData subscriptionData = FilterAPI.buildSubscriptionData(topic, sql, ExpressionType.SQL92);
178+
for (BrokerController controller : brokerControllerList) {
179+
controller.getConsumerFilterManager().register(topic, sqlConsumer.getConsumerGroup(), sql, ExpressionType.SQL92, subscriptionData.getSubVersion());
180+
}
181+
174182
// wait for building filter data
175183
await().atMost(5, TimeUnit.SECONDS).until(() -> sqlListener.isBlocked() && tagListener.isBlocked());
176184

@@ -210,7 +218,6 @@ public void testEstimateLag() throws Exception {
210218
for (MessageQueue mq : mqs) {
211219
if (mq.getBrokerName().equals(controller.getBrokerConfig().getBrokerName())) {
212220
long brokerOffset = controller.getMessageStore().getMaxOffsetInQueue(topic, mq.getQueueId());
213-
SubscriptionData subscriptionData = controller.getConsumerManager().findSubscriptionData(sqlConsumer.getConsumerGroup(), topic);
214221
ConsumerFilterData consumerFilterData = controller.getConsumerFilterManager().get(topic, sqlConsumer.getConsumerGroup());
215222
long estimateMessageCount = controller.getMessageStore()
216223
.estimateMessageCount(topic, mq.getQueueId(), 0, brokerOffset,

0 commit comments

Comments
 (0)