org.redisson.rx.CommandRxBatchService 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.rx;
import io.reactivex.rxjava3.core.Flowable;
import org.redisson.api.BatchOptions;
import org.redisson.api.BatchResult;
import org.redisson.api.RFuture;
import org.redisson.client.codec.Codec;
import org.redisson.client.protocol.RedisCommand;
import org.redisson.command.BatchService;
import org.redisson.command.CommandAsyncExecutor;
import org.redisson.command.CommandBatchService;
import org.redisson.connection.ConnectionManager;
import org.redisson.connection.NodeSource;
import org.redisson.liveobject.core.RedissonObjectBuilder;
import org.redisson.misc.CompletableFutureWrapper;
import java.util.concurrent.Callable;
import java.util.concurrent.CompletableFuture;
import java.util.concurrent.atomic.AtomicBoolean;
/**
*
* @author Nikita Koksharov
*
*/
public class CommandRxBatchService extends CommandRxService implements BatchService {
private final CommandBatchService batchService;
public CommandRxBatchService(ConnectionManager connectionManager, CommandAsyncExecutor executor, BatchOptions options) {
super(connectionManager, executor.getObjectBuilder());
batchService = new CommandBatchService(executor, options, RedissonObjectBuilder.ReferenceType.RXJAVA);
}
@Override
public Flowable flowable(Callable> supplier) {
Flowable flowable = super.flowable(new Callable>() {
final CompletableFuture future = new CompletableFuture<>();
final AtomicBoolean lock = new AtomicBoolean();
@Override
public RFuture call() throws Exception {
if (lock.compareAndSet(false, true)) {
transfer(supplier.call().toCompletableFuture(), future);
}
return new CompletableFutureWrapper<>(future);
}
});
flowable.subscribe();
return flowable;
}
@Override
protected CompletableFuture createPromise() {
return batchService.createPromise();
}
@Override
public RFuture async(boolean readOnlyMode, NodeSource nodeSource,
Codec codec, RedisCommand command, Object[] params, boolean ignoreRedirect, boolean noRetry) {
return batchService.async(readOnlyMode, nodeSource, codec, command, params, ignoreRedirect, noRetry);
}
public RFuture> executeAsync() {
return batchService.executeAsync();
}
public RFuture discardAsync() {
return batchService.discardAsync();
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy