refactoring
Nikita Koksharov
6 years ago
parent
ecd3a2cb35
commit
f58c083653
@ -0,0 +1,93 @@
|
||||
/**
|
||||
* Copyright (c) 2013-2019 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.command;
|
||||
|
||||
import java.util.concurrent.ConcurrentMap;
|
||||
import java.util.concurrent.atomic.AtomicBoolean;
|
||||
import java.util.concurrent.atomic.AtomicInteger;
|
||||
|
||||
import org.redisson.api.BatchOptions;
|
||||
import org.redisson.api.RFuture;
|
||||
import org.redisson.client.RedisConnection;
|
||||
import org.redisson.client.codec.Codec;
|
||||
import org.redisson.client.protocol.BatchCommandData;
|
||||
import org.redisson.client.protocol.RedisCommand;
|
||||
import org.redisson.client.protocol.RedisCommands;
|
||||
import org.redisson.command.CommandBatchService.ConnectionEntry;
|
||||
import org.redisson.command.CommandBatchService.Entry;
|
||||
import org.redisson.connection.ConnectionManager;
|
||||
import org.redisson.connection.MasterSlaveEntry;
|
||||
import org.redisson.connection.NodeSource;
|
||||
import org.redisson.liveobject.core.RedissonObjectBuilder;
|
||||
import org.redisson.misc.RPromise;
|
||||
import org.redisson.pubsub.AsyncSemaphore;
|
||||
|
||||
/**
|
||||
*
|
||||
* @author Nikita Koksharov
|
||||
*
|
||||
* @param <V> type of value
|
||||
* @param <R> type of returned value
|
||||
*/
|
||||
public class BaseRedisBatchExecutor<V, R> extends RedisExecutor<V, R> {
|
||||
|
||||
final ConcurrentMap<MasterSlaveEntry, Entry> commands;
|
||||
final ConcurrentMap<MasterSlaveEntry, ConnectionEntry> connections;
|
||||
final BatchOptions options;
|
||||
final AtomicInteger index;
|
||||
|
||||
final AtomicBoolean executed;
|
||||
final AsyncSemaphore semaphore;
|
||||
|
||||
@SuppressWarnings("ParameterNumber")
|
||||
public BaseRedisBatchExecutor(boolean readOnlyMode, NodeSource source, Codec codec, RedisCommand<V> command,
|
||||
Object[] params, RPromise<R> mainPromise, boolean ignoreRedirect,
|
||||
ConnectionManager connectionManager, RedissonObjectBuilder objectBuilder,
|
||||
ConcurrentMap<MasterSlaveEntry, Entry> commands, ConcurrentMap<MasterSlaveEntry, ConnectionEntry> connections,
|
||||
BatchOptions options, AtomicInteger index, AtomicBoolean executed, AsyncSemaphore semaphore) {
|
||||
|
||||
super(readOnlyMode, source, codec, command, params, mainPromise, ignoreRedirect, connectionManager,
|
||||
objectBuilder);
|
||||
this.commands = commands;
|
||||
this.connections = connections;
|
||||
this.options = options;
|
||||
this.index = index;
|
||||
this.executed = executed;
|
||||
this.semaphore = semaphore;
|
||||
}
|
||||
|
||||
protected final void addBatchCommandData(Object[] batchParams) {
|
||||
if (source.getEntry() != null) {
|
||||
Entry entry = commands.get(source.getEntry());
|
||||
if (entry == null) {
|
||||
entry = new Entry();
|
||||
Entry oldEntry = commands.putIfAbsent(source.getEntry(), entry);
|
||||
if (oldEntry != null) {
|
||||
entry = oldEntry;
|
||||
}
|
||||
}
|
||||
|
||||
if (!readOnlyMode) {
|
||||
entry.setReadOnlyMode(false);
|
||||
}
|
||||
|
||||
Codec codecToUse = getCodec(codec);
|
||||
BatchCommandData<V, R> commandData = new BatchCommandData<V, R>(mainPromise, codecToUse, command, batchParams, index.incrementAndGet());
|
||||
entry.getCommands().add(commandData);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,222 @@
|
||||
/**
|
||||
* Copyright (c) 2013-2019 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.command;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.concurrent.ConcurrentMap;
|
||||
import java.util.concurrent.atomic.AtomicBoolean;
|
||||
import java.util.concurrent.atomic.AtomicInteger;
|
||||
|
||||
import org.redisson.api.BatchOptions;
|
||||
import org.redisson.api.BatchOptions.ExecutionMode;
|
||||
import org.redisson.api.RFuture;
|
||||
import org.redisson.client.RedisConnection;
|
||||
import org.redisson.client.codec.Codec;
|
||||
import org.redisson.client.protocol.BatchCommandData;
|
||||
import org.redisson.client.protocol.CommandData;
|
||||
import org.redisson.client.protocol.CommandsData;
|
||||
import org.redisson.client.protocol.RedisCommand;
|
||||
import org.redisson.client.protocol.RedisCommands;
|
||||
import org.redisson.command.CommandBatchService.ConnectionEntry;
|
||||
import org.redisson.command.CommandBatchService.Entry;
|
||||
import org.redisson.connection.ConnectionManager;
|
||||
import org.redisson.connection.MasterSlaveEntry;
|
||||
import org.redisson.connection.NodeSource;
|
||||
import org.redisson.connection.NodeSource.Redirect;
|
||||
import org.redisson.liveobject.core.RedissonObjectBuilder;
|
||||
import org.redisson.misc.LogHelper;
|
||||
import org.redisson.misc.RPromise;
|
||||
import org.redisson.misc.RedissonPromise;
|
||||
import org.redisson.pubsub.AsyncSemaphore;
|
||||
|
||||
/**
|
||||
*
|
||||
* @author Nikita Koksharov
|
||||
*
|
||||
* @param <V> type of value
|
||||
* @param <R> type of returned value
|
||||
*/
|
||||
public class RedisQueuedBatchExecutor<V, R> extends BaseRedisBatchExecutor<V, R> {
|
||||
|
||||
@SuppressWarnings("ParameterNumber")
|
||||
public RedisQueuedBatchExecutor(boolean readOnlyMode, NodeSource source, Codec codec, RedisCommand<V> command,
|
||||
Object[] params, RPromise<R> mainPromise, boolean ignoreRedirect, ConnectionManager connectionManager,
|
||||
RedissonObjectBuilder objectBuilder, ConcurrentMap<MasterSlaveEntry, Entry> commands,
|
||||
ConcurrentMap<MasterSlaveEntry, ConnectionEntry> connections, BatchOptions options, AtomicInteger index,
|
||||
AtomicBoolean executed, AsyncSemaphore semaphore) {
|
||||
super(readOnlyMode, source, codec, command, params, mainPromise, ignoreRedirect, connectionManager, objectBuilder,
|
||||
commands, connections, options, index, executed, semaphore);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void execute() {
|
||||
addBatchCommandData(null);
|
||||
|
||||
if (!readOnlyMode && this.options.getExecutionMode() == ExecutionMode.REDIS_READ_ATOMIC) {
|
||||
throw new IllegalStateException("Data modification commands can't be used with queueStore=REDIS_READ_ATOMIC");
|
||||
}
|
||||
|
||||
super.execute();
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
protected void releaseConnection(RPromise<R> attemptPromise, RFuture<RedisConnection> connectionFuture) {
|
||||
if (RedisCommands.EXEC.getName().equals(command.getName())) {
|
||||
super.releaseConnection(attemptPromise, connectionFuture);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void handleSuccess(RPromise<R> promise, RFuture<RedisConnection> connectionFuture, R res)
|
||||
throws ReflectiveOperationException {
|
||||
if (RedisCommands.EXEC.getName().equals(command.getName())) {
|
||||
super.handleSuccess(promise, connectionFuture, res);
|
||||
return;
|
||||
}
|
||||
if (RedisCommands.DISCARD.getName().equals(command.getName())) {
|
||||
super.handleSuccess(promise, connectionFuture, null);
|
||||
if (executed.compareAndSet(false, true)) {
|
||||
connectionFuture.getNow().forceFastReconnectAsync().onComplete((r, e) -> {
|
||||
releaseConnection(promise, connectionFuture);
|
||||
});
|
||||
}
|
||||
return;
|
||||
}
|
||||
|
||||
BatchPromise<R> batchPromise = (BatchPromise<R>) promise;
|
||||
RPromise<R> sentPromise = (RPromise<R>) batchPromise.getSentPromise();
|
||||
super.handleSuccess(sentPromise, connectionFuture, null);
|
||||
semaphore.release();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void handleError(RFuture<RedisConnection> connectionFuture, Throwable cause) {
|
||||
if (mainPromise instanceof BatchPromise) {
|
||||
BatchPromise<R> batchPromise = (BatchPromise<R>) mainPromise;
|
||||
RPromise<R> sentPromise = (RPromise<R>) batchPromise.getSentPromise();
|
||||
sentPromise.tryFailure(cause);
|
||||
mainPromise.tryFailure(cause);
|
||||
if (executed.compareAndSet(false, true)) {
|
||||
connectionFuture.getNow().forceFastReconnectAsync().onComplete((res, e) -> {
|
||||
RedisQueuedBatchExecutor.super.releaseConnection(mainPromise, connectionFuture);
|
||||
});
|
||||
}
|
||||
semaphore.release();
|
||||
return;
|
||||
}
|
||||
|
||||
super.handleError(connectionFuture, cause);
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void sendCommand(RPromise<R> attemptPromise, RedisConnection connection) {
|
||||
ConnectionEntry connectionEntry = connections.get(source.getEntry());
|
||||
|
||||
if (source.getRedirect() == Redirect.ASK) {
|
||||
List<CommandData<?, ?>> list = new ArrayList<CommandData<?, ?>>(2);
|
||||
RPromise<Void> promise = new RedissonPromise<Void>();
|
||||
list.add(new CommandData<Void, Void>(promise, codec, RedisCommands.ASKING, new Object[]{}));
|
||||
if (connectionEntry.isFirstCommand()) {
|
||||
list.add(new CommandData<Void, Void>(promise, codec, RedisCommands.MULTI, new Object[]{}));
|
||||
connectionEntry.setFirstCommand(false);
|
||||
}
|
||||
list.add(new CommandData<V, R>(attemptPromise, codec, command, params));
|
||||
RPromise<Void> main = new RedissonPromise<Void>();
|
||||
writeFuture = connection.send(new CommandsData(main, list, true));
|
||||
} else {
|
||||
if (log.isDebugEnabled()) {
|
||||
log.debug("acquired connection for command {} and params {} from slot {} using node {}... {}",
|
||||
command, LogHelper.toString(params), source, connection.getRedisClient().getAddr(), connection);
|
||||
}
|
||||
|
||||
if (connectionEntry.isFirstCommand()) {
|
||||
List<CommandData<?, ?>> list = new ArrayList<CommandData<?, ?>>(2);
|
||||
list.add(new CommandData<Void, Void>(new RedissonPromise<Void>(), codec, RedisCommands.MULTI, new Object[]{}));
|
||||
list.add(new CommandData<V, R>(attemptPromise, codec, command, params));
|
||||
RPromise<Void> main = new RedissonPromise<Void>();
|
||||
writeFuture = connection.send(new CommandsData(main, list, true));
|
||||
connectionEntry.setFirstCommand(false);
|
||||
} else {
|
||||
if (RedisCommands.EXEC.getName().equals(command.getName())) {
|
||||
Entry entry = commands.get(source.getEntry());
|
||||
|
||||
List<CommandData<?, ?>> list = new ArrayList<>();
|
||||
|
||||
if (options.isSkipResult()) {
|
||||
list.add(new CommandData<Void, Void>(new RedissonPromise<Void>(), codec, RedisCommands.CLIENT_REPLY, new Object[]{ "OFF" }));
|
||||
}
|
||||
|
||||
list.add(new CommandData<V, R>(attemptPromise, codec, command, params));
|
||||
|
||||
if (options.isSkipResult()) {
|
||||
list.add(new CommandData<Void, Void>(new RedissonPromise<Void>(), codec, RedisCommands.CLIENT_REPLY, new Object[]{ "ON" }));
|
||||
}
|
||||
if (options.getSyncSlaves() > 0) {
|
||||
BatchCommandData<?, ?> waitCommand = new BatchCommandData(RedisCommands.WAIT,
|
||||
new Object[] { this.options.getSyncSlaves(), this.options.getSyncTimeout() }, index.incrementAndGet());
|
||||
list.add(waitCommand);
|
||||
entry.getCommands().add(waitCommand);
|
||||
}
|
||||
|
||||
RPromise<Void> main = new RedissonPromise<Void>();
|
||||
writeFuture = connection.send(new CommandsData(main, list, new ArrayList(entry.getCommands()), options.isSkipResult(), false, true));
|
||||
} else {
|
||||
RPromise<Void> main = new RedissonPromise<Void>();
|
||||
List<CommandData<?, ?>> list = new ArrayList<>();
|
||||
list.add(new CommandData<V, R>(attemptPromise, codec, command, params));
|
||||
writeFuture = connection.send(new CommandsData(main, list, true));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
protected RFuture<RedisConnection> getConnection() {
|
||||
ConnectionEntry entry = connections.get(source.getEntry());
|
||||
if (entry == null) {
|
||||
entry = new ConnectionEntry();
|
||||
ConnectionEntry oldEntry = connections.putIfAbsent(source.getEntry(), entry);
|
||||
if (oldEntry != null) {
|
||||
entry = oldEntry;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
if (entry.getConnectionFuture() != null) {
|
||||
return entry.getConnectionFuture();
|
||||
}
|
||||
|
||||
synchronized (this) {
|
||||
if (entry.getConnectionFuture() != null) {
|
||||
return entry.getConnectionFuture();
|
||||
}
|
||||
|
||||
RFuture<RedisConnection> connectionFuture;
|
||||
if (this.options.getExecutionMode() == ExecutionMode.REDIS_WRITE_ATOMIC) {
|
||||
connectionFuture = connectionManager.connectionWriteOp(source, null);
|
||||
} else {
|
||||
connectionFuture = connectionManager.connectionReadOp(source, null);
|
||||
}
|
||||
connectionFuture.syncUninterruptibly();
|
||||
entry.setConnectionFuture(connectionFuture);
|
||||
return connectionFuture;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue