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

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

import java.util.Collection;
import java.util.function.Function;

import org.springframework.util.CollectionUtils;

import io.lettuce.core.RedisFuture;
import io.lettuce.core.ScoredValue;
import io.lettuce.core.ZAddArgs;
import io.lettuce.core.api.async.BaseRedisAsyncCommands;
import io.lettuce.core.api.async.RedisSortedSetAsyncCommands;

public class ZaddAll extends AbstractKeyValueOperation>, T> {

	private Function argsFunction = t -> null;

	public ZaddAll(Function keyFunction, Function>> valuesFunction) {
		super(keyFunction, valuesFunction, CollectionUtils::isEmpty);
	}

	public void setArgs(ZAddArgs args) {
		this.argsFunction = t -> args;
	}

	public void setArgsFunction(Function function) {
		this.argsFunction = function;
	}

	@SuppressWarnings({ "unchecked", "rawtypes" })
	@Override
	protected RedisFuture execute(BaseRedisAsyncCommands commands, T item, K key,
			Collection> value) {
		return ((RedisSortedSetAsyncCommands) commands).zadd(key, argsFunction.apply(item),
				value.toArray(new ScoredValue[0]));
	}

}