Java Map 通过 key 或者 value 过滤

摘要：今天写根过滤的时候一下子有点愣眼，先是想到用原生的循环查出来，但是觉得太，后面思考了一下可以用的，写完了，又发现有现成的方法，这里一一列出来，供参考使用。

今天写根过滤的时候一下子有点愣眼，先是想到用 Java 原生的 map 循环查出来，但是觉得太 low， 后面思考了一下可以用 Java8 的 Lambda，写完了，又发现 Google Guava 有现成的方法，这里一一列出来，供参考使用。

首先提示，如果照搬我的代码的话别忘了引这些依赖

        
            junit
            junit
            4.12
            test
            
                
                    org.hamcrest
                    hamcrest-core
                
            
        

        
            org.hamcrest
            hamcrest-library
            1.3
            test
        

        
            com.google.guava
            guava
            25.1-jre
        

public class FilterMapByKeyTest {
    private Map WEEK = new HashMap<>();

    @Before
    public void setUp () {
        WEEK.put(1, "Monday");
        WEEK.put(2, "Tuesday");
        WEEK.put(3, "Wednesday");
        WEEK.put(4, "Thursday");
        WEEK.put(5, "Friday");
        WEEK.put(6, "Saturday");
        WEEK.put(7, "Sunday");
    }

    /**
     * Java 8之前的版本
     */
    @Test
    public void filterMapByKey () {
        Map map = new HashMap<>();
        for (Map.Entry entry : WEEK.entrySet()) {
            if (entry.getKey() <= 3) {
                map.put(entry.getKey(), entry.getValue());
            }
        }
        assertThat(map.keySet(), contains(1, 2, 3));
    }

    /**
     * Java 8 Lambda
     */
    @Test
    public void filterMapByKeyJava8Lambda () {
        Map map = WEEK.entrySet().stream().filter(r -> r.getKey() <= 3)
                        .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
        assertThat(map.keySet(), contains(1, 2, 3));
    }

    /**
     * Google Guava
     */
    @Test
    public void filterMapByKeyGuava () {
        Map map = Maps.filterKeys(WEEK, r -> r <= 3);
        assertThat(map.keySet(), contains(1, 2, 3));
    }

}

public class FilterMapByValueTest {
    private Map WEEK = new HashMap<>();

    @Before
    public void setUp () {
        WEEK.put(1, "Monday");
        WEEK.put(2, "Tuesday");
        WEEK.put(3, "Wednesday");
        WEEK.put(4, "Thursday");
        WEEK.put(5, "Friday");
        WEEK.put(6, "Saturday");
        WEEK.put(7, "Sunday");
    }

    /**
     * Java 8之前的版本
     */
    @Test
    public void filterMapByValue () {
        Map map = new HashMap<>();
        for (Map.Entry entry : WEEK.entrySet()) {
            if (entry.getValue().startsWith("S")) {
                map.put(entry.getKey(), entry.getValue());
            }
        }
        assertThat(map.values(), contains("Saturday","Sunday"));
    }

    /**
     * Java 8 Lambda
     */
    @Test
    public void filterMapByValueJava8Lambda () {
        Map map = WEEK.entrySet().stream().filter(r -> r.getValue().startsWith("S"))
                .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
        assertThat(map.values(), contains("Saturday","Sunday"));
    }

    /**
     * Google Guava
     */
    @Test
    public void filterMapByValueGuava () {
        Map map = Maps.filterValues(WEEK, r -> r.startsWith("S"));
        assertThat(map.values(), contains("Saturday","Sunday"));
    }
}

如果觉得内容还不错，可以关注一下我哦
微信公众号：志哥 （ID: zhige-me）
期待与你相遇，一同成长前行！