com.redis.spring.batch.writer.operation.Restore Maven / Gradle / Ivy

Go to download

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

Spring Batch Redis

There is a newer version: 4.0.7

package com.redis.spring.batch.writer.operation;

import java.util.List;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.function.ToLongFunction;

import com.redis.spring.batch.util.Predicates;

import io.lettuce.core.RedisFuture;
import io.lettuce.core.RestoreArgs;
import io.lettuce.core.api.async.BaseRedisAsyncCommands;
import io.lettuce.core.api.async.RedisKeyAsyncCommands;

public class Restore extends AbstractOperation {

    public static final long TTL_KEY_DOES_NOT_EXIST = -2;

    private Function bytes;

    private ToLongFunction absoluteTtl;

    private Predicate replace = Predicates.isFalse();

    public void setBytes(Function function) {
        this.bytes = function;
    }

    public void setTtl(ToLongFunction function) {
        this.absoluteTtl = function;
    }

    public void setReplace(boolean replace) {
        this.replace = Predicates.is(replace);
    }

    @SuppressWarnings("unchecked")
    @Override
    public void execute(BaseRedisAsyncCommands commands, T item, List> futures) {
        byte[] dump = bytes.apply(item);
        long ttl = absoluteTtl.applyAsLong(item);
        if (dump == null || ttl == TTL_KEY_DOES_NOT_EXIST) {
            futures.add(((RedisKeyAsyncCommands) commands).del(key(item)));
        } else {
            RestoreArgs args = new RestoreArgs().absttl().replace(replace(item));
            if (ttl > 0) {
                args.ttl(ttl);
            }
            futures.add(((RedisKeyAsyncCommands) commands).restore(key(item), dump, args));
        }
    }

    private boolean replace(T item) {
        return replace.test(item);
    }

}