From c866f62262994c30eeeaafb24cd318556f70fa13 Mon Sep 17 00:00:00 2001 From: yzang2019 Date: Thu, 23 Jan 2025 13:36:06 +0800 Subject: [PATCH 1/2] Add num of keys for state dump --- tools/cmd/seidb/operations/state_size.go | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tools/cmd/seidb/operations/state_size.go b/tools/cmd/seidb/operations/state_size.go index 2afe46b9..f6920e22 100644 --- a/tools/cmd/seidb/operations/state_size.go +++ b/tools/cmd/seidb/operations/state_size.go @@ -69,6 +69,7 @@ func PrintStateSize(module string, db *memiavl.DB) error { fmt.Printf("Calculating for module: %s \n", moduleName) keySizeByPrefix := map[string]int64{} valueSizeByPrefix := map[string]int64{} + numKeysByPrefix := map[string]int64{} tree.ScanPostOrder(func(node memiavl.Node) bool { if node.IsLeaf() { totalNumKeys++ @@ -81,6 +82,7 @@ func PrintStateSize(module string, db *memiavl.DB) error { prefix = prefix[:2] keySizeByPrefix[prefix] += int64(keySize) valueSizeByPrefix[prefix] += int64(valueSize) + numKeysByPrefix[prefix] += 1 } return true }) @@ -89,6 +91,8 @@ func PrintStateSize(module string, db *memiavl.DB) error { fmt.Printf("Module %s prefix key size breakdown (bytes): %s \n", moduleName, prefixKeyResult) prefixValueResult, _ := json.MarshalIndent(valueSizeByPrefix, "", " ") fmt.Printf("Module %s prefix value size breakdown (bytes): %s \n", moduleName, prefixValueResult) + numKeysResult, _ := json.MarshalIndent(numKeysByPrefix, "", " ") + fmt.Printf("Module %s prefix num of keys breakdown: %s \n", moduleName, numKeysResult) // Print top 20 contracts by total size numToShow := 20 From 7930272bc9735d75265b10b79a0b2efffc62013b Mon Sep 17 00:00:00 2001 From: yzang2019 Date: Fri, 24 Jan 2025 10:27:27 +0800 Subject: [PATCH 2/2] Fix lint --- tools/cmd/seidb/operations/state_size.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/cmd/seidb/operations/state_size.go b/tools/cmd/seidb/operations/state_size.go index f6920e22..780fd3e6 100644 --- a/tools/cmd/seidb/operations/state_size.go +++ b/tools/cmd/seidb/operations/state_size.go @@ -82,7 +82,7 @@ func PrintStateSize(module string, db *memiavl.DB) error { prefix = prefix[:2] keySizeByPrefix[prefix] += int64(keySize) valueSizeByPrefix[prefix] += int64(valueSize) - numKeysByPrefix[prefix] += 1 + numKeysByPrefix[prefix]++ } return true })