Kotlin groupby value as set instead of list

Question

I'm going to groupby Set<Pair<String,Int>> values to Map<String, Set<Int>> so simply. Find all matching ints for string and create set. I've got solution that works:

setStringInt.stream()
            .collect(
                groupingBy(
                    Projection::stringObj,
                    mapping(Projection::intObj,
                        toSet<Int>())
                )
            )

I've got kind of cleaner:

            .groupBy { it.stringobj }
            .mapValues { it.value.map { it.intobj }.toSet() }

But it's looking quite dirty. Do you have any idea how to simplify this? Am I able to do this without using stream?

Answer 1

我认为这更具可读性：

setStringInt.groupBy({ it.first }, { it.second }).mapValues { it.value.toSet() }

使用

inline fun <T, K, V> Iterable<T>.groupBy(
    keySelector: (T) -> K,
    valueTransform: (T) -> V
): Map<K, List<V>>

超载。

Answer 2

You can just use the Kotlin functions groupBy and then mapValues directly on the Set.

您第一次使用groupBy上it.first获得Map<String, List<Pair<String, Int>>。然后，您需要映射结果映射的值，以便List<Int>从获得List<Pair<String, Int>>。像这样：

setStringInt.groupBy { it.first }
            .mapValues { entry -> entry.value.map { it.second }.toSet() }