com.redis.spring.batch.writer.Expire Maven / Gradle / Ivy

Go to download
package com.redis.spring.batch.writer;

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

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

public class Expire extends AbstractKeyOperation {

	private ToLongFunction ttlFunction = t -> 0;

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

	public Expire ttl(long millis) {
		return ttl(t -> millis);
	}

	public Expire ttl(ToLongFunction function) {
		this.ttlFunction = function;
		return this;
	}

	@SuppressWarnings({ "unchecked", "rawtypes" })
	@Override
	protected void execute(BaseRedisAsyncCommands commands, T item, K key, List> outputs) {
		long ttl = ttlFunction.applyAsLong(item);
		if (ttl > 0) {
			outputs.add((RedisFuture) ((RedisKeyAsyncCommands) commands).pexpire(key, ttl));
		}
	}

}