java:基于guava实现Map的Key类型替换

Posted 2022-12-15 10km

google 的 guava是功能很丰富的基础工具库，它的com.google.common.collect.Maps工具类提供了很多对Map方便工具方法

参见 Javdoc https://guava.dev/releases/snapshot/api/docs/com/google/common/collect/Maps.html

Maps提供了一系列transform方法可以实现对Value的转换
比如 public static <K, V1, V2> Map<K, V2> transformValues(Map<K, V1> fromMap, Function<? super V1, V2> function)可以通过一个Function将Map的所有Value转为另一个类型V2,即将Map<K,V1>转为Map<K,V2>

但是Maps没有提供Key类型的转换方法。不过知道了原理，自己实现也不复杂。如下是我基于Maps.uniqueIndex和Maps.transformEntries方法实现的Key类型转换工具方法，实现将Map<K1,V>转为Map<K2,V>

public class TransUtils
	/** 
     * convert @code Map<K1,V> to @code Map<K2,V>   
     * @return @linkplain ImmutableMap
     */
    public static final <K1,K2,V>Map<K2,V> transform(Map<K1,V>fromMap,final Function<K1,K2>transformer)
        checkNotNull(fromMap,"fromMap is null");
        checkNotNull(transformer,"transformer is null");
        // 新的Map对象Key类型已经是K2了，只是Value类型成了Entry
        ImmutableMap<K2, Entry<K1, V>> k2Entry = Maps.uniqueIndex(fromMap.entrySet(), new Function<Entry<K1, V>,K2>()
            @Override
            public K2 apply(Entry<K1, V> input)                
                return transformer.apply(input.getKey());
            );
        // 再做一次转换将Entry<K1, V>转换成V,这个过程并没有创建新的Map,只是创建了k2Entry的代理对象
        Map<K2, V> k2V = Maps.transformEntries(k2Entry, new EntryTransformer<K2,Entry<K1,V>,V>()
            @Override
            public V transformEntry(K2 key, Entry<K1, V> value) 
                return value.getValue();
            );
        return k2V;
    

调用示例：

public enum MQParam 
ONE,TOW

/** Key为枚举类型的Map(MQParam 为枚举类型) */
Map<MQParam,String> map = ImmutableMap.of(MQParam.ONE,"hello",MQParam.TWO,"world");
/** 将map的Key由枚举类型MQParam转为String */
Map<String,String> m2 = TransUtils.transform(map, MQParam::name);

参考资料

Maps.uniqueIndex(Iterable, Function)
Maps.transformEntries​(Map, Maps.EntryTransformer)