org.apache.rocketmq.client.java.impl.ClientManagerImpl 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.client.java.impl;
import apache.rocketmq.v2.AckMessageRequest;
import apache.rocketmq.v2.AckMessageResponse;
import apache.rocketmq.v2.ChangeInvisibleDurationRequest;
import apache.rocketmq.v2.ChangeInvisibleDurationResponse;
import apache.rocketmq.v2.EndTransactionRequest;
import apache.rocketmq.v2.EndTransactionResponse;
import apache.rocketmq.v2.ForwardMessageToDeadLetterQueueRequest;
import apache.rocketmq.v2.ForwardMessageToDeadLetterQueueResponse;
import apache.rocketmq.v2.HeartbeatRequest;
import apache.rocketmq.v2.HeartbeatResponse;
import apache.rocketmq.v2.NotifyClientTerminationRequest;
import apache.rocketmq.v2.NotifyClientTerminationResponse;
import apache.rocketmq.v2.QueryAssignmentRequest;
import apache.rocketmq.v2.QueryAssignmentResponse;
import apache.rocketmq.v2.QueryRouteRequest;
import apache.rocketmq.v2.QueryRouteResponse;
import apache.rocketmq.v2.ReceiveMessageRequest;
import apache.rocketmq.v2.ReceiveMessageResponse;
import apache.rocketmq.v2.SendMessageRequest;
import apache.rocketmq.v2.SendMessageResponse;
import apache.rocketmq.v2.TelemetryCommand;
import org.apache.rocketmq.shaded.com.google.common.util.concurrent.ListenableFuture;
import org.apache.rocketmq.shaded.com.google.errorprone.annotations.concurrent.GuardedBy;
import org.apache.rocketmq.shaded.io.grpc.Metadata;
import org.apache.rocketmq.shaded.io.grpc.stub.StreamObserver;
import java.io.IOException;
import java.time.Duration;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.ScheduledThreadPoolExecutor;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;
import javax.net.ssl.SSLException;
import org.apache.rocketmq.client.apis.ClientException;
import org.apache.rocketmq.client.java.exception.InternalErrorException;
import org.apache.rocketmq.client.java.misc.ClientId;
import org.apache.rocketmq.client.java.misc.ExecutorServices;
import org.apache.rocketmq.client.java.misc.MetadataUtils;
import org.apache.rocketmq.client.java.misc.ThreadFactoryImpl;
import org.apache.rocketmq.client.java.route.Endpoints;
import org.apache.rocketmq.client.java.rpc.Context;
import org.apache.rocketmq.client.java.rpc.RpcClient;
import org.apache.rocketmq.client.java.rpc.RpcClientImpl;
import org.apache.rocketmq.client.java.rpc.RpcFuture;
import org.apache.rocketmq.shaded.org.slf4j.Logger;
import org.apache.rocketmq.shaded.org.slf4j.LoggerFactory;
/**
* @see ClientManager
*/
public class ClientManagerImpl extends ClientManager {
public static final Duration RPC_CLIENT_MAX_IDLE_DURATION = Duration.ofMinutes(30);
public static final Duration RPC_CLIENT_IDLE_CHECK_INITIAL_DELAY = Duration.ofSeconds(5);
public static final Duration RPC_CLIENT_IDLE_CHECK_PERIOD = Duration.ofMinutes(1);
public static final Duration HEART_BEAT_INITIAL_DELAY = Duration.ofSeconds(1);
public static final Duration HEART_BEAT_PERIOD = Duration.ofSeconds(10);
public static final Duration LOG_STATS_INITIAL_DELAY = Duration.ofSeconds(1);
public static final Duration LOG_STATS_PERIOD = Duration.ofSeconds(60);
public static final Duration SYNC_SETTINGS_DELAY = Duration.ofSeconds(1);
public static final Duration SYNC_SETTINGS_PERIOD = Duration.ofMinutes(5);
private static final Logger log = LoggerFactory.getLogger(ClientManagerImpl.class);
private final Client client;
@GuardedBy("rpcClientTableLock")
private final Map rpcClientTable;
private final ReadWriteLock rpcClientTableLock;
/**
* In charge of all scheduled tasks.
*/
private final ScheduledExecutorService scheduler;
/**
* Public executor for all async RPCs, should never submit a heavy task.
*/
private final ExecutorService asyncWorker;
public ClientManagerImpl(Client client) {
this.client = client;
this.rpcClientTable = new HashMap<>();
this.rpcClientTableLock = new ReentrantReadWriteLock();
final long clientIndex = client.getClientId().getIndex();
this.scheduler = new ScheduledThreadPoolExecutor(
Runtime.getRuntime().availableProcessors(),
new ThreadFactoryImpl("ClientScheduler", clientIndex));
this.asyncWorker = new ThreadPoolExecutor(
Runtime.getRuntime().availableProcessors(),
Runtime.getRuntime().availableProcessors(),
60,
TimeUnit.SECONDS,
new LinkedBlockingQueue<>(),
new ThreadFactoryImpl("ClientAsyncWorker", clientIndex));
}
/**
* It is well-founded that a {@link RpcClient} is deprecated if it is idle for a long time, so it is essential to
* clear it.
*
* @throws InterruptedException if the thread has been interrupted
*/
private void clearIdleRpcClients() throws InterruptedException {
rpcClientTableLock.writeLock().lock();
try {
final Iterator> it = rpcClientTable.entrySet().iterator();
while (it.hasNext()) {
final Map.Entry entry = it.next();
final Endpoints endpoints = entry.getKey();
final RpcClient rpcClient = entry.getValue();
final Duration idleDuration = rpcClient.idleDuration();
if (idleDuration.compareTo(RPC_CLIENT_MAX_IDLE_DURATION) > 0) {
it.remove();
rpcClient.shutdown();
log.info("Rpc client has been idle for a long time, endpoints={}, idleDuration={}, " +
"rpcClientMaxIdleDuration={}, clientId={}", endpoints, idleDuration,
RPC_CLIENT_MAX_IDLE_DURATION, client.getClientId());
}
}
} finally {
rpcClientTableLock.writeLock().unlock();
}
}
/**
* Return the RPC client by remote {@link Endpoints}, would create the client automatically if it does not exist.
*
* In case of the occasion that {@link RpcClient} is garbage collected before shutdown when invoked
* concurrently, lock here is essential.
*
* @param endpoints remote endpoints.
* @return RPC client.
*/
private RpcClient getRpcClient(Endpoints endpoints) throws ClientException {
RpcClient rpcClient;
rpcClientTableLock.readLock().lock();
try {
rpcClient = rpcClientTable.get(endpoints);
if (null != rpcClient) {
return rpcClient;
}
} finally {
rpcClientTableLock.readLock().unlock();
}
rpcClientTableLock.writeLock().lock();
try {
rpcClient = rpcClientTable.get(endpoints);
if (null != rpcClient) {
return rpcClient;
}
try {
rpcClient = new RpcClientImpl(endpoints);
} catch (SSLException e) {
log.error("Failed to get RPC client, endpoints={}, clientId={}", endpoints, client.getClientId(), e);
throw new ClientException("Failed to generate RPC client", e);
}
rpcClientTable.put(endpoints, rpcClient);
return rpcClient;
} finally {
rpcClientTableLock.writeLock().unlock();
}
}
@Override
public RpcFuture queryRoute(Endpoints endpoints, QueryRouteRequest request,
Duration duration) {
try {
final Metadata metadata = client.sign();
final Context context = new Context(endpoints, metadata);
final RpcClient rpcClient = getRpcClient(endpoints);
final ListenableFuture future = rpcClient.queryRoute(metadata, request, asyncWorker,
duration);
return new RpcFuture<>(context, request, future);
} catch (Throwable t) {
return new RpcFuture<>(t);
}
}
@Override
public RpcFuture heartbeat(Endpoints endpoints, HeartbeatRequest request,
Duration duration) {
try {
final Metadata metadata = client.sign();
final Context context = new Context(endpoints, metadata);
final RpcClient rpcClient = getRpcClient(endpoints);
ListenableFuture future = rpcClient.heartbeat(metadata, request, asyncWorker, duration);
return new RpcFuture<>(context, request, future);
} catch (Throwable t) {
return new RpcFuture<>(t);
}
}
@Override
public RpcFuture sendMessage(Endpoints endpoints,
SendMessageRequest request, Duration duration) {
try {
final Metadata metadata = client.sign();
final Context context = new Context(endpoints, metadata);
final RpcClient rpcClient = getRpcClient(endpoints);
final ListenableFuture future =
rpcClient.sendMessage(metadata, request, asyncWorker, duration);
return new RpcFuture<>(context, request, future);
} catch (Throwable t) {
return new RpcFuture<>(t);
}
}
@Override
public RpcFuture queryAssignment(Endpoints endpoints,
QueryAssignmentRequest request, Duration duration) {
try {
final Metadata metadata = client.sign();
final Context context = new Context(endpoints, metadata);
final RpcClient rpcClient = getRpcClient(endpoints);
final ListenableFuture future =
rpcClient.queryAssignment(metadata, request, asyncWorker, duration);
return new RpcFuture<>(context, request, future);
} catch (Throwable t) {
return new RpcFuture<>(t);
}
}
@Override
public RpcFuture> receiveMessage(Endpoints endpoints,
ReceiveMessageRequest request, Duration duration) {
try {
final Metadata metadata = client.sign();
final Context context = new Context(endpoints, metadata);
final RpcClient rpcClient = getRpcClient(endpoints);
final ListenableFuture> future =
rpcClient.receiveMessage(metadata, request, asyncWorker, duration);
return new RpcFuture<>(context, request, future);
} catch (Throwable t) {
return new RpcFuture<>(t);
}
}
@Override
public RpcFuture ackMessage(Endpoints endpoints, AckMessageRequest request,
Duration duration) {
try {
final Metadata metadata = client.sign();
final Context context = new Context(endpoints, metadata);
final RpcClient rpcClient = getRpcClient(endpoints);
final ListenableFuture future =
rpcClient.ackMessage(metadata, request, asyncWorker, duration);
return new RpcFuture<>(context, request, future);
} catch (Throwable t) {
return new RpcFuture<>(t);
}
}
@Override
public RpcFuture
changeInvisibleDuration(Endpoints endpoints, ChangeInvisibleDurationRequest request,
Duration duration) {
try {
final Metadata metadata = client.sign();
final Context context = new Context(endpoints, metadata);
final RpcClient rpcClient = getRpcClient(endpoints);
final ListenableFuture future =
rpcClient.changeInvisibleDuration(metadata, request, asyncWorker, duration);
return new RpcFuture<>(context, request, future);
} catch (Throwable t) {
return new RpcFuture<>(t);
}
}
@Override
public RpcFuture
forwardMessageToDeadLetterQueue(Endpoints endpoints, ForwardMessageToDeadLetterQueueRequest request,
Duration duration) {
try {
final Metadata metadata = client.sign();
final Context context = new Context(endpoints, metadata);
final RpcClient rpcClient = getRpcClient(endpoints);
final ListenableFuture future =
rpcClient.forwardMessageToDeadLetterQueue(metadata, request, asyncWorker, duration);
return new RpcFuture<>(context, request, future);
} catch (Throwable t) {
return new RpcFuture<>(t);
}
}
@Override
public RpcFuture endTransaction(Endpoints endpoints,
EndTransactionRequest request, Duration duration) {
try {
final Metadata metadata = client.sign();
final Context context = new Context(endpoints, metadata);
final RpcClient rpcClient = getRpcClient(endpoints);
final ListenableFuture future =
rpcClient.endTransaction(metadata, request, asyncWorker, duration);
return new RpcFuture<>(context, request, future);
} catch (Throwable t) {
return new RpcFuture<>(t);
}
}
@Override
public RpcFuture
notifyClientTermination(Endpoints endpoints, NotifyClientTerminationRequest request,
Duration duration) {
try {
final Metadata metadata = client.sign();
final Context context = new Context(endpoints, metadata);
final RpcClient rpcClient = getRpcClient(endpoints);
final ListenableFuture future =
rpcClient.notifyClientTermination(metadata, request, asyncWorker, duration);
return new RpcFuture<>(context, request, future);
} catch (Throwable t) {
return new RpcFuture<>(t);
}
}
@Override
public StreamObserver telemetry(Endpoints endpoints, Duration duration,
StreamObserver responseObserver) throws ClientException {
try {
final Metadata metadata = client.sign();
final RpcClient rpcClient = getRpcClient(endpoints);
return rpcClient.telemetry(metadata, asyncWorker, duration, responseObserver);
} catch (Throwable t) {
throw new InternalErrorException(t);
}
}
@Override
public ScheduledExecutorService getScheduler() {
return this.scheduler;
}
@Override
protected void startUp() {
final ClientId clientId = client.getClientId();
log.info("Begin to start the client manager, clientId={}", clientId);
scheduler.scheduleWithFixedDelay(
() -> {
try {
clearIdleRpcClients();
} catch (Throwable t) {
log.error("Exception raised during the clearing of idle rpc clients, clientId={}", clientId, t);
}
},
RPC_CLIENT_IDLE_CHECK_INITIAL_DELAY.toNanos(),
RPC_CLIENT_IDLE_CHECK_PERIOD.toNanos(),
TimeUnit.NANOSECONDS
);
scheduler.scheduleWithFixedDelay(
() -> {
try {
client.doHeartbeat();
} catch (Throwable t) {
log.error("Exception raised during heartbeat, clientId={}", clientId, t);
}
},
HEART_BEAT_INITIAL_DELAY.toNanos(),
HEART_BEAT_PERIOD.toNanos(),
TimeUnit.NANOSECONDS
);
scheduler.scheduleWithFixedDelay(
() -> {
try {
log.info("Start to log statistics, clientVersion={}, clientWrapperVersion={}, "
+ "clientEndpoints={}, clientId={}", MetadataUtils.getVersion(),
MetadataUtils.getWrapperVersion(), client.getEndpoints(), clientId);
client.doStats();
} catch (Throwable t) {
log.error("Exception raised during statistics logging, clientId={}", clientId, t);
}
},
LOG_STATS_INITIAL_DELAY.toNanos(),
LOG_STATS_PERIOD.toNanos(),
TimeUnit.NANOSECONDS
);
scheduler.scheduleWithFixedDelay(
() -> {
try {
client.syncSettings();
} catch (Throwable t) {
log.error("Exception raised during the setting synchronization, clientId={}", clientId, t);
}
},
SYNC_SETTINGS_DELAY.toNanos(),
SYNC_SETTINGS_PERIOD.toNanos(),
TimeUnit.NANOSECONDS
);
log.info("The client manager starts successfully, clientId={}", clientId);
}
@Override
protected void shutDown() throws IOException {
final ClientId clientId = client.getClientId();
log.info("Begin to shutdown the client manager, clientId={}", clientId);
scheduler.shutdown();
try {
if (!ExecutorServices.awaitTerminated(scheduler)) {
log.error("[Bug] Timeout to shutdown the client scheduler, clientId={}", clientId);
} else {
log.info("Shutdown the client scheduler successfully, clientId={}", clientId);
}
rpcClientTableLock.writeLock().lock();
try {
final Iterator> it = rpcClientTable.entrySet().iterator();
while (it.hasNext()) {
final Map.Entry entry = it.next();
final RpcClient rpcClient = entry.getValue();
it.remove();
rpcClient.shutdown();
}
} finally {
rpcClientTableLock.writeLock().unlock();
}
log.info("Shutdown all rpc client(s) successfully, clientId={}", clientId);
asyncWorker.shutdown();
if (!ExecutorServices.awaitTerminated(asyncWorker)) {
log.error("[Bug] Timeout to shutdown the client async worker, clientId={}", clientId);
} else {
log.info("Shutdown the client async worker successfully, clientId={}", clientId);
}
} catch (InterruptedException e) {
log.error("[Bug] Unexpected exception raised while shutdown client manager, clientId={}", clientId, e);
throw new IOException(e);
}
log.info("Shutdown the client manager successfully, clientId={}", clientId);
}
@SuppressWarnings("NullableProblems")
@Override
protected String serviceName() {
return super.serviceName() + "-" + client.getClientId().getIndex();
}
}