java - 将映射转换为 java 8 流中更扁平的结构

Question

我有以下代码:

private Map<String, Object> flatten(Map<String, Object> source) {
    Map<String, Object> result = new LinkedHashMap<>();

    source.forEach((k, v) -> {
        if (v instanceof Map) {
            Map<String, Object> subMap = flatten((Map<String, Object>) v);
            for (String subkey : subMap.keySet()) {
                result.put(k + "." + subkey, subMap.get(subkey));
            }
        } else result.put(k, v);
    });

    return result;
}

上面的代码压平给定的 map

Ex: 
{
  "hello": {
    "hi": {
      "hola": 1
    }
  }
}

to 
{
  "hello.hi.hola": 1
}

我想使用 Java8 流并实现相同的逻辑，我该怎么做？

Answer 1

我本来打算发布类似于 @John Bollinger's answer 的内容。相反，我将使用 StreamEx 提供一个更干净的版本。 :

static Map<String, Object> flatten(Map<?, ?> map) {
    return flatten("", map)
            .mapKeys(k -> k.substring(1))
            .toMap();
}

static EntryStream<String, Object> flatten(String key, Object value) {
    if (value instanceof Map) {
        return EntryStream.of((Map<?, ?>)value)
                .flatMapKeyValue((k, v) -> flatten(key + "." + k, v))
                .chain(EntryStream::of);
    } else {
        return EntryStream.of(key, value);
    }
}