com.redis.spring.batch.writer.operation.XAddAll 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.util.Collection;
import java.util.function.Function;

import org.springframework.batch.item.Chunk;
import org.springframework.util.CollectionUtils;

import com.redis.spring.batch.common.Operation;

import io.lettuce.core.RedisFuture;
import io.lettuce.core.StreamMessage;
import io.lettuce.core.XAddArgs;
import io.lettuce.core.api.async.BaseRedisAsyncCommands;
import io.lettuce.core.api.async.RedisStreamAsyncCommands;

public class XAddAll implements Operation {

	private Function>> messagesFunction;

	private Function, XAddArgs> argsFunction = m -> new XAddArgs().id(m.getId());

	public void setMessagesFunction(Function>> function) {
		this.messagesFunction = function;
	}

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

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

	@SuppressWarnings({ "unchecked", "rawtypes" })
	@Override
	public void execute(BaseRedisAsyncCommands commands, Chunk items,
			Chunk> futures) {
		RedisStreamAsyncCommands streamCommands = (RedisStreamAsyncCommands) commands;
		for (T item : items) {
			Collection> messages = messagesFunction.apply(item);
			if (CollectionUtils.isEmpty(messages)) {
				continue;
			}
			for (StreamMessage message : messages) {
				XAddArgs args = argsFunction.apply(message);
				futures.add((RedisFuture) streamCommands.xadd(message.getStream(), args, message.getBody()));
			}
		}
	}

}