com.redislabs.riot.redis.AbstractRedisCommand Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of riot-redis Show documentation

RIOT Core module

The newest version!

package com.redislabs.riot.redis;

import com.redislabs.riot.AbstractTaskCommand;
import com.redislabs.riot.RedisOptions;
import io.lettuce.core.AbstractRedisClient;
import io.lettuce.core.RedisClient;
import io.lettuce.core.api.StatefulConnection;
import io.lettuce.core.api.StatefulRedisConnection;
import io.lettuce.core.api.sync.BaseRedisCommands;
import io.lettuce.core.cluster.RedisClusterClient;
import io.lettuce.core.cluster.api.StatefulRedisClusterConnection;
import org.springframework.batch.core.configuration.annotation.StepBuilderFactory;
import org.springframework.batch.core.job.flow.Flow;
import org.springframework.batch.repeat.RepeatStatus;
import org.springframework.util.ClassUtils;
import picocli.CommandLine.Command;

@Command
public abstract class AbstractRedisCommand extends AbstractTaskCommand {

    @Override
    protected Flow flow(StepBuilderFactory stepBuilderFactory) {

        return flow(stepBuilderFactory.get(ClassUtils.getShortName(getClass()) + "-step").tasklet((contribution, chunkContext) -> {
            RedisOptions redisOptions = getRedisOptions();
            AbstractRedisClient client = redisOptions.client();
            try {
                try (StatefulConnection connection = redisOptions.isCluster() ? ((RedisClusterClient) client).connect() : ((RedisClient) client).connect()) {
                    BaseRedisCommands commands = redisOptions.isCluster() ? ((StatefulRedisClusterConnection) connection).sync() : ((StatefulRedisConnection) connection).sync();
                    execute(commands);
                    return RepeatStatus.FINISHED;
                }
            } finally {
                RedisOptions.shutdown(client);
            }
        }).build());
    }

    protected abstract void execute(BaseRedisCommands commands);


}