com.redis.spring.batch.writer.operation.Expire 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

The newest version!

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

import java.time.Duration;
import java.util.function.Function;

import org.springframework.batch.item.Chunk;

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

public class Expire extends AbstractKeyWriteOperation {

	private Function ttlFunction = t -> null;

	public Expire(Function keyFunction) {
		super(keyFunction);
	}

	public void setTtl(Duration duration) {
		this.ttlFunction = t -> duration;
	}

	public void setTtlFunction(Function function) {
		this.ttlFunction = function;
	}

	@SuppressWarnings({ "unchecked", "rawtypes" })
	@Override
	protected void execute(BaseRedisAsyncCommands commands, T item, K key, Chunk> outputs) {
		Duration duration = ttlFunction.apply(item);
		if (duration != null && !duration.isNegative() && !duration.isZero()) {
			outputs.add((RedisFuture) ((RedisKeyAsyncCommands) commands).pexpire(key, duration));
		}
	}

}