element-ios/Riot/Categories/Sequence.swift

20 lines
523 B
Swift

//
// Copyright 2022-2024 New Vector Ltd.
//
// SPDX-License-Identifier: AGPL-3.0-only
// Please see LICENSE in the repository root for full details.
//
extension Sequence {
func group<GroupID: Hashable>(by keyPath: KeyPath<Element, GroupID>) -> [GroupID: [Element]] {
var result: [GroupID: [Element]] = .init()
for item in self {
let groupId = item[keyPath: keyPath]
result[groupId] = (result[groupId] ?? []) + [item]
}
return result
}
}