com.redis.spring.batch.RedisItemWriter 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;

import java.util.function.ToLongFunction;

import com.redis.spring.batch.writer.Operation;
import com.redis.spring.batch.writer.OperationItemWriter;
import com.redis.spring.batch.writer.StructOperation;
import com.redis.spring.batch.writer.operation.Restore;

import io.lettuce.core.AbstractRedisClient;
import io.lettuce.core.codec.RedisCodec;

public class RedisItemWriter extends OperationItemWriter> {

    public enum MergePolicy {
        MERGE, OVERWRITE
    }

    public enum TtlPolicy {
        PROPAGATE, DROP
    }

    public enum StreamIdPolicy {
        PROPAGATE, DROP
    }

    public static final MergePolicy DEFAULT_MERGE_POLICY = MergePolicy.OVERWRITE;

    public static final StreamIdPolicy DEFAULT_STREAM_ID_POLICY = StreamIdPolicy.PROPAGATE;

    public static final TtlPolicy DEFAULT_TTL_POLICY = TtlPolicy.PROPAGATE;

    public static final ValueType DEFAULT_VALUE_TYPE = ValueType.DUMP;

    private TtlPolicy ttlPolicy = DEFAULT_TTL_POLICY;

    private MergePolicy mergePolicy = DEFAULT_MERGE_POLICY;

    private StreamIdPolicy streamIdPolicy = DEFAULT_STREAM_ID_POLICY;

    private ValueType valueType = DEFAULT_VALUE_TYPE;

    public RedisItemWriter(AbstractRedisClient client, RedisCodec codec) {
        super(client, codec);
    }

    public TtlPolicy getTtlPolicy() {
        return ttlPolicy;
    }

    public void setTtlPolicy(TtlPolicy policy) {
        this.ttlPolicy = policy;
    }

    public MergePolicy getMergePolicy() {
        return mergePolicy;
    }

    public void setMergePolicy(MergePolicy policy) {
        this.mergePolicy = policy;
    }

    public StreamIdPolicy getStreamIdPolicy() {
        return streamIdPolicy;
    }

    public void setStreamIdPolicy(StreamIdPolicy policy) {
        this.streamIdPolicy = policy;
    }

    public ValueType getValueType() {
        return valueType;
    }

    public void setValueType(ValueType valueType) {
        this.valueType = valueType;
    }

    @Override
    protected void doOpen() {
        setOperation(operation());
        super.doOpen();
    }

    private Operation> operation() {
        if (valueType == ValueType.DUMP) {
            Restore> operation = new Restore<>();
            operation.setKey(KeyValue::getKey);
            operation.setBytes(v -> (byte[]) v.getValue());
            operation.setTtl(keyValueTtl());
            operation.setReplace(true);
            return operation;
        }
        StructOperation operation = new StructOperation<>();
        operation.mergePolicy(mergePolicy);
        operation.streamIdPolicy(streamIdPolicy);
        operation.ttlPolicy(ttlPolicy);
        return operation;
    }

    private ToLongFunction> keyValueTtl() {
        if (ttlPolicy == TtlPolicy.PROPAGATE) {
            return KeyValue::getTtl;
        }
        return kv -> 0;
    }

}