一致性哈希算法的简单实现和在redis集群的应用

    项目中需要搭建一个简单的redis集群，用来存储系统中的特征值，利用到一致性哈希算法，简介可以看一致性哈希算法，本文简单实现了一致性哈希的算法，主要阐述一下应用场景和一些感想。

#ifndef _CONSISTENTHASH_H #define _CONSISTENTHASH_H #include<vector> #include<map> #include<utility> #include<string> using namespace std; class ConsistentHash { public: ConsistentHash(); ~ConsistentHash(); ConsistentHash(int cachedNodeNum, int virtualNodeNum); int getCachedNodeId(uint64_t key); int getCachedNodeId(string key); private: void getVirtualNodeForString(int cachedNodeNum, int virtualNodeNum, string &virtualNodeString); private: vector<uint64_t> vHashValues; map<uint64_t, pair<int int=""> > mVirtualNodeMap; int cachedNodeNum; int virtualNodeNum; }; #endif

#include"ConsistentHash.h" #include"city.h" #include<iostream> #include<algorithm> #include<vector> using namespace std; ConsistentHash::ConsistentHash() { } ConsistentHash::~ConsistentHash() { } /* * cachedNodeNum:cache节点的数量 * virtualNodeNum:虚拟节点的个数 * 根据cache节点数量和虚拟节点个数生成环 */ ConsistentHash::ConsistentHash(int cachedNodeNum, int virtualNodeNum) { this->cachedNodeNum = cachedNodeNum; this->virtualNodeNum = virtualNodeNum; for(int i = 0; i < cachedNodeNum; ++i) { for(int j = 0; j < virtualNodeNum; ++j) { string hashStr = ""; uint64 hashInt = 0; //可以根据节点信息计算hash值 getVirtualNodeForString(i, j, hashStr); //利用Google的CityHash算法计算hash值 hashInt = CityHash64(hashStr.c_str(), hashStr.size()); vHashValues.push_back(hashInt); mVirtualNodeMap[hashInt] = make_pair(i, j); } } sort(vHashValues.begin(), vHashValues.end()); } /* * key:要存储的字符串 * 获取该字符串要存储的cache节点id */ int ConsistentHash::getCachedNodeId(string key) { vector<uint64_t>::iterator low; uint64_t hashInt = CityHash64(key.c_str(), key.size()); low = lower_bound(vHashValues.begin(), vHashValues.end(), hashInt); uint64_t nextHash; if(low == vHashValues.end()) { nextHash = vHashValues[0]; } else { nextHash = vHashValues[low - vHashValues.begin()]; } return mVirtualNodeMap[nextHash].first; } int ConsistentHash::getCachedNodeId(uint64_t key) { vector<uint64_t>::iterator low; uint64_t hashInt = key; low = lower_bound(vHashValues.begin(), vHashValues.end(), hashInt); uint64_t nextHash; if(low == vHashValues.end()) { nextHash = vHashValues[0]; } else { nextHash = vHashValues[low - vHashValues.begin()]; } return mVirtualNodeMap[nextHash].first; } void ConsistentHash::getVirtualNodeForString(int cachedNodeNum, int virtualNodeNum, string &virtualNodeString) { virtualNodeString = ("node:" + cachedNodeNum); virtualNodeString += ":vnode:"; virtualNodeString = virtualNodeString + ("" + virtualNodeNum); }

以上代码比较简单的实现了一致性哈希的算法。算法对来自客户端的请求进行分发，实现数据的存储和读取。redis也提供了相关的备份机制，但是是以牺牲了一部分的性能为代价的，结合我们的应用场景，并没有采用相关的备份机制，即使redis挂了，将模型重新计算和相关的特征值存入就可以了，因为特征值丢失的代价是我们可以承受的，而且模型更新的间隔比较短，所以就没有采用相关的备份机制，后续我们加入了双写的功能，即一份数据存储到不同的两个节点上，只需要在原有代码基础上稍加改动就行了，这样即使一个redis挂了，可以到另一个节点读取数据。