Java实现的简单双向Map，支持重复Value

Java  /  管理员 发布于 1年前   248

关键字：Java双向Map、DualHashBidiMap

 

 

有个需求，需要根据即时修改Map结构中的Value值，比如，将Map中所有value=V1的记录改成value=V2，key保持不变。

 

数据量比较大，遍历Map性能太差，这就需要根据Value先找到Key，然后去修改。

 

即：既要根据Key找Value，又要根据Value找Key。

 

commons-collections中的DualHashBidiMap实现了双向Map的功能，但悲剧的是，Value必须唯一。

 

自己简单实现了一个双向Map,支持根据Key和Value查找，核心思想是相当于额外保存了一份V->K的映射关系，当根据V查找时候，

 

先快速找到哪些V存在于哪些K中，然后再根据这些K去找，就很快了。类似于索引表。

 

代码不够完善，发现Bug还望多多指正。

 

package com.lxw1234.map import java.util.HashMap;import java.util.Map;import java.util.Set;import java.util.concurrent.ConcurrentHashMap;/**** @author lxw的大数据田地 - http://lxw1234.com* @param* @param*/public class MyMap<K,V> {private Map<K,V> kMap = new HashMap<K,V>();private Map<V,ConcurrentHashMap<K,String>> vMap = new HashMap<V,ConcurrentHashMap<K,String>>();private final static String VV = "1";public boolean put(K k,V v) {boolean result = false;if(null == k || null == v) return result;if(kMap.containsKey(k)) {vMap.get(kMap.get(k)).remove(k);//vMap.remove(kMap.get(k));}kMap.put(k, v);if(vMap.containsKey(v)) {vMap.get(v).put(k,VV);} else {ConcurrentHashMap<K,String> set = new ConcurrentHashMap<K,String>();set.put(k,VV);vMap.put(v, set);}return true;}public boolean containsKey(K k) {return kMap.containsKey(k);}public boolean containsValue(V v) {return vMap.containsKey(v);}public Set<K> keySet(){return kMap.keySet();}public Set<V> valueSet() {return vMap.keySet();}public V getByKey(K k) {return kMap.get(k);}public ConcurrentHashMap<K,String> getByValue(V v) {return vMap.get(v);}public String toString() {return kMap.toString();}public static void main(String[] args) {MyMap<String,String> map = new MyMap<String,String>();map.put("k1", "v1");map.put("k2", "v2");map.put("k3", "v1");map.put("k4", "v2");System.out.println("k1 -> " + map.getByKey("k1"));System.out.println("v1 -> " + map.getByValue("v1").keySet());System.out.println("v2 -> " + map.getByValue("v2").keySet());map.put("k5", "v2");System.out.println("v2 -> " + map.getByValue("v2").keySet());map.put("k5", "v1");System.out.println("v1 -> " + map.getByValue("v1").keySet());System.out.println("v2 -> " + map.getByValue("v2").keySet());//替换所有的v1为v111System.out.println("before update : " + map);for(String k : map.getByValue("v1").keySet()) {map.put(k, "v111");}System.out.println("after update : " + map);}}

 

 参考更多大数据Hadoop、Spark、Hive相关：lxw的大数据田地

 

运行结果：

k1 -> v1
v1 -> [k3, k1]
v2 -> [k4, k2]
v2 -> [k5, k4, k2]
v1 -> [k5, k3, k1]
v2 -> [k4, k2]
before update : {k3=v1, k4=v2, k5=v1, k1=v1, k2=v2}
after update : {k3=v111, k4=v2, k5=v111, k1=v111, k2=v2}