45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
/*
|
|
* Copyright 2023 The Dragonfly Authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package slice
|
|
|
|
func KeyBy[K comparable, V any](collection []V, iteratee func(item V) K) map[K]V {
|
|
result := make(map[K]V, len(collection))
|
|
|
|
for i := range collection {
|
|
k := iteratee(collection[i])
|
|
result[k] = collection[i]
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
func Values[K comparable, V any](in ...map[K]V) []V {
|
|
size := 0
|
|
for i := range in {
|
|
size += len(in[i])
|
|
}
|
|
result := make([]V, 0, size)
|
|
|
|
for i := range in {
|
|
for k := range in[i] {
|
|
result = append(result, in[i][k])
|
|
}
|
|
}
|
|
|
|
return result
|
|
}
|