Java使用TreeMap實現(xiàn)一致性hash算法

關(guān)于一致性Hash算法原理可參考文章:一致性hash算法原理與實現(xiàn)

public class TreeMapConsistentHash {

    /**
     * 虛擬節(jié)點數(shù)量
     */
    private static final int VIRTUAL_NODE_SIZE = 4;

    /**
     * 定義treeMap
     */
    private TreeMap<Long, String> tree = new TreeMap<>();

    /**
     * 在服務(wù)器列表中根據(jù)key定位節(jié)點
     * @param values 服務(wù)器列表
     * @param key token
     * @return
     */
    public String choose(List<String> values, String key) {
        for (String value: values) {
           //value作為key,hash值作為value
            add(hash(value), value);
        }

        return selectNode(key);
    }

    /**
     * hash落環(huán)走越,并加入虛擬節(jié)點
     * @param key
     * @param value
     */
    private void add(long key, String value) {
        tree.clear();
        //虛擬節(jié)點
        for (int i = 0; i < VIRTUAL_NODE_SIZE; i++) {
            Long hash = this.hash("vir" + key + i);
            tree.put(hash, value);
        }
        tree.put(key, value);
    }

    /**
     * 在環(huán)中根據(jù)傳入的值找到第一個server
     * 使用tailMap特性模擬hash環(huán)旨指,如果tailMap返回空,則表示已經(jīng)到了hash環(huán)的末尾谆构,那么需要使用第一個key
     * @param value
     * @return
     */
    private String selectNode(String value) {
        long hash = hash(value);
        SortedMap<Long, String> after = tree.tailMap(hash);
        if (after != null && !after.isEmpty()) {

            String server = after.get(after.firstKey());
            System.out.println("路由成功:value: " + value + ", route server: " + server );
            return server;
        }
        return tree.firstEntry().getValue();
    }

    /**
     * hash計算
     * @param value
     * @return
     */
    private Long hash(String value) {
        byte[] digest = DigestUtils.md5(value);

        // hash code, Truncate to 32-bits
        long hashCode = ((long) (digest[3] & 0xFF) << 24) | ((long) (digest[2] & 0xFF) << 16) | ((long) (digest[1] & 0xFF) << 8) | (digest[0] & 0xFF);

        long truncateHashCode = hashCode & 0xffffffffL;
        return truncateHashCode;
    }
}

如何使用
比如我們使用zookeeper作為注冊中心呵晨,那么將應(yīng)用注冊到zookeeper的特定節(jié)點下摸屠,取出節(jié)點下所有應(yīng)用的服務(wù)地址猪杭,根據(jù)consumer方的
key hash值來選擇具體調(diào)用那個provider

public class ConsistentHashRouter implements Router {

    private TreeMapConsistentHash hash = new TreeMapConsistentHash();

    /**
     * values: List<String> servers = Lists.newArrayList("127.0.0.1","192.168.11.1" ...)
     * key: consumer的特定數(shù)值妥衣,比如根據(jù)userId
     **/
    @Override
    public String chooseServer(List<String> values, String key) {
        return hash.choose(values,key);
    }
}
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末蜂筹,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子艺挪,更是在濱河造成了極大的恐慌兵扬,老刑警劉巖器钟,帶你破解...
    沈念sama閱讀 211,123評論 6 490
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異傲霸,居然都是意外死亡昙啄,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,031評論 2 384
  • 文/潘曉璐 我一進店門耿币,熙熙樓的掌柜王于貴愁眉苦臉地迎上來淹接,“玉大人,你說我怎么就攤上這事烁试÷K粒” “怎么了?”我有些...
    開封第一講書人閱讀 156,723評論 0 345
  • 文/不壞的土叔 我叫張陵支示,是天一觀的道長颂鸿。 經(jīng)常有香客問我嘴纺,道長浓冒,這世上最難降的妖魔是什么稳懒? 我笑而不...
    開封第一講書人閱讀 56,357評論 1 283
  • 正文 為了忘掉前任,我火速辦了婚禮墅冷,結(jié)果婚禮上寞忿,老公的妹妹穿的比我還像新娘装哆。我一直安慰自己,他們只是感情好蜕琴,可當(dāng)我...
    茶點故事閱讀 65,412評論 5 384
  • 文/花漫 我一把揭開白布萍桌。 她就那樣靜靜地躺著,像睡著了一般凌简。 火紅的嫁衣襯著肌膚如雪上炎。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,760評論 1 289
  • 那天,我揣著相機與錄音藕施,去河邊找鬼寇损。 笑死,一個胖子當(dāng)著我的面吹牛裳食,可吹牛的內(nèi)容都是我干的矛市。 我是一名探鬼主播,決...
    沈念sama閱讀 38,904評論 3 405
  • 文/蒼蘭香墨 我猛地睜開眼诲祸,長吁一口氣:“原來是場噩夢啊……” “哼浊吏!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起找田,我...
    開封第一講書人閱讀 37,672評論 0 266
  • 序言:老撾萬榮一對情侶失蹤漆改,失蹤者是張志新(化名)和其女友劉穎于个,沒想到半個月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 44,118評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡档押,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,456評論 2 325
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了筛婉。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片入蛆。...
    茶點故事閱讀 38,599評論 1 340
  • 序言:一個原本活蹦亂跳的男人離奇死亡源武,死狀恐怖迎捺,靈堂內(nèi)的尸體忽然破棺而出凳枝,到底是詐尸還是另有隱情砂代,我是刑警寧澤刻伊,帶...
    沈念sama閱讀 34,264評論 4 328
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響证九,放射性物質(zhì)發(fā)生泄漏愧怜。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點故事閱讀 39,857評論 3 312
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望雪位。 院中可真熱鬧,春花似錦时肿、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,731評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽鞋仍。三九已至,卻和暖如春谎懦,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背截碴。 一陣腳步聲響...
    開封第一講書人閱讀 31,956評論 1 264
  • 我被黑心中介騙來泰國打工丙躏, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人敢朱。 一個月前我還...
    沈念sama閱讀 46,286評論 2 360
  • 正文 我出身青樓,卻偏偏與公主長得像岸梨,于是被迫代替她去往敵國和親赃份。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 43,465評論 2 348