JDK源码之AbstractMap 类分析

一 概述 && Map接口

AbstractMap 是一个基础抽象实现类,实现了 Map接口 的主要方法.
Map接口定义了双列集合的规范 Map<K,V>,一个元素包含两值(一个key,一个value),
key和value的数据类型可以相同,也可以不同,key是不允许重复的,value是可以重复的

Map接口源码分析

定义一般方法

        int size();

        boolean isEmpty();

        boolean containsKey(Object key);

        boolean containsValue(Object value);

        V get(Object key); //可能返回null

        V put(K key, V value); //同一个key,多次put会覆盖

        V remove(Object key);

        void putAll(java.util.Map<? extends K, ? extends V> m);

        void clear();//清除map

        Set<K> keySet(); //返回key的set集合

        Collection<V> values(); // 返回value的Collection集合

        Set<java.util.Map.Entry<K, V>> entrySet(); //返回Entry内部接口的Set集合,内部接口看下面
        boolean equals(Object o);
        int hashCode();

内部接口

Map.Entry是map声明的一个内部接口,此接口为泛型,定义为Entry<K,V>.它表示map中的一个元素的实体

public interface Map<K, V> {
        interface Entry<K, V> {
            K getKey();
            V getValue();
            V setValue(V value);
            boolean equals(Object o);
            int hashCode();
            // 下面四个方法其实都是获取比较器,只不过需要传递的参数不一样,主要是为了StreamAPI和lamdba添加的排序器
            // 1.8新增静态方法,获取key的比较器
            public static <K extends Comparable<? super K>, V> Comparator<java.util.Map.Entry<K, V>> comparingByKey() {
                return (Comparator<java.util.Map.Entry<K, V>> & Serializable)
                        (c1, c2) -> c1.getKey().compareTo(c2.getKey());
            }
            //1.8新增静态方法,获取value的比较器
            public static <K, V extends Comparable<? super V>> Comparator<java.util.Map.Entry<K, V>> comparingByValue() {
                return (Comparator<java.util.Map.Entry<K, V>> & Serializable)
                        (c1, c2) -> c1.getValue().compareTo(c2.getValue());
            }
            //1.8新增静态方法,获取参数是Comparator的比较器,按key排序,这里可传入一个lamdba,自定义排序结果,按正序还是倒叙
            public static <K, V> Comparator<java.util.Map.Entry<K, V>> comparingByKey(Comparator<? super K> cmp) {
                Objects.requireNonNull(cmp);
                return (Comparator<java.util.Map.Entry<K, V>> & Serializable)
                        (c1, c2) -> cmp.compare(c1.getKey(), c2.getKey());
            }
            //1.8新增静态方法,获取参数是Comparator的比较器,按value排序,这里可传入一个lamdba,自定义排序结果,按正序还是倒叙
            public static <K, V> Comparator<java.util.Map.Entry<K, V>> comparingByValue(Comparator<? super V> cmp) {
                Objects.requireNonNull(cmp);
                return (Comparator<java.util.Map.Entry<K, V>> & Serializable)
                        (c1, c2) -> cmp.compare(c1.getValue(), c2.getValue());
            }
        }

//新添加的几个排序器方法使用demo:

        final Map<String, Integer> map = new HashMap<>();
        map.put("10",4);
        map.put("2",20);
        map.put("4",44);
        final List<Map.Entry<String, Integer>> collect = map.entrySet().stream()
                //.sorted(Map.Entry.comparingByKey()) // 使用比较器排序,直接Map.Entry获取,有默认的排序,也可以方法参数里面实现自定义排序规则
                .sorted(Map.Entry.comparingByValue((x, y) -> { // value
                    // 自定义倒序规则
                    if (x > y) return -1;
                    if (x < y) return 1;
                    return 0;
                }))
                .collect(Collectors.toList());

default方法

default方法都是jdk1.8新添加的方法,主要是功能补充的一些API

        // Default方法都是1.8之后新增的

        // 如果map中没有key对应的value,返回默认的value,负责返回对应的value值(包括null)
        default V getOrDefault(Object key, V defaultValue) {
            V v;
            return (((v = get(key)) != null) || containsKey(key))
                    ? v
                    : defaultValue;
        }

        //提供forEach遍历,参数为BiConsumer labdba表达式,自定义处理逻辑
        default void forEach(BiConsumer<? super K, ? super V> action) {
            Objects.requireNonNull(action);
            for (java.util.Map.Entry<K, V> entry : entrySet()) {
                K k;
                V v;
                try {
                    k = entry.getKey();
                    v = entry.getValue();
                } catch (IllegalStateException ise) {
                    // this usually means the entry is no longer in the map.
                    throw new ConcurrentModificationException(ise);
                }
                action.accept(k, v);
            }
        }

        // 对map中所有元素Entry的value进行修改操作
        default void replaceAll(BiFunction<? super K, ? super V, ? extends V> function) {
            Objects.requireNonNull(function);
            for (java.util.Map.Entry<K, V> entry : entrySet()) {
                K k;
                V v;
                try {
                    k = entry.getKey();
                    v = entry.getValue();
                } catch (IllegalStateException ise) { throw new ConcurrentModificationException(ise); }
                v = function.apply(k, v);
                try {
                    entry.setValue(v); //将lamdba返回的值设置给value
                } catch (IllegalStateException ise) { throw new ConcurrentModificationException(ise); }
            }
        }

        //如果key对应的value不存在或者为null就添加,否则返回value
        default V putIfAbsent(K key, V value) {
            V v = get(key);
            if (v == null) {
                v = put(key, value);
            }
            return v;
        }
        //根据 key和value 删除元素
        default boolean remove(Object key, Object value) {
            Object curValue = get(key);
            if (!Objects.equals(curValue, value) ||   // 传递的value不一样,不会删除
                    (curValue == null && !containsKey(key))) {
                return false;
            }
            remove(key);
            return true;
        }
        // 将key对应的值进行替换并返回结果,需要传递原value值
        default boolean replace(K key, V oldValue, V newValue) {
            Object curValue = get(key);
            if (!Objects.equals(curValue, oldValue) ||   //如果传递的old值和实际获取的值不一样,则不替换,即必须传真实的值
                    (curValue == null && !containsKey(key))) { //key不存在也不会新增
                return false;
            }
            put(key, newValue);
            return true;
        }

        //将key对应的value替换并返回结果
        default V replace(K key, V value) {
            V curValue;
            if (((curValue = get(key)) != null) || containsKey(key)) {
                curValue = put(key, value);
            }
            return curValue;
        }
        // 如果key对应v为null,则进行put,根据key,可以在lamdba设置想要的值,如果是null,则不添加
        default V computeIfAbsent(K key,Function<? super K, ? extends V> mappingFunction) {
            Objects.requireNonNull(mappingFunction);
            V v;
            if ((v = get(key)) == null) {
                V newValue;
                if ((newValue = mappingFunction.apply(key)) != null) {
                    put(key, newValue);
                    return newValue;
                }
            }
            return v;
        }
        // 如果key对应v为null,则进行put,根据key和value,可以在lamdba设置想要的值,如果是null,则不添加
        default V computeIfPresent(K key, BiFunction<? super K, ? super V, ? extends V> remappingFunction) {
            Objects.requireNonNull(remappingFunction);
            V oldValue;
            if ((oldValue = get(key)) != null) {
                V newValue = remappingFunction.apply(key, oldValue);
                if (newValue != null) {
                    put(key, newValue);
                    return newValue;
                } else {
                    remove(key);
                    return null;
                }
            } else {
                return null;
            }
        }
        //如果lamdba返回值不为null则直接put操作,否则remove,key这个元素
        default V compute(K key,BiFunction<? super K, ? super V, ? extends V> remappingFunction) {
            Objects.requireNonNull(remappingFunction);
            V oldValue = get(key);
            V newValue = remappingFunction.apply(key, oldValue);
            if (newValue == null) {
                if (oldValue != null || containsKey(key)) {
                    remove(key);  //lamdba返回null,且原key值存在,则直接remove
                    return null;
                } else {
                    return null;
                }
            } else {
                put(key, newValue);
                return newValue;
            }
        }

        //当key对应的value不存在时,直接使用传递的value,否则用lamdba自定义的value值进行put
        default V merge(K key, V value,BiFunction<? super V, ? super V, ? extends V> remappingFunction) {
            Objects.requireNonNull(remappingFunction);
            Objects.requireNonNull(value);
            V oldValue = get(key);
            //当key对应的value不存在时,直接使用传递的value,否则用lamdba自定义的value值
            V newValue = (oldValue == null) ? value :
                    remappingFunction.apply(oldValue, value);
            if (newValue == null) {
                remove(key);
            } else {
                put(key, newValue);
            }
            return newValue;
        }

静态方法

static方法都是jdk1.9新添加的方法,主要是新建map的一些方法

        //of方法都是新建map,jdk9开始新增的,从空map到最多10个k,v的map
        static <K, V> java.util.Map<K, V> of() {
            return ImmutableCollections.emptyMap();
        }
        static <K, V> java.util.Map<K, V> of(K k1, V v1) {
            return new ImmutableCollections.Map1<>(k1, v1);
        }
        // .........
        static <K, V> java.util.Map<K, V> of(K k1, V v1, K k2, V v2, K k3, V v3, K k4, V v4, K k5, V v5,
                                             K k6, V v6, K k7, V v7, K k8, V v8, K k9, V v9, K k10, V v10) {
            return new ImmutableCollections.MapN<>(k1, v1, k2, v2, k3, v3, k4, v4, k5, v5,
                    k6, v6, k7, v7, k8, v8, k9, v9, k10, v10);
        }

        //根据参数构造一个不可变map
        @SafeVarargs
        @SuppressWarnings("varargs")
        static <K, V> java.util.Map<K, V> ofEntries(java.util.Map.Entry<? extends K, ? extends V>... entries) {
            if (entries.length == 0) { // implicit null check of entries array
                return ImmutableCollections.emptyMap();
            } else if (entries.length == 1) {
                // implicit null check of the array slot
                return new ImmutableCollections.Map1<>(entries[0].getKey(),
                        entries[0].getValue());
            } else {
                Object[] kva = new Object[entries.length << 1];
                int a = 0;
                for (java.util.Map.Entry<? extends K, ? extends V> entry : entries) {
                    // implicit null checks of each array slot
                    kva[a++] = entry.getKey();
                    kva[a++] = entry.getValue();
                }
                return new ImmutableCollections.MapN<>(kva);
            }
        }

        //返回Map.Entry空的实例
        static <K, V> java.util.Map.Entry<K, V> entry(K k, V v) {
            // KeyValueHolder checks for nulls
            return new KeyValueHolder<>(k, v);
        }

        // jdk10 新增方法,根据map参数返回一个copy的不可变的map
        @SuppressWarnings({"rawtypes","unchecked"})
        static <K, V> java.util.Map<K, V> copyOf(java.util.Map<? extends K, ? extends V> map) {
            if (map instanceof ImmutableCollections.AbstractImmutableMap) {
                return (java.util.Map<K,V>)map;
            } else {
                return (java.util.Map<K,V>) java.util.Map.ofEntries(map.entrySet().toArray(new java.util.Map.Entry[0]));
            }
        }
    }

二 源码分析

三 总结

猜你喜欢

转载自www.cnblogs.com/houzheng/p/12687883.html