io.craft.atom.redis.AbstractMurmurHashSharded Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of craft-atom-redis Show documentation

There is a newer version: 3.1.2

package io.craft.atom.redis;

import io.craft.atom.redis.api.RedisCommand;

import java.util.ArrayList;
import java.util.List;
import java.util.SortedMap;
import java.util.TreeMap;

import lombok.ToString;


/**
 * @author mindwind
 * @version 1.0, Jun 25, 2013
 */
@ToString(of = {"shards"})
public abstract class AbstractMurmurHashSharded {
	
	
	private TreeMap nodes ;
	private List          shards; 
	private MurmurHash       murmur;
	
	
	// ~ ------------------------------------------------------------------------------------------------------------
	
	
	public AbstractMurmurHashSharded(List shards) {
		this.shards = shards;
		this.murmur = new MurmurHash();
		init(shards);
	}
	
	private void init(List shards) {
		nodes = new TreeMap();
		for (int i = 0; i < shards.size(); i++) {
			R redis = shards.get(i);
            for (int n = 0; n < 160; n++) {
            	Long k = murmur.hash("SHARD-" + i + "-NODE-" + n);
            	nodes.put(k, redis);
            }
        }
	}

	public R shard(String shardkey) {
		SortedMap tail = nodes.tailMap(murmur.hash(shardkey));
		if (tail.isEmpty()) {
			return nodes.get(nodes.firstKey());
		}
		return tail.get(tail.firstKey());
	}

	public List shards() {
		return new ArrayList(shards);
	}

}