refactor: share collapse field validation lists (task 387)

1 files changed, 3 insertions, 12 deletions

diff --git a/internal/flamegraph/livetrie.go b/internal/flamegraph/livetrie.go
index 9f1fd91..04137bf 100644
--- a/internal/flamegraph/livetrie.go
+++ b/internal/flamegraph/livetrie.go
@@ -9,6 +9,7 @@ import (
 	"sync"
 	"sync/atomic"
 
+	"ior/internal/collapse"
 	"ior/internal/event"
 )
 
@@ -261,21 +262,11 @@ func normalizeLiveTrieFields(fields []string) ([]string, error) {
 }
 
 func isLiveTrieField(field string) bool {
-	switch field {
-	case "path", "comm", "tracepoint", "pid", "tid", "flags":
-		return true
-	default:
-		return false
-	}
+	return collapse.IsValidField(field)
 }
 
 func isLiveTrieCountField(field string) bool {
-	switch field {
-	case "count", "duration", "durationToPrev", "bytes":
-		return true
-	default:
-		return false
-	}
+	return collapse.IsValidCountField(field)
 }
 
 func subtreeTotal(node *trieNode) uint64 {