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

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.uniqueIndexMaps.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);

补充

上面 transform方法返回的是不可修改的Map实例,是原Map实例的一个视图(View),原Map实例如果在其他地方被修改了,会同步影响返回的这个Map实例。
如果要返回可修改的Map实例就要在最后返回时将类型转为HashMap。
即: Maps.newHashMap(k2V)。这样返回的Map实例就与原Map实例无关了。
完整代码如下:

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 Maps.newHashMap(k2V);
    }
}

参考资料

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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

10km

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值