com.redislabs.riot.redis.writer.map.AbstractFtAdd Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of riot Show documentation

Redis Input/Output Tool

There is a newer version: 1.11.2

package com.redislabs.riot.redis.writer.map;

import java.util.Map;

import com.redislabs.lettusearch.search.AddOptions;
import com.redislabs.riot.redis.RedisCommands;
import com.redislabs.riot.redis.writer.KeyBuilder;
import com.redislabs.riot.redis.writer.RediSearchCommandWriter;

import lombok.Setter;

@SuppressWarnings({ "rawtypes", "unchecked" })
public abstract class AbstractFtAdd extends AbstractKeyMapCommandWriter
		implements RediSearchCommandWriter> {

	private @Setter String index;
	private @Setter AddOptions options;
	private @Setter String score;
	private @Setter double defaultScore = 1d;

	protected AbstractFtAdd(KeyBuilder keyBuilder, boolean keepKeyFields, String index, String score,
			double defaultScore, AddOptions options) {
		super(keyBuilder, keepKeyFields);
		this.index = index;
		this.score = score;
		this.defaultScore = defaultScore;
		this.options = options;
	}

	@Override
	protected Object write(RedisCommands commands, Object redis, String key, Map item) {
		return commands.ftadd(redis, index, key, convert(item.getOrDefault(score, defaultScore), Double.class),
				stringMap(item), payload(item), options);
	}

	protected abstract String payload(Map item);

}