Java 8 Map of Collections 从集合中删除元素，如果为空则删除条目

Question

我有一张地图，其中的值是一个集合。给定一个键，我想删除集合的一个元素并返回它，但如果集合为空，我也想删除该条目。有没有办法使用 Java 8 的众多新 Map 方法之一在短时间内完成此操作？

一个简单的例子（我使用堆栈，但它可以是列表、集合等）。举例来说，假设已经检查了映射是否包含键。

public static String removeOne(Map<Integer, Stack<String>> map, int key) {
    Stack<String> stack = map.get(key);
    String result = stack.pop();
    if(stack.isEmpty()){
        map.remove(key);
    }
    return result;
}

我尝试做类似的事情

map.compute(1, (k, v) -> {v.pop(); return v.size() == 0 ? null : v;});

但即使它确实删除了为空的条目，我也不知道如何获取pop() 返回的值。

Answer 1

/* quite ugly
String rv = Optional.ofNullable(map.get(1)).map(stack -> {
            if (!stack.isEmpty()) {
                String v = stack.pop();
                if (stack.isEmpty()) {
                    map.remove(1);
                }
                return v;
            }
            return null;
        }).orElse(null);
*/ 

@Test
public void test() {
    {
        Map<Integer, Stack<String>> map = new HashMap<>();
        Stack<String> s = new Stack<String>();
        s.addAll(Arrays.asList("a", "b"));
        map.put(1, s);
        String rv = Optional.ofNullable(map.get(1)).map(stack -> {
            if (!stack.isEmpty()) {
                String v = stack.pop();
                if (stack.isEmpty()) {
                    map.remove(1);
                }
                return v;
            }
            return null;
        }).orElse(null);
        Assert.assertEquals("b", rv);
        Assert.assertEquals(1, map.get(1).size());
        Assert.assertEquals("a", map.get(1).iterator().next());
    }
    {
        Map<Integer, Stack<String>> map = new HashMap<>();
        Stack<String> s = new Stack<String>();
        s.add("a");
        map.put(1, s);
        String rv = Optional.ofNullable(map.get(1)).map(stack -> {
            if (!stack.isEmpty()) {
                String v = stack.pop();
                if (stack.isEmpty()) {
                    map.remove(1);
                }
                return v;
            }
            return null;
        }).orElse(null);
        Assert.assertEquals("a", rv);
        Assert.assertNull(map.get(1));
    }
}

Answer 2

嗯，它甚至比你已有的更难看，但我想有一种方法：

public static String removeOne(Map<Integer, Stack<String>> map, int key) {
    String[] removed = new String[1];
    map.compute(key, (k, v) -> {
        removed[0] = v.pop();
        return v.size() == 0 ? null : v;
    });
    return removed[0];
}

问题是merge/compute 等返回值，在您的情况下是Stack/Set/List，而不是该集合中的单个元素。

Answer 3

有没有一种方法可以使用众多新方法中的一种在短时间内完成此操作 Java 8 的映射方法？

从 JDK8 开始，没有任何新方法可以改进您的代码，无论是在可读性还是效率方面。

如果您将其作为自己的练习，那么我可以在一定程度上理解您为什么要缩短代码（如果可能），但是在生产代码打高尔夫球时应该避免，而是采用最易读和可维护的方法；再长也没关系。

你的方法很好。

Answer 4

我完全同意@NicholasK。这里没有理由使用任何流或 lambda。

你的方法很不错。我唯一想补充的是使其通用：

public static <K, E, C extends Collection<E>> E removeOne(Map<K, C> map, K key) {
    C col = map.get(key);
    Iterator<E> it = col.iterator();
    E e = it.next();
    it.remove();
    if (!it.hasNext()) {
        map.remove(key);
    }
    return e;
}

此方法将适用于任何返回有效迭代器的集合（映射值）。

Answer 5

Guava's Multimap 为您处理 remove-collection-if-empty 逻辑。您可以在两行中获得与您的方法等效的行为：

public static String removeOne(ListMultimap<Integer, String> map, int key) {
    List<String> stack = map.get(key);
    return stack.remove(stack.size() - 1);
}

如果地图没有给定键的条目，您现有的解决方案和上述解决方案都会抛出异常。您可以选择更改代码来处理此问题：

public static String removeOne(ListMultimap<Integer, String> map, int key) {
    List<String> stack = map.get(key);
    if (stack.isEmpty()) {
        return null;
    }
    return stack.remove(stack.size() - 1);
}

当然你也可以把它变成通用的：

public static <K, V> V removeOne(ListMultimap<K, V> map, K key) {
    List<V> stack = map.get(key);
    if (stack.isEmpty()) {
        return null;
    }
    return stack.remove(stack.size() - 1);
}

Answer 6

或者您可以使用size 将其重写为：

public static String removeOne(Map<Integer, Stack<String>> map, int key) {
    return map.get(key).size() == 1 ? map.remove(key).pop() : map.get(key).pop();
}