com.redis.spring.batch.reader.AbstractKeyValueItemReader 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.reader;

import java.io.IOException;

import org.apache.commons.pool2.impl.GenericObjectPoolConfig;
import org.springframework.batch.item.Chunk;
import org.springframework.util.unit.DataSize;

import com.redis.spring.batch.RedisItemReader;
import com.redis.spring.batch.common.KeyValue;
import com.redis.spring.batch.common.Operation;
import com.redis.spring.batch.common.ValueReader;

import io.lettuce.core.AbstractRedisClient;
import io.lettuce.core.codec.RedisCodec;

public abstract class AbstractKeyValueItemReader extends RedisItemReader> {

	/**
	 * Default to no memory usage calculation
	 */
	public static final DataSize DEFAULT_MEMORY_USAGE_LIMIT = DataSize.ofBytes(0);
	public static final int DEFAULT_MEMORY_USAGE_SAMPLES = 5;
	public static final int DEFAULT_POOL_SIZE = GenericObjectPoolConfig.DEFAULT_MAX_TOTAL;

	private int poolSize = DEFAULT_POOL_SIZE;
	protected DataSize memLimit = DEFAULT_MEMORY_USAGE_LIMIT;
	protected int memSamples = DEFAULT_MEMORY_USAGE_SAMPLES;

	private ValueReader> valueReader;

	protected AbstractKeyValueItemReader(AbstractRedisClient client, RedisCodec codec) {
		super(client, codec);
	}

	@Override
	protected synchronized void doOpen() throws Exception {
		if (valueReader == null) {
			valueReader = valueReader();
			valueReader.open();
		}
		super.doOpen();
	}

	@Override
	protected synchronized void doClose() throws Exception {
		super.doClose();
		if (valueReader != null) {
			valueReader.close();
		}
	}

	public ValueReader> valueReader() throws IOException {
		ValueReader> reader = new ValueReader<>(getClient(), getCodec(), operation());
		reader.setPoolSize(poolSize);
		reader.setReadFrom(getReadFrom());
		return reader;
	}

	@Override
	public Chunk> values(Chunk chunk) {
		return valueReader.execute(chunk);
	}

	protected abstract Operation> operation() throws IOException;

	public void setMemoryUsageLimit(DataSize limit) {
		this.memLimit = limit;
	}

	public void setMemoryUsageSamples(int samples) {
		this.memSamples = samples;
	}

	public void setPoolSize(int poolSize) {
		this.poolSize = poolSize;
	}

}