05aeeab581
* snapshotter ignores nodes not along a path along those derived from a list of account addresses if one is provided * config and env updates * cmd update * Encode watched address path bytes to hex for comparison * actually ignore the subtries that are not along the paths of interest * Fixes for account selective snapshot * Use non-concurrent iterator when having a single worker * Only index root node when starting path of an iterator is nil * Upgrade deps * Avoid tracking iterators and skip recovery test * Fix recovery mechanism, use sync Map instead of buffered channels * Add test for account selective snapshot * Continue traversal with concurrent iterators with starting path nil * Use errgroup to simplify error handling with concurrent iterators * Check if all the nodes are indexed in the recovery test * Use concurrency safe sync Map in account selective snapshot test * Only track concurrent iterators and refactor code * Fix node and recovered path comparison * Revert back to using buffered channels for tracking iterators * Add a metric to monitor number of active iterators * Update docs * Update seeked path after node is processed * Return error on context cancellation from subtrie iteration * Add tests for account selective snapshot recovery * Explicity enforce concurrent iterator bounds to avoid duplicate nodes * Update full snapshot test to check nodes being indexed * Refactor code to simplify snapshot logic * Remove unnecessary function argument * Use ctx cancellation for handling signals * Add descriptive comments Co-authored-by: prathamesh0 <prathamesh.musale0@gmail.com>
114 lines
2.8 KiB
Go
114 lines
2.8 KiB
Go
// VulcanizeDB
|
|
// Copyright © 2022 Vulcanize
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// This program 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 Affero General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package prom
|
|
|
|
import (
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/promauto"
|
|
)
|
|
|
|
const (
|
|
namespace = "ipld_eth_state_snapshot"
|
|
|
|
connSubsystem = "connections"
|
|
statsSubsystem = "stats"
|
|
)
|
|
|
|
var (
|
|
metrics bool
|
|
|
|
stateNodeCount prometheus.Counter
|
|
storageNodeCount prometheus.Counter
|
|
codeNodeCount prometheus.Counter
|
|
|
|
activeIteratorCount prometheus.Gauge
|
|
)
|
|
|
|
func Init() {
|
|
metrics = true
|
|
|
|
stateNodeCount = promauto.NewCounter(prometheus.CounterOpts{
|
|
Namespace: namespace,
|
|
Subsystem: statsSubsystem,
|
|
Name: "state_node_count",
|
|
Help: "Number of state nodes processed",
|
|
})
|
|
|
|
storageNodeCount = promauto.NewCounter(prometheus.CounterOpts{
|
|
Namespace: namespace,
|
|
Subsystem: statsSubsystem,
|
|
Name: "storage_node_count",
|
|
Help: "Number of storage nodes processed",
|
|
})
|
|
|
|
codeNodeCount = promauto.NewCounter(prometheus.CounterOpts{
|
|
Namespace: namespace,
|
|
Subsystem: statsSubsystem,
|
|
Name: "code_node_count",
|
|
Help: "Number of code nodes processed",
|
|
})
|
|
|
|
activeIteratorCount = promauto.NewGauge(prometheus.GaugeOpts{
|
|
Namespace: namespace,
|
|
Subsystem: statsSubsystem,
|
|
Name: "active_iterator_count",
|
|
Help: "Number of active iterators",
|
|
})
|
|
}
|
|
|
|
// RegisterDBCollector create metric collector for given connection
|
|
func RegisterDBCollector(name string, db DBStatsGetter) {
|
|
if metrics {
|
|
prometheus.Register(NewDBStatsCollector(name, db))
|
|
}
|
|
}
|
|
|
|
// IncStateNodeCount increments the number of state nodes processed
|
|
func IncStateNodeCount() {
|
|
if metrics {
|
|
stateNodeCount.Inc()
|
|
}
|
|
}
|
|
|
|
// IncStorageNodeCount increments the number of storage nodes processed
|
|
func IncStorageNodeCount() {
|
|
if metrics {
|
|
storageNodeCount.Inc()
|
|
}
|
|
}
|
|
|
|
// IncCodeNodeCount increments the number of code nodes processed
|
|
func IncCodeNodeCount() {
|
|
if metrics {
|
|
codeNodeCount.Inc()
|
|
}
|
|
}
|
|
|
|
// IncActiveIterCount increments the number of active iterators
|
|
func IncActiveIterCount() {
|
|
if metrics {
|
|
activeIteratorCount.Inc()
|
|
}
|
|
}
|
|
|
|
// DecActiveIterCount decrements the number of active iterators
|
|
func DecActiveIterCount() {
|
|
if metrics {
|
|
activeIteratorCount.Dec()
|
|
}
|
|
}
|