com.github.obase.jedis.impl.JedisClientShardingImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of spring-jedis Show documentation

A simple jedis client

The newest version!

package com.github.obase.jedis.impl;

import static redis.clients.util.Hashing.MURMUR_HASH;

import java.math.BigDecimal;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

import com.github.obase.jedis.JedisClient;
import com.github.obase.jedis.JedisClientException;
import com.github.obase.jedis.JedisClientSharding;
import com.github.obase.jedis.ShardInfo;

public class JedisClientShardingImpl implements JedisClientSharding {

	final List shards;
	final TreeMap clients = new TreeMap();

	public JedisClientShardingImpl(List shards) throws JedisClientException {
		this.shards = shards;

		if (shards != null) {
			for (ShardInfo info : shards) {
				if (info.getHash() == null) {
					if (info.getRate() < 0 || info.getRate() > 100) {
						throw new JedisClientException("Invalid shard rate: " + info.getRate());
					}
					info.setHash(convRateToLong(info.getRate()));
				}
				if (clients.put(info.getHash(), new JedisClientImpl(info.getJedisPool())) != null) {
					throw new JedisClientException("Duplicate shard hash: " + info.getHash());
				}
			}
		}
	}

	private static long convRateToLong(byte rate) {
		BigDecimal min = BigDecimal.valueOf(Integer.MIN_VALUE);
		return BigDecimal.valueOf(Integer.MAX_VALUE).subtract(min).divide(BigDecimal.valueOf(100)).multiply(BigDecimal.valueOf(rate)).add(min).longValue();
	}

	@Override
	public JedisClient shard(long key) {
		Map.Entry entry = clients.floorEntry(key);
		if (entry == null) {
			entry = clients.firstEntry();
		}
		return entry.getValue();
	}

	@Override
	public JedisClient shard(String key) {
		return shard(MURMUR_HASH.hash(key));
	}

	@Override
	public JedisClient shard(byte[] key) {
		return shard(MURMUR_HASH.hash(key));
	}

	@Override
	public Collection getAllShardInfo() {
		return shards;
	}

	@Override
	public Collection getAllShard() {
		return clients.values();
	}

}