org.redisson.client.RedisConnection Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of redisson-all Show documentation
Show all versions of redisson-all Show documentation
Easy Redis Java client and Real-Time Data Platform. Valkey compatible. Sync/Async/RxJava3/Reactive API. Client side caching. Over 50 Redis based Java objects and services: JCache API, Apache Tomcat, Hibernate, Spring, Set, Multimap, SortedSet, Map, List, Queue, Deque, Semaphore, Lock, AtomicLong, Map Reduce, Bloom filter, Scheduler, RPC
/**
* Copyright (c) 2013-2024 Nikita Koksharov
*
* Licensed 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.redisson.client;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelFutureListener;
import io.netty.util.AttributeKey;
import io.netty.util.Timeout;
import org.redisson.RedissonShutdownException;
import org.redisson.api.RFuture;
import org.redisson.client.codec.Codec;
import org.redisson.client.handler.CommandsQueue;
import org.redisson.client.handler.CommandsQueuePubSub;
import org.redisson.client.protocol.*;
import org.redisson.misc.CompletableFutureWrapper;
import org.redisson.misc.LogHelper;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.util.Deque;
import java.util.Queue;
import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicInteger;
/**
*
* @author Nikita Koksharov
*
*/
public class RedisConnection implements RedisCommands {
public enum Status {OPEN, CLOSED, CLOSED_IDLE}
private static final Logger LOG = LoggerFactory.getLogger(RedisConnection.class);
private static final AttributeKey CONNECTION = AttributeKey.valueOf("connection");
final RedisClient redisClient;
private volatile CompletableFuture fastReconnect;
private volatile Status status = Status.OPEN;
volatile Channel channel;
private CompletableFuture> connectionPromise;
private volatile long lastUsageTime;
@Deprecated
private Runnable connectedListener;
@Deprecated
private Runnable disconnectedListener;
private final AtomicInteger usage = new AtomicInteger();
public RedisConnection(RedisClient redisClient, Channel channel, CompletableFuture connectionPromise) {
this.redisClient = redisClient;
this.connectionPromise = connectionPromise;
updateChannel(channel);
lastUsageTime = System.nanoTime();
LOG.debug("Connection created {}", redisClient);
}
protected RedisConnection(RedisClient redisClient) {
this.redisClient = redisClient;
}
public void fireConnected() {
if (connectedListener != null) {
connectedListener.run();
}
if (redisClient.getConfig().getConnectedListener() != null) {
redisClient.getConfig().getConnectedListener().accept(redisClient.getAddr());
}
}
public int incUsage() {
return usage.incrementAndGet();
}
public int getUsage() {
return usage.get();
}
public int decUsage() {
return usage.decrementAndGet();
}
@Deprecated
public void setConnectedListener(Runnable connectedListener) {
this.connectedListener = connectedListener;
}
public void fireDisconnected() {
if (disconnectedListener != null) {
disconnectedListener.run();
}
if (redisClient.getConfig().getDisconnectedListener() != null) {
redisClient.getConfig().getDisconnectedListener().accept(redisClient.getAddr());
}
}
@Deprecated
public void setDisconnectedListener(Runnable disconnectedListener) {
this.disconnectedListener = disconnectedListener;
}
public CompletableFuture getConnectionPromise() {
return (CompletableFuture) connectionPromise;
}
public static C getFrom(Channel channel) {
return (C) channel.attr(RedisConnection.CONNECTION).get();
}
public CommandData, ?> getLastCommand() {
Deque queue = channel.attr(CommandsQueue.COMMANDS_QUEUE).get();
if (queue != null) {
QueueCommandHolder holder = queue.peekLast();
if (holder != null) {
if (holder.getCommand() instanceof CommandData) {
return (CommandData, ?>) holder.getCommand();
}
}
}
return null;
}
public QueueCommand getCurrentCommandData() {
Queue queue = channel.attr(CommandsQueue.COMMANDS_QUEUE).get();
if (queue != null) {
QueueCommandHolder holder = queue.peek();
if (holder != null) {
return holder.getCommand();
}
}
QueueCommandHolder holder = channel.attr(CommandsQueuePubSub.CURRENT_COMMAND).get();
if (holder != null) {
return holder.getCommand();
}
return null;
}
public CommandData, ?> getCurrentCommand() {
Queue queue = channel.attr(CommandsQueue.COMMANDS_QUEUE).get();
if (queue != null) {
QueueCommandHolder holder = queue.peek();
if (holder != null) {
if (holder.getCommand() instanceof CommandData) {
return (CommandData, ?>) holder.getCommand();
}
}
}
QueueCommandHolder holder = channel.attr(CommandsQueuePubSub.CURRENT_COMMAND).get();
if (holder != null && holder.getCommand() instanceof CommandData) {
return (CommandData, ?>) holder.getCommand();
}
return null;
}
public long getLastUsageTime() {
return lastUsageTime;
}
public void setLastUsageTime(long lastUsageTime) {
this.lastUsageTime = lastUsageTime;
}
public boolean isOpen() {
return channel.isOpen();
}
/**
* Check is channel connected and ready for transfer
*
* @return true if so
*/
public boolean isActive() {
return channel.isActive();
}
public void updateChannel(Channel channel) {
if (channel == null) {
throw new NullPointerException();
}
this.channel = channel;
channel.attr(CONNECTION).set(this);
}
public RedisClient getRedisClient() {
return redisClient;
}
public R await(CompletableFuture future) {
try {
return future.get(redisClient.getCommandTimeout(), TimeUnit.MILLISECONDS);
} catch (ExecutionException e) {
if (e.getCause() instanceof RedisException) {
throw (RedisException) e.getCause();
}
throw new RedisException("Unexpected exception while processing command", e.getCause());
} catch (TimeoutException e) {
RedisTimeoutException ex = new RedisTimeoutException("Command execution timeout for " + redisClient.getAddr());
future.completeExceptionally(ex);
throw ex;
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
return null;
}
}
public T sync(RedisCommand command, Object... params) {
return sync(null, command, params);
}
public ChannelFuture send(CommandData data) {
return channel.writeAndFlush(data);
}
public ChannelFuture send(CommandsData data) {
return channel.writeAndFlush(data);
}
public R sync(Codec encoder, RedisCommand command, Object... params) {
CompletableFuture promise = new CompletableFuture<>();
send(new CommandData(promise, encoder, command, params));
return await(promise);
}
public RFuture async(RedisCommand command, Object... params) {
return async(-1, command, params);
}
public RFuture async(long timeout, RedisCommand command, Object... params) {
return async(timeout, null, command, params);
}
public RFuture async(Codec codec, RedisCommand command, Object... params) {
return async(-1, codec, command, params);
}
public RFuture async(int retryAttempts, int retryInterval, long timeout, Codec codec, RedisCommand command, Object... params) {
CompletableFuture result = new CompletableFuture<>();
AtomicInteger attempts = new AtomicInteger(retryAttempts);
async(result, attempts, retryInterval, timeout, codec, command, params);
return new CompletableFutureWrapper<>(result);
}
private void async(CompletableFuture promise, AtomicInteger attempts, int retryInterval, long timeout, Codec codec, RedisCommand command, Object... params) {
RFuture f = async(timeout, codec, command, params);
f.whenComplete((r, e) -> {
if (e != null) {
if (attempts.decrementAndGet() >= 0) {
redisClient.getTimer().newTimeout(t -> {
async(promise, attempts, retryInterval, timeout, codec, command, params);
}, retryInterval, TimeUnit.MILLISECONDS);
return;
}
promise.completeExceptionally(e);
return;
}
promise.complete(r);
});
}
public RFuture async(long timeout, Codec codec, RedisCommand command, Object... params) {
CompletableFuture promise = new CompletableFuture<>();
if (timeout == -1) {
timeout = redisClient.getCommandTimeout();
}
if (redisClient.isShutdown()) {
RedissonShutdownException cause = new RedissonShutdownException("Redis client " + redisClient.getAddr() + " is shutdown");
return new CompletableFutureWrapper<>(cause);
}
Timeout scheduledFuture = redisClient.getTimer().newTimeout(t -> {
RedisTimeoutException ex = new RedisTimeoutException("Command execution timeout for "
+ LogHelper.toString(command, params) + ", Redis client: " + redisClient);
promise.completeExceptionally(ex);
}, timeout, TimeUnit.MILLISECONDS);
promise.whenComplete((res, e) -> {
scheduledFuture.cancel();
});
ChannelFuture writeFuture = send(new CommandData<>(promise, codec, command, params));
writeFuture.addListener((ChannelFutureListener) future -> {
if (!future.isSuccess()) {
promise.completeExceptionally(future.cause());
}
});
return new CompletableFutureWrapper<>(promise);
}
public CommandData create(Codec encoder, RedisCommand command, Object... params) {
CompletableFuture promise = new CompletableFuture<>();
return new CommandData<>(promise, encoder, command, params);
}
public boolean isClosed() {
return status != Status.OPEN;
}
public boolean isFastReconnect() {
return fastReconnect != null;
}
public void clearFastReconnect() {
fastReconnect.complete(null);
fastReconnect = null;
}
public void close() {
try {
closeAsync().sync();
} catch (InterruptedException e) {
Thread.currentThread().interrupt();
} catch (Exception e) {
throw e;
}
}
private void closeInternal() {
QueueCommand command = getCurrentCommandData();
if ((command != null && command.isBlockingCommand())
|| !connectionPromise.isDone()) {
channel.close();
} else {
RFuture f = async(RedisCommands.QUIT);
f.whenComplete((res, e) -> {
if (redisClient.isShutdown()) {
return;
}
channel.close();
});
}
}
public CompletionStage forceFastReconnectAsync() {
CompletableFuture promise = new CompletableFuture<>();
fastReconnect = promise;
closeInternal();
return promise;
}
/**
* Access to Netty channel.
* This method is provided to use in debug info only.
*
* @return channel
*/
public Channel getChannel() {
return channel;
}
public ChannelFuture closeIdleAsync() {
status = Status.CLOSED_IDLE;
closeInternal();
return channel.closeFuture();
}
public boolean isClosedIdle() {
return status == Status.CLOSED_IDLE;
}
public ChannelFuture closeAsync() {
if (status == Status.CLOSED) {
return channel.closeFuture();
}
status = Status.CLOSED;
closeInternal();
return channel.closeFuture();
}
@Override
public String toString() {
return getClass().getSimpleName() + "@" + System.identityHashCode(this)
+ " [redisClient=" + redisClient
+ ", channel=" + channel
+ ", currentCommand=" + getCurrentCommandData()
+ ", usage=" + usage + "]";
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy