mirror of
https://github.com/ethereum/go-ethereum.git
synced 2026-02-28 00:27:26 +00:00
e9656238a7
Co-authored-by: shantichanal <158101918+shantichanal@users.noreply.github.com> Co-authored-by: Gary Rong <garyrong0905@gmail.com> Co-authored-by: Guillaume Ballet <3272758+gballet@users.noreply.github.com>
108 lines
3.3 KiB
Go
108 lines
3.3 KiB
Go
// Copyright 2025 The go-ethereum Authors
|
|
// This file is part of the go-ethereum library.
|
|
//
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package stateless
|
|
|
|
import (
|
|
"maps"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/ethereum/go-ethereum/metrics"
|
|
)
|
|
|
|
var (
|
|
accountTrieDepthAvg = metrics.NewRegisteredGauge("witness/trie/account/depth/avg", nil)
|
|
accountTrieDepthMin = metrics.NewRegisteredGauge("witness/trie/account/depth/min", nil)
|
|
accountTrieDepthMax = metrics.NewRegisteredGauge("witness/trie/account/depth/max", nil)
|
|
|
|
storageTrieDepthAvg = metrics.NewRegisteredGauge("witness/trie/storage/depth/avg", nil)
|
|
storageTrieDepthMin = metrics.NewRegisteredGauge("witness/trie/storage/depth/min", nil)
|
|
storageTrieDepthMax = metrics.NewRegisteredGauge("witness/trie/storage/depth/max", nil)
|
|
)
|
|
|
|
// depthStats tracks min/avg/max statistics for trie access depths.
|
|
type depthStats struct {
|
|
totalDepth int64
|
|
samples int64
|
|
minDepth int64
|
|
maxDepth int64
|
|
}
|
|
|
|
// newDepthStats creates a new depthStats with default values.
|
|
func newDepthStats() *depthStats {
|
|
return &depthStats{minDepth: -1}
|
|
}
|
|
|
|
// add records a new depth sample.
|
|
func (d *depthStats) add(n int64) {
|
|
if n < 0 {
|
|
return
|
|
}
|
|
d.totalDepth += n
|
|
d.samples++
|
|
|
|
if d.minDepth == -1 || n < d.minDepth {
|
|
d.minDepth = n
|
|
}
|
|
if n > d.maxDepth {
|
|
d.maxDepth = n
|
|
}
|
|
}
|
|
|
|
// report uploads the collected statistics into the provided gauges.
|
|
func (d *depthStats) report(maxGauge, minGauge, avgGauge *metrics.Gauge) {
|
|
if d.samples == 0 {
|
|
return
|
|
}
|
|
maxGauge.Update(d.maxDepth)
|
|
minGauge.Update(d.minDepth)
|
|
avgGauge.Update(d.totalDepth / d.samples)
|
|
}
|
|
|
|
// WitnessStats aggregates statistics for account and storage trie accesses.
|
|
type WitnessStats struct {
|
|
accountTrie *depthStats
|
|
storageTrie *depthStats
|
|
}
|
|
|
|
// NewWitnessStats creates a new WitnessStats collector.
|
|
func NewWitnessStats() *WitnessStats {
|
|
return &WitnessStats{
|
|
accountTrie: newDepthStats(),
|
|
storageTrie: newDepthStats(),
|
|
}
|
|
}
|
|
|
|
// Add records trie access depths from the given node paths.
|
|
// If `owner` is the zero hash, accesses are attributed to the account trie;
|
|
// otherwise, they are attributed to the storage trie of that account.
|
|
func (s *WitnessStats) Add(nodes map[string][]byte, owner common.Hash) {
|
|
if owner == (common.Hash{}) {
|
|
for path := range maps.Keys(nodes) {
|
|
s.accountTrie.add(int64(len(path)))
|
|
}
|
|
} else {
|
|
for path := range maps.Keys(nodes) {
|
|
s.storageTrie.add(int64(len(path)))
|
|
}
|
|
}
|
|
}
|
|
|
|
// ReportMetrics reports the collected statistics to the global metrics registry.
|
|
func (s *WitnessStats) ReportMetrics() {
|
|
s.accountTrie.report(accountTrieDepthMax, accountTrieDepthMin, accountTrieDepthAvg)
|
|
s.storageTrie.report(storageTrieDepthMax, storageTrieDepthMin, storageTrieDepthAvg)
|
|
}
|