com.redislabs.riot.batch.redis.reader.JedisHashReader Maven / Gradle / Ivy

Go to download

package com.redislabs.riot.batch.redis.reader;

import java.util.Map;

import org.springframework.batch.item.support.AbstractItemCountingItemStreamItemReader;
import org.springframework.util.Assert;
import org.springframework.util.ClassUtils;

import lombok.Setter;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.util.Pool;

public class JedisHashReader extends AbstractItemCountingItemStreamItemReader> {

	@Setter
	private Integer count;
	@Setter
	private String match;
	@Setter
	private String separator;
	@Setter
	private String keyspace;
	@Setter
	private String[] keys;
	private Pool jedisPool;
	private volatile boolean initialized = false;
	private Object lock = new Object();
	private JedisKeyScanIterator redisIterator;
	private Jedis jedis;

	public JedisHashReader(Pool jedisPool) {
		setName(ClassUtils.getShortName(JedisHashReader.class));
		this.jedisPool = jedisPool;
	}

	@Override
	protected void doOpen() throws Exception {
		Assert.state(!initialized, "Cannot open an already open ItemReader, call close first");
		jedis = jedisPool.getResource();
		redisIterator = new JedisKeyScanIterator(jedis, count, match);
		initialized = true;
	}

	@Override
	protected void doClose() throws Exception {
		synchronized (lock) {
			redisIterator = null;
			jedis.close();
			jedisPool.close();
			jedis = null;
			jedisPool = null;
			initialized = false;
		}
	}

	@SuppressWarnings({ "rawtypes", "unchecked" })
	@Override
	protected Map doRead() throws Exception {
		synchronized (lock) {
			if (redisIterator.hasNext()) {
				String key = redisIterator.next();
				Map map = (Map) jedis.hgetAll(key);
				if (keys.length > 0) {
					String[] keyValues = key.split(separator);
					for (int index = 0; index < keys.length; index++) {
						String keyName = keys[index];
						String keyValue = keyValues[keyspace == null ? index : index + 1];
						map.put(keyName, keyValue);
					}
				}
				return map;
			}
		}
		return null;
	}

}