org.apache.rocketmq.proxy.service.mqclient.MQClientAPIExt Maven / Gradle / Ivy
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.rocketmq.proxy.service.mqclient;
import java.nio.ByteBuffer;
import java.util.Collections;
import java.util.List;
import java.util.Set;
import java.util.concurrent.CompletableFuture;
import org.apache.rocketmq.client.ClientConfig;
import org.apache.rocketmq.client.consumer.AckCallback;
import org.apache.rocketmq.client.consumer.AckResult;
import org.apache.rocketmq.client.consumer.PopCallback;
import org.apache.rocketmq.client.consumer.PopResult;
import org.apache.rocketmq.client.consumer.PullCallback;
import org.apache.rocketmq.client.consumer.PullResult;
import org.apache.rocketmq.client.consumer.PullStatus;
import org.apache.rocketmq.client.exception.MQBrokerException;
import org.apache.rocketmq.client.exception.MQClientException;
import org.apache.rocketmq.client.exception.OffsetNotFoundException;
import org.apache.rocketmq.client.impl.ClientRemotingProcessor;
import org.apache.rocketmq.client.impl.CommunicationMode;
import org.apache.rocketmq.client.impl.MQClientAPIImpl;
import org.apache.rocketmq.client.impl.consumer.PullResultExt;
import org.apache.rocketmq.client.producer.SendResult;
import org.apache.rocketmq.common.constant.LoggerName;
import org.apache.rocketmq.common.message.Message;
import org.apache.rocketmq.common.message.MessageBatch;
import org.apache.rocketmq.common.message.MessageClientIDSetter;
import org.apache.rocketmq.common.message.MessageDecoder;
import org.apache.rocketmq.common.message.MessageExt;
import org.apache.rocketmq.common.message.MessageQueue;
import org.apache.rocketmq.common.protocol.RequestCode;
import org.apache.rocketmq.common.protocol.ResponseCode;
import org.apache.rocketmq.common.protocol.body.LockBatchRequestBody;
import org.apache.rocketmq.common.protocol.body.LockBatchResponseBody;
import org.apache.rocketmq.common.protocol.body.UnlockBatchRequestBody;
import org.apache.rocketmq.common.protocol.header.AckMessageRequestHeader;
import org.apache.rocketmq.common.protocol.header.ChangeInvisibleTimeRequestHeader;
import org.apache.rocketmq.common.protocol.header.ConsumerSendMsgBackRequestHeader;
import org.apache.rocketmq.common.protocol.header.GetConsumerListByGroupRequestHeader;
import org.apache.rocketmq.common.protocol.header.GetConsumerListByGroupResponseBody;
import org.apache.rocketmq.common.protocol.header.GetMaxOffsetRequestHeader;
import org.apache.rocketmq.common.protocol.header.GetMaxOffsetResponseHeader;
import org.apache.rocketmq.common.protocol.header.GetMinOffsetRequestHeader;
import org.apache.rocketmq.common.protocol.header.GetMinOffsetResponseHeader;
import org.apache.rocketmq.common.protocol.header.PopMessageRequestHeader;
import org.apache.rocketmq.common.protocol.header.PullMessageRequestHeader;
import org.apache.rocketmq.common.protocol.header.QueryConsumerOffsetRequestHeader;
import org.apache.rocketmq.common.protocol.header.QueryConsumerOffsetResponseHeader;
import org.apache.rocketmq.common.protocol.header.SearchOffsetRequestHeader;
import org.apache.rocketmq.common.protocol.header.SearchOffsetResponseHeader;
import org.apache.rocketmq.common.protocol.header.SendMessageRequestHeader;
import org.apache.rocketmq.common.protocol.header.SendMessageRequestHeaderV2;
import org.apache.rocketmq.common.protocol.header.UpdateConsumerOffsetRequestHeader;
import org.apache.rocketmq.common.protocol.heartbeat.HeartbeatData;
import org.apache.rocketmq.logging.InternalLogger;
import org.apache.rocketmq.logging.InternalLoggerFactory;
import org.apache.rocketmq.remoting.RPCHook;
import org.apache.rocketmq.remoting.exception.RemotingCommandException;
import org.apache.rocketmq.remoting.netty.NettyClientConfig;
import org.apache.rocketmq.remoting.netty.ResponseFuture;
import org.apache.rocketmq.remoting.protocol.RemotingCommand;
public class MQClientAPIExt extends MQClientAPIImpl {
private static final InternalLogger log = InternalLoggerFactory.getLogger(LoggerName.PROXY_LOGGER_NAME);
private final ClientConfig clientConfig;
public MQClientAPIExt(
ClientConfig clientConfig,
NettyClientConfig nettyClientConfig,
ClientRemotingProcessor clientRemotingProcessor,
RPCHook rpcHook
) {
super(nettyClientConfig, clientRemotingProcessor, rpcHook, clientConfig);
this.clientConfig = clientConfig;
}
public boolean updateNameServerAddressList() {
if (this.clientConfig.getNamesrvAddr() != null) {
this.updateNameServerAddressList(this.clientConfig.getNamesrvAddr());
log.info("user specified name server address: {}", this.clientConfig.getNamesrvAddr());
return true;
}
return false;
}
protected static MQClientException processNullResponseErr(ResponseFuture responseFuture) {
MQClientException ex;
if (!responseFuture.isSendRequestOK()) {
ex = new MQClientException("send request failed", responseFuture.getCause());
} else if (responseFuture.isTimeout()) {
ex = new MQClientException("wait response timeout " + responseFuture.getTimeoutMillis() + "ms",
responseFuture.getCause());
} else {
ex = new MQClientException("unknown reason", responseFuture.getCause());
}
return ex;
}
public CompletableFuture sendHeartbeatOneway(
String brokerAddr,
HeartbeatData heartbeatData,
long timeoutMillis
) {
CompletableFuture future = new CompletableFuture<>();
try {
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.HEART_BEAT, null);
request.setLanguage(clientConfig.getLanguage());
request.setBody(heartbeatData.encode());
this.getRemotingClient().invokeOneway(brokerAddr, request, timeoutMillis);
future.complete(null);
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture sendHeartbeatAsync(
String brokerAddr,
HeartbeatData heartbeatData,
long timeoutMillis
) {
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.HEART_BEAT, null);
request.setLanguage(clientConfig.getLanguage());
request.setBody(heartbeatData.encode());
CompletableFuture future = new CompletableFuture<>();
try {
this.getRemotingClient().invokeAsync(brokerAddr, request, timeoutMillis, responseFuture -> {
RemotingCommand response = responseFuture.getResponseCommand();
if (response != null) {
if (ResponseCode.SUCCESS == response.getCode()) {
future.complete(response.getVersion());
} else {
future.completeExceptionally(new MQBrokerException(response.getCode(), response.getRemark(), brokerAddr));
}
} else {
future.completeExceptionally(processNullResponseErr(responseFuture));
}
});
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture sendMessageAsync(
String brokerAddr,
String brokerName,
Message msg,
SendMessageRequestHeader requestHeader,
long timeoutMillis
) {
SendMessageRequestHeaderV2 requestHeaderV2 = SendMessageRequestHeaderV2.createSendMessageRequestHeaderV2(requestHeader);
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.SEND_MESSAGE_V2, requestHeaderV2);
request.setBody(msg.getBody());
CompletableFuture future = new CompletableFuture<>();
try {
this.getRemotingClient().invokeAsync(brokerAddr, request, timeoutMillis, responseFuture -> {
RemotingCommand response = responseFuture.getResponseCommand();
if (response != null) {
try {
future.complete(this.processSendResponse(brokerName, msg, response, brokerAddr));
} catch (Exception e) {
future.completeExceptionally(e);
}
} else {
future.completeExceptionally(processNullResponseErr(responseFuture));
}
});
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture sendMessageAsync(
String brokerAddr,
String brokerName,
List msgList,
SendMessageRequestHeader requestHeader,
long timeoutMillis
) {
SendMessageRequestHeaderV2 requestHeaderV2 = SendMessageRequestHeaderV2.createSendMessageRequestHeaderV2(requestHeader);
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.SEND_BATCH_MESSAGE, requestHeaderV2);
CompletableFuture future = new CompletableFuture<>();
try {
requestHeader.setBatch(true);
MessageBatch msgBatch = MessageBatch.generateFromList(msgList);
MessageClientIDSetter.setUniqID(msgBatch);
byte[] body = msgBatch.encode();
msgBatch.setBody(body);
request.setBody(body);
this.getRemotingClient().invokeAsync(brokerAddr, request, timeoutMillis, responseFuture -> {
RemotingCommand response = responseFuture.getResponseCommand();
if (response != null) {
try {
future.complete(this.processSendResponse(brokerName, msgBatch, response, brokerAddr));
} catch (Exception e) {
future.completeExceptionally(e);
}
} else {
future.completeExceptionally(processNullResponseErr(responseFuture));
}
});
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture sendMessageBackAsync(
String brokerAddr,
ConsumerSendMsgBackRequestHeader requestHeader,
long timeoutMillis
) {
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.CONSUMER_SEND_MSG_BACK, requestHeader);
CompletableFuture future = new CompletableFuture<>();
try {
this.getRemotingClient().invokeAsync(brokerAddr, request, timeoutMillis, responseFuture -> {
RemotingCommand response = responseFuture.getResponseCommand();
if (response != null) {
future.complete(response);
} else {
future.completeExceptionally(processNullResponseErr(responseFuture));
}
});
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture popMessageAsync(
String brokerAddr,
String brokerName,
PopMessageRequestHeader requestHeader,
long timeoutMillis
) {
CompletableFuture future = new CompletableFuture<>();
try {
this.popMessageAsync(brokerName, brokerAddr, requestHeader, timeoutMillis, new PopCallback() {
@Override
public void onSuccess(PopResult popResult) {
future.complete(popResult);
}
@Override
public void onException(Throwable t) {
future.completeExceptionally(t);
}
});
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture ackMessageAsync(
String brokerAddr,
AckMessageRequestHeader requestHeader,
long timeoutMillis
) {
CompletableFuture future = new CompletableFuture<>();
try {
this.ackMessageAsync(brokerAddr, timeoutMillis, new AckCallback() {
@Override
public void onSuccess(AckResult ackResult) {
future.complete(ackResult);
}
@Override
public void onException(Throwable t) {
future.completeExceptionally(t);
}
}, requestHeader);
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture changeInvisibleTimeAsync(
String brokerAddr,
String brokerName,
ChangeInvisibleTimeRequestHeader requestHeader,
long timeoutMillis
) {
CompletableFuture future = new CompletableFuture<>();
try {
this.changeInvisibleTimeAsync(brokerName, brokerAddr, requestHeader, timeoutMillis,
new AckCallback() {
@Override
public void onSuccess(AckResult ackResult) {
future.complete(ackResult);
}
@Override
public void onException(Throwable t) {
future.completeExceptionally(t);
}
}
);
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture pullMessageAsync(
String brokerAddr,
PullMessageRequestHeader requestHeader,
long timeoutMillis
) {
CompletableFuture future = new CompletableFuture<>();
try {
this.pullMessage(brokerAddr, requestHeader, timeoutMillis, CommunicationMode.ASYNC,
new PullCallback() {
@Override
public void onSuccess(PullResult pullResult) {
if (pullResult instanceof PullResultExt) {
PullResultExt pullResultExt = (PullResultExt) pullResult;
if (PullStatus.FOUND.equals(pullResult.getPullStatus())) {
List messageExtList = MessageDecoder.decodesBatch(
ByteBuffer.wrap(pullResultExt.getMessageBinary()),
true,
false,
true
);
pullResult.setMsgFoundList(messageExtList);
}
}
future.complete(pullResult);
}
@Override
public void onException(Throwable t) {
future.completeExceptionally(t);
}
}
);
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture queryConsumerOffsetWithFuture(
String brokerAddr,
QueryConsumerOffsetRequestHeader requestHeader,
long timeoutMillis
) {
CompletableFuture future = new CompletableFuture<>();
try {
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.QUERY_CONSUMER_OFFSET, requestHeader);
this.getRemotingClient().invokeAsync(brokerAddr, request, timeoutMillis, responseFuture -> {
RemotingCommand response = responseFuture.getResponseCommand();
if (response != null) {
switch (response.getCode()) {
case ResponseCode.SUCCESS: {
try {
QueryConsumerOffsetResponseHeader responseHeader =
(QueryConsumerOffsetResponseHeader) response.decodeCommandCustomHeader(QueryConsumerOffsetResponseHeader.class);
future.complete(responseHeader.getOffset());
} catch (RemotingCommandException e) {
future.completeExceptionally(e);
}
break;
}
case ResponseCode.QUERY_NOT_FOUND: {
future.completeExceptionally(new OffsetNotFoundException(response.getCode(), response.getRemark(), brokerAddr));
break;
}
default:
break;
}
} else {
future.completeExceptionally(processNullResponseErr(responseFuture));
}
});
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture updateConsumerOffsetOneWay(
String brokerAddr,
UpdateConsumerOffsetRequestHeader header,
long timeoutMillis
) {
CompletableFuture future = new CompletableFuture<>();
try {
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.UPDATE_CONSUMER_OFFSET, header);
this.getRemotingClient().invokeOneway(brokerAddr, request, timeoutMillis);
future.complete(null);
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture> getConsumerListByGroupAsync(
String brokerAddr,
GetConsumerListByGroupRequestHeader requestHeader,
long timeoutMillis
) {
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.GET_CONSUMER_LIST_BY_GROUP, requestHeader);
CompletableFuture> future = new CompletableFuture<>();
try {
this.getRemotingClient().invokeAsync(brokerAddr, request, timeoutMillis, responseFuture -> {
RemotingCommand response = responseFuture.getResponseCommand();
if (response != null) {
switch (response.getCode()) {
case ResponseCode.SUCCESS: {
if (response.getBody() != null) {
GetConsumerListByGroupResponseBody body =
GetConsumerListByGroupResponseBody.decode(response.getBody(), GetConsumerListByGroupResponseBody.class);
future.complete(body.getConsumerIdList());
return;
}
}
/*
@see org.apache.rocketmq.broker.processor.ConsumerManageProcessor#getConsumerListByGroup,
* broker will return {@link ResponseCode.SYSTEM_ERROR} if there is no consumer.
*/
case ResponseCode.SYSTEM_ERROR: {
future.complete(Collections.emptyList());
return;
}
default:
break;
}
future.completeExceptionally(new MQBrokerException(response.getCode(), response.getRemark()));
} else {
future.completeExceptionally(processNullResponseErr(responseFuture));
}
});
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture getMaxOffset(String brokerAddr, GetMaxOffsetRequestHeader requestHeader,
long timeoutMillis) {
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.GET_MAX_OFFSET, requestHeader);
CompletableFuture future = new CompletableFuture<>();
try {
this.getRemotingClient().invokeAsync(brokerAddr, request, timeoutMillis, responseFuture -> {
RemotingCommand response = responseFuture.getResponseCommand();
if (response != null) {
if (ResponseCode.SUCCESS == response.getCode()) {
try {
GetMaxOffsetResponseHeader responseHeader = (GetMaxOffsetResponseHeader) response.decodeCommandCustomHeader(GetMaxOffsetResponseHeader.class);
future.complete(responseHeader.getOffset());
} catch (Throwable t) {
future.completeExceptionally(t);
}
}
future.completeExceptionally(new MQBrokerException(response.getCode(), response.getRemark()));
} else {
future.completeExceptionally(processNullResponseErr(responseFuture));
}
});
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture getMinOffset(String brokerAddr, GetMinOffsetRequestHeader requestHeader,
long timeoutMillis) {
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.GET_MIN_OFFSET, requestHeader);
CompletableFuture future = new CompletableFuture<>();
try {
this.getRemotingClient().invokeAsync(brokerAddr, request, timeoutMillis, responseFuture -> {
RemotingCommand response = responseFuture.getResponseCommand();
if (response != null) {
if (ResponseCode.SUCCESS == response.getCode()) {
try {
GetMinOffsetResponseHeader responseHeader = (GetMinOffsetResponseHeader) response.decodeCommandCustomHeader(GetMinOffsetResponseHeader.class);
future.complete(responseHeader.getOffset());
} catch (Throwable t) {
future.completeExceptionally(t);
}
}
future.completeExceptionally(new MQBrokerException(response.getCode(), response.getRemark()));
} else {
future.completeExceptionally(processNullResponseErr(responseFuture));
}
});
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture searchOffset(String brokerAddr, SearchOffsetRequestHeader requestHeader,
long timeoutMillis) {
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.SEARCH_OFFSET_BY_TIMESTAMP, requestHeader);
CompletableFuture future = new CompletableFuture<>();
try {
this.getRemotingClient().invokeAsync(brokerAddr, request, timeoutMillis, responseFuture -> {
RemotingCommand response = responseFuture.getResponseCommand();
if (response != null) {
if (response.getCode() == ResponseCode.SUCCESS) {
try {
SearchOffsetResponseHeader responseHeader = (SearchOffsetResponseHeader) response.decodeCommandCustomHeader(SearchOffsetResponseHeader.class);
future.complete(responseHeader.getOffset());
} catch (Throwable t) {
future.completeExceptionally(t);
}
}
future.completeExceptionally(new MQBrokerException(response.getCode(), response.getRemark()));
} else {
future.completeExceptionally(processNullResponseErr(responseFuture));
}
});
} catch (Throwable t) {
future.completeExceptionally(t);
}
return future;
}
public CompletableFuture> lockBatchMQWithFuture(String brokerAddr,
LockBatchRequestBody requestBody, long timeoutMillis) {
CompletableFuture> future = new CompletableFuture<>();
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.LOCK_BATCH_MQ, null);
request.setBody(requestBody.encode());
try {
this.getRemotingClient().invokeAsync(brokerAddr, request, timeoutMillis, responseFuture -> {
RemotingCommand response = responseFuture.getResponseCommand();
if (response != null) {
if (response.getCode() == ResponseCode.SUCCESS) {
try {
LockBatchResponseBody responseBody = LockBatchResponseBody.decode(response.getBody(), LockBatchResponseBody.class);
Set messageQueues = responseBody.getLockOKMQSet();
future.complete(messageQueues);
} catch (Throwable t) {
future.completeExceptionally(t);
}
}
future.completeExceptionally(new MQBrokerException(response.getCode(), response.getRemark()));
} else {
future.completeExceptionally(processNullResponseErr(responseFuture));
}
});
} catch (Exception e) {
future.completeExceptionally(e);
}
return future;
}
public CompletableFuture unlockBatchMQOneway(String brokerAddr,
UnlockBatchRequestBody requestBody, long timeoutMillis) {
CompletableFuture future = new CompletableFuture<>();
RemotingCommand request = RemotingCommand.createRequestCommand(RequestCode.UNLOCK_BATCH_MQ, null);
request.setBody(requestBody.encode());
try {
this.getRemotingClient().invokeOneway(brokerAddr, request, timeoutMillis);
future.complete(null);
} catch (Exception e) {
future.completeExceptionally(e);
}
return future;
}
public CompletableFuture invoke(String brokerAddr, RemotingCommand request, long timeoutMillis) {
CompletableFuture future = new CompletableFuture<>();
try {
this.getRemotingClient().invokeAsync(brokerAddr, request, timeoutMillis, responseFuture -> {
RemotingCommand response = responseFuture.getResponseCommand();
if (response != null) {
future.complete(response);
} else {
future.completeExceptionally(processNullResponseErr(responseFuture));
}
});
} catch (Exception e) {
future.completeExceptionally(e);
}
return future;
}
public CompletableFuture invokeOneway(String brokerAddr, RemotingCommand request, long timeoutMillis) {
CompletableFuture future = new CompletableFuture<>();
try {
this.getRemotingClient().invokeOneway(brokerAddr, request, timeoutMillis);
future.complete(null);
} catch (Exception e) {
future.completeExceptionally(e);
}
return future;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy