From 43de646806e4179701024fd54f632db6ab78d4ce Mon Sep 17 00:00:00 2001 From: cuiweixie Date: Tue, 26 Aug 2025 18:29:45 +0800 Subject: [PATCH] Refactor to use maps.Keys --- dive/filetree/order_strategy.go | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/dive/filetree/order_strategy.go b/dive/filetree/order_strategy.go index 1838dd8d..610ec335 100644 --- a/dive/filetree/order_strategy.go +++ b/dive/filetree/order_strategy.go @@ -1,6 +1,8 @@ package filetree import ( + "maps" + "slices" "sort" ) @@ -30,10 +32,7 @@ func GetSortOrderStrategy(sortOrder SortOrder) OrderStrategy { type orderByNameStrategy struct{} func (orderByNameStrategy) orderKeys(files map[string]*FileNode) []string { - var keys []string - for key := range files { - keys = append(keys, key) - } + keys := slices.Collect(maps.Keys(files)) sort.Strings(keys) @@ -43,10 +42,7 @@ func (orderByNameStrategy) orderKeys(files map[string]*FileNode) []string { type orderBySizeDescStrategy struct{} func (orderBySizeDescStrategy) orderKeys(files map[string]*FileNode) []string { - var keys []string - for key := range files { - keys = append(keys, key) - } + keys := slices.Collect(maps.Keys(files)) sort.Slice(keys, func(i, j int) bool { ki, kj := keys[i], keys[j]