
com.redis.spring.batch.writer.operation.XAddAll Maven / Gradle / Ivy
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 extends T> 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()));
}
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy