27 lines
693 B
Go
27 lines
693 B
Go
package util
|
|
|
|
import (
|
|
"golang.org/x/exp/constraints"
|
|
"golang.org/x/exp/maps"
|
|
"golang.org/x/exp/slices"
|
|
)
|
|
|
|
// IterateMapOrdered iterates over the map with keys sorted in ascending order
|
|
// calling forEach for each key-value pair as long as forEach does not return an error.
|
|
func IterateMapOrdered[K constraints.Ordered, V any](m map[K]V, forEach func(k K, v V) error) error {
|
|
keys := OrderedMapKeys(m)
|
|
for _, k := range keys {
|
|
if err := forEach(k, m[k]); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// OrderedMapKeys returns the map keys in ascending order.
|
|
func OrderedMapKeys[K constraints.Ordered, V any](m map[K]V) []K {
|
|
keys := maps.Keys(m)
|
|
slices.Sort(keys)
|
|
return keys
|
|
}
|