Java 8中递归展平嵌套映射的值

Question

给定一个 Map<String, Object>，其中的值要么是一个 String，要么是另一个 Map<String, Object>，如何使用 Java 8 将这些 Map 打平为单个值列表？

示例：

Map - "key1" -> "value1"
    - "key2" -> "value2"
    - "key3" -> Map - "key3.1" -> "value3.1"
                    - "key3.2" -> "value3.2"
                    - "key3.3" -> Map - "key3.3.1" -> "value3.3.1"
                                      - "key3.3.2" -> "value3.3.2" 

针对上述示例，我希望得到以下列表：

value1
value2
value3.1
value3.2
value3.3.1
value3.3.2

我知道可以这样做：

public static void main(String args[]) throws Exception {
    //Map with nested maps with nested maps with nested maps with nested......
    Map<String, Object> map = getSomeMapWithNestedMaps();

    List<Object> values = new ArrayList<>();
    addToList(map, values);

    for (Object o:values) {
        System.out.println(o);
    }
}

static void addToList(Map<String, Object>map, List<Object> list) {
    for (Object o:map.values()) {
        if (o instanceof Map) {
            addToList((Map<String, Object>)o, list);
        } else {
            list.add(o);
        }
    }
}

如何使用 Stream 实现这个功能？

编辑：

经过一些尝试，我找到了答案：

public static void main(String args[]) throws Exception {
    //Map with nested maps with nested maps with nested maps with nested......
    Map<String, Object> map = getSomeMapWithNestedMaps();
    //Recursively flatten maps and print out all values
    List<Object> list= flatten(map.values().stream()).collect(Collectors.toList());
}

static Stream<Object> flatten(Stream<Object> stream) {
    return stream.flatMap((o) ->
        (o instanceof Map) ? flatten(((Map<String, Object>)o).values().stream()) : Stream.of(o)
    );
}

- Ian2thedv

---

https://dev59.com/YmEi5IYBdhLWcg3wG45o 建议使用方法引用。 - jaco0646

类似的问题，但只涉及键：https://stackoverflow.com/a/62470056/435605 - AlikElzin-kilaka

Answer 1

你可以定义一个递归方法，用于展开一个map，并将其作为Stream#flatMap函数的参数，或者直接调用该方法。

示例：

public class FlatMap {

    public static Stream<Object> flatten(Object o) {
        if (o instanceof Map<?, ?>) {
            return ((Map<?, ?>) o).values().stream().flatMap(FlatMap::flatten);
        }
        return Stream.of(o);
    }

    public static void main(String[] args) {
        Map<String, Object> map0 = new TreeMap<>();
        map0.put("key1", "value1");
        map0.put("key2", "value2");
        Map<String, Object> map1 = new TreeMap<>();
        map0.put("key3", map1);
        map1.put("key3.1", "value3.1");
        map1.put("key3.2", "value3.2");
        Map<String, Object> map2 = new TreeMap<>();
        map1.put("key3.3", map2);
        map2.put("key3.3.1", "value3.3.1");
        map2.put("key3.3.2", "value3.3.2");

        List<Object> collect = map0.values().stream()
                                            .flatMap(FlatMap::flatten)
                                            .collect(Collectors.toList());
        // or
        List<Object> collect2 = flatten(map0).collect(Collectors.toList());
        System.out.println(collect); 
    }
}

对于给定的嵌套映射，它会打印出：

[value1, value2, value3.1, value3.2, value3.3.1, value3.3.2]

- Modus Tollens

---

不确定结果如何被认为是扁平化的。结果：[value1，value2，value3.1，value3.2，value3.3.1，value3.3.2] - AlikElzin-kilaka

是的，那就是答案中给出的结果。嵌套映射的值被合并成一个流。 - Modus Tollens