com.redis.spring.batch.writer.operation.ZaddAll 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

There is a newer version: 4.0.7

package com.redis.spring.batch.writer.operation;

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

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 AbstractOperation {

    private Function>> values;

    private ZAddArgs args;

    public void setArgs(ZAddArgs args) {
        this.args = args;
    }

    public void setValues(Function>> values) {
        this.values = values;
    }

    @SuppressWarnings("unchecked")
    @Override
    public void execute(BaseRedisAsyncCommands commands, T item, List> futures) {
        Collection> collection = values(item);
        if (!collection.isEmpty()) {
            futures.add(((RedisSortedSetAsyncCommands) commands).zadd(key(item), args,
                    collection.toArray(new ScoredValue[0])));
        }
    }

    private Collection> values(T item) {
        return values.apply(item);
    }

}