提交 b421d48c 编写于 作者: Z Zhanhui Li
上级 9cb0a0cd
...@@ -23,6 +23,9 @@ import java.util.List; ...@@ -23,6 +23,9 @@ import java.util.List;
import java.util.Set; import java.util.Set;
import java.util.concurrent.CountDownLatch; import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit; import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;
import org.apache.rocketmq.client.QueryResult; import org.apache.rocketmq.client.QueryResult;
import org.apache.rocketmq.client.exception.MQBrokerException; import org.apache.rocketmq.client.exception.MQBrokerException;
import org.apache.rocketmq.client.exception.MQClientException; import org.apache.rocketmq.client.exception.MQClientException;
...@@ -285,6 +288,7 @@ public class MQAdminImpl { ...@@ -285,6 +288,7 @@ public class MQAdminImpl {
if (!brokerAddrs.isEmpty()) { if (!brokerAddrs.isEmpty()) {
final CountDownLatch countDownLatch = new CountDownLatch(brokerAddrs.size()); final CountDownLatch countDownLatch = new CountDownLatch(brokerAddrs.size());
final List<QueryResult> queryResultList = new LinkedList<QueryResult>(); final List<QueryResult> queryResultList = new LinkedList<QueryResult>();
final ReadWriteLock lock = new ReentrantReadWriteLock(false);
for (String addr : brokerAddrs) { for (String addr : brokerAddrs) {
try { try {
...@@ -318,7 +322,12 @@ public class MQAdminImpl { ...@@ -318,7 +322,12 @@ public class MQAdminImpl {
MessageDecoder.decodes(ByteBuffer.wrap(response.getBody()), true); MessageDecoder.decodes(ByteBuffer.wrap(response.getBody()), true);
QueryResult qr = new QueryResult(responseHeader.getIndexLastUpdateTimestamp(), wrappers); QueryResult qr = new QueryResult(responseHeader.getIndexLastUpdateTimestamp(), wrappers);
queryResultList.add(qr); try {
lock.writeLock().lock();
queryResultList.add(qr);
} finally {
lock.writeLock().unlock();
}
break; break;
} }
default: default:
......
...@@ -69,7 +69,7 @@ public class QueryMsgByKeySubCommand implements SubCommand { ...@@ -69,7 +69,7 @@ public class QueryMsgByKeySubCommand implements SubCommand {
} }
} }
void queryByKey(final DefaultMQAdminExt admin, final String topic, final String key) private void queryByKey(final DefaultMQAdminExt admin, final String topic, final String key)
throws MQClientException, InterruptedException { throws MQClientException, InterruptedException {
admin.start(); admin.start();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册