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

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

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

import org.springframework.util.CollectionUtils;

import com.redis.lettucemod.api.async.RedisTimeSeriesAsyncCommands;
import com.redis.lettucemod.timeseries.AddOptions;
import com.redis.lettucemod.timeseries.Sample;

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

public class TsAddAll implements WriteOperation {

	private final Function keyFunction;
	private final Function> samplesFunction;

	private Function> optionsFunction = t -> null;

	public TsAddAll(Function keyFunction, Function> samplesFunction) {
		this.keyFunction = keyFunction;
		this.samplesFunction = samplesFunction;
	}

	public TsAddAll options(AddOptions options) {
		return options(t -> options);
	}

	public TsAddAll options(Function> function) {
		this.optionsFunction = function;
		return this;
	}

	@SuppressWarnings({ "unchecked", "rawtypes" })
	@Override
	public void execute(BaseRedisAsyncCommands commands, Iterable items,
			List> futures) {
		RedisTimeSeriesAsyncCommands timeseriesCommands = (RedisTimeSeriesAsyncCommands) commands;
		for (T item : items) {
			K key = keyFunction.apply(item);
			AddOptions options = optionsFunction.apply(item);
			Collection samples = samplesFunction.apply(item);
			if (CollectionUtils.isEmpty(samples)) {
				continue;
			}
			for (Sample sample : samples) {
				futures.add((RedisFuture) timeseriesCommands.tsAdd(key, sample, options));
			}
		}
	}

	public static  TsAddAll of(Function key, Function> samples,
			AddOptions options) {
		TsAddAll operation = new TsAddAll<>(key, samples);
		operation.options(options);
		return operation;
	}

}