org.redisson.RedissonBatch Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2018 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;

import java.util.concurrent.TimeUnit;

import org.redisson.api.BatchOptions;
import org.redisson.api.BatchResult;
import org.redisson.api.RAtomicDoubleAsync;
import org.redisson.api.RAtomicLongAsync;
import org.redisson.api.RBatch;
import org.redisson.api.RBitSetAsync;
import org.redisson.api.RBlockingDequeAsync;
import org.redisson.api.RBlockingQueueAsync;
import org.redisson.api.RBucketAsync;
import org.redisson.api.RDequeAsync;
import org.redisson.api.RFuture;
import org.redisson.api.RGeoAsync;
import org.redisson.api.RHyperLogLogAsync;
import org.redisson.api.RKeysAsync;
import org.redisson.api.RLexSortedSetAsync;
import org.redisson.api.RListAsync;
import org.redisson.api.RMapAsync;
import org.redisson.api.RMapCacheAsync;
import org.redisson.api.RMultimapAsync;
import org.redisson.api.RMultimapCacheAsync;
import org.redisson.api.RQueueAsync;
import org.redisson.api.RScoredSortedSetAsync;
import org.redisson.api.RScript;
import org.redisson.api.RScriptAsync;
import org.redisson.api.RSetAsync;
import org.redisson.api.RSetCacheAsync;
import org.redisson.api.RStreamAsync;
import org.redisson.api.RTopicAsync;
import org.redisson.client.codec.Codec;
import org.redisson.command.CommandBatchService;
import org.redisson.connection.ConnectionManager;
import org.redisson.eviction.EvictionScheduler;

/**
 *
 *
 * @author Nikita Koksharov
 *
 */
public class RedissonBatch implements RBatch {

    private final EvictionScheduler evictionScheduler;
    private final CommandBatchService executorService;
    private final BatchOptions options;

    public RedissonBatch(EvictionScheduler evictionScheduler, ConnectionManager connectionManager, BatchOptions options) {
        this.executorService = new CommandBatchService(connectionManager, options);
        this.evictionScheduler = evictionScheduler;
        this.options = options;
    }

    @Override
    public  RBucketAsync getBucket(String name) {
        return new RedissonBucket(executorService, name);
    }

    @Override
    public  RBucketAsync getBucket(String name, Codec codec) {
        return new RedissonBucket(codec, executorService, name);
    }

    @Override
    public  RHyperLogLogAsync getHyperLogLog(String name) {
        return new RedissonHyperLogLog(executorService, name);
    }

    @Override
    public  RHyperLogLogAsync getHyperLogLog(String name, Codec codec) {
        return new RedissonHyperLogLog(codec, executorService, name);
    }

    @Override
    public  RListAsync getList(String name) {
        return new RedissonList(executorService, name, null);
    }

    @Override
    public  RListAsync getList(String name, Codec codec) {
        return new RedissonList(codec, executorService, name, null);
    }

    @Override
    public  RMapAsync getMap(String name) {
        return new RedissonMap(executorService, name, null, null);
    }

    @Override
    public  RMapAsync getMap(String name, Codec codec) {
        return new RedissonMap(codec, executorService, name, null, null);
    }

    @Override
    public  RSetAsync getSet(String name) {
        return new RedissonSet(executorService, name, null);
    }

    @Override
    public  RSetAsync getSet(String name, Codec codec) {
        return new RedissonSet(codec, executorService, name, null);
    }

    @Override
    public RTopicAsync getTopic(String name) {
        return new RedissonTopic(executorService, name);
    }

    @Override
    public RTopicAsync getTopic(String name, Codec codec) {
        return new RedissonTopic(codec, executorService, name);
    }

    @Override
    public  RQueueAsync getQueue(String name) {
        return new RedissonQueue(executorService, name, null);
    }

    @Override
    public  RQueueAsync getQueue(String name, Codec codec) {
        return new RedissonQueue(codec, executorService, name, null);
    }

    @Override
    public  RBlockingQueueAsync getBlockingQueue(String name) {
        return new RedissonBlockingQueue(executorService, name, null);
    }

    @Override
    public  RBlockingQueueAsync getBlockingQueue(String name, Codec codec) {
        return new RedissonBlockingQueue(codec, executorService, name, null);
    }

    @Override
    public  RBlockingDequeAsync getBlockingDeque(String name) {
        return new RedissonBlockingDeque(executorService, name, null);
    }

    @Override
    public  RBlockingDequeAsync getBlockingDeque(String name, Codec codec) {
        return new RedissonBlockingDeque(codec, executorService, name, null);
    }

    @Override
    public  RDequeAsync getDeque(String name) {
        return new RedissonDeque(executorService, name, null);
    }

    @Override
    public  RDequeAsync getDeque(String name, Codec codec) {
        return new RedissonDeque(codec, executorService, name, null);
    }

    @Override
    public RAtomicLongAsync getAtomicLong(String name) {
        return new RedissonAtomicLong(executorService, name);
    }

    @Override
    public RAtomicDoubleAsync getAtomicDouble(String name) {
        return new RedissonAtomicDouble(executorService, name);
    }

    @Override
    public  RScoredSortedSetAsync getScoredSortedSet(String name) {
        return new RedissonScoredSortedSet(executorService, name, null);
    }

    @Override
    public  RScoredSortedSetAsync getScoredSortedSet(String name, Codec codec) {
        return new RedissonScoredSortedSet(codec, executorService, name, null);
    }

    @Override
    public RLexSortedSetAsync getLexSortedSet(String name) {
        return new RedissonLexSortedSet(executorService, name, null);
    }

    @Override
    public RBitSetAsync getBitSet(String name) {
        return new RedissonBitSet(executorService, name);
    }

    @Override
    public  RMapCacheAsync getMapCache(String name, Codec codec) {
        return new RedissonMapCache(codec, evictionScheduler, executorService, name, null, null);
    }

    @Override
    public  RMapCacheAsync getMapCache(String name) {
        return new RedissonMapCache(evictionScheduler, executorService, name, null, null);
    }

    @Override
    public RScriptAsync getScript() {
        return new RedissonScript(executorService);
    }
    
    @Override
    public RScript getScript(Codec codec) {
        return new RedissonScript(executorService, codec);
    }

    @Override
    public RKeysAsync getKeys() {
        return new RedissonKeys(executorService);
    }

    @Override
    public  RSetCacheAsync getSetCache(String name) {
        return new RedissonSetCache(evictionScheduler, executorService, name, null);
    }

    @Override
    public  RSetCacheAsync getSetCache(String name, Codec codec) {
        return new RedissonSetCache(codec, evictionScheduler, executorService, name, null);
    }

    @Override
    public RBatch syncSlaves(int slaves, long timeout, TimeUnit unit) {
        options.syncSlaves(slaves, timeout, unit);
        return this;
    }
    
    @Override
    public RBatch atomic() {
        options.atomic();
        return this;
    }
    
    @Override
    public RBatch skipResult() {
        options.skipResult();
        return this;
    }
    
    @Override
    public RBatch retryAttempts(int retryAttempts) {
        options.retryAttempts(retryAttempts);
        return this;
    }
    
    @Override
    public RBatch retryInterval(long retryInterval, TimeUnit unit) {
        options.retryInterval(retryInterval, unit);
        return this;
    }
    
    @Override
    public RBatch timeout(long timeout, TimeUnit unit) {
        options.responseTimeout(timeout, unit);
        return this;
    }
    
    @Override
    public BatchResult execute() {
        return executorService.execute(BatchOptions.defaults());
    }

    @Override
    public RFuture> executeAsync() {
        return executorService.executeAsync(BatchOptions.defaults());
    }
    
    @Override
    public  RMultimapAsync getSetMultimap(String name) {
        return new RedissonSetMultimap(executorService, name);
    }

    @Override
    public  RMultimapAsync getSetMultimap(String name, Codec codec) {
        return new RedissonSetMultimap(codec, executorService, name);
    }

    @Override
    public  RMultimapAsync getListMultimap(String name) {
        return new RedissonListMultimap(executorService, name);
    }

    @Override
    public  RMultimapAsync getListMultimap(String name, Codec codec) {
        return new RedissonListMultimap(codec, executorService, name);
    }

    @Override
    public  RGeoAsync getGeo(String name) {
        return new RedissonGeo(executorService, name, null);
    }
    
    @Override
    public  RGeoAsync getGeo(String name, Codec codec) {
        return new RedissonGeo(codec, executorService, name, null);
    }
    
    @Override
    public  RMultimapCacheAsync getSetMultimapCache(String name) {
        return new RedissonSetMultimapCache(evictionScheduler, executorService, name);
    }
    
    @Override
    public  RMultimapCacheAsync getSetMultimapCache(String name, Codec codec) {
        return new RedissonSetMultimapCache(evictionScheduler, codec, executorService, name);
    }

    @Override
    public  RMultimapCacheAsync getListMultimapCache(String name) {
        return new RedissonListMultimapCache(evictionScheduler, executorService, name);
    }
    
    @Override
    public  RMultimapCacheAsync getListMultimapCache(String name, Codec codec) {
        return new RedissonListMultimapCache(evictionScheduler, codec, executorService, name);
    }

    protected void enableRedissonReferenceSupport(Redisson redisson) {
        this.executorService.enableRedissonReferenceSupport(redisson);
    }

    @Override
    public  RStreamAsync getStream(String name) {
        return new RedissonStream(executorService, name);
    }

    @Override
    public  RStreamAsync getStream(String name, Codec codec) {  
        return new RedissonStream(codec, executorService, name);
    }

}