[Kotlin] associateBy, groupBy

collections API 중 하나로 컬렉션을 Map 형태로 변형시켜주는 API 이다. 

associateBy 함수와 groupBy 함수는 지정된 key에 의해 인덱싱된 컬렉션의 원소로부터 Map을 만들어준다.

associateBy

public inline fun <T, K> Iterable<T>.associateBy(keySelector: (T) -> K): Map<K, T> {
    val capacity = mapCapacity(collectionSizeOrDefault(10)).coerceAtLeast(16)
    return associateByTo(LinkedHashMap<K, T>(capacity), keySelector)
}

매개변수에는 T를 전달하고 key를 리턴하는 람다함수 keySelector를 인자로 받는다. 

매개변수로 받은 keySelector를 key로, List의 인자를 value로 Map<K,V>를 만들어 리턴한다. 

 

예시

fun main() { 
	val statuses = listOf(Status("Running"), Status("Pending"), Status("Failed"))
	private val mapping = statuses.associateBy(Phase::value)
	// 결과 : {Running=Status("Running"), Pending=Status("Pending"), Failed=Status("Failed")}
}

 

groupBy

public inline fun <T, K> Array<out T>.groupBy(keySelector: (T) -> K): Map<K, List<T>> {
    return groupByTo(LinkedHashMap<K, MutableList<T>>(), keySelector)
}

List를 Map 형태로 변형시켜주는 함수이다. 

associateBy와 달리, key가 중복되어도 value를 리스트로 만들어 주기 때문에 값 유실이 없다. (associateBy는 key 중복시, 마지막에 있는 원소가 value가 된다.)

 

예시

fun main() { 
	val statuses = listOf(Status("Running"), Status("Pending"), Status("Failed"))
	private val mapping = statuses.groupBy(Phase::value)
	// 결과 : {Running=[Status("Running")], Pending=[Status("Pending")], Failed=[Status("Failed")]}
}

value가 list로 묶여있는 것을 볼 수 있다. 

매개변수로 2개를 넣는다면 각각은 key와 value이다.