Remove optional gatherer

api/metrics/multi_gatherer_test.go

@@ -4,13 +4,17 @@
 package metrics
 
 import (
+	"errors"
 	"testing"
 
+	"github.com/prometheus/client_golang/prometheus"
 	"github.com/stretchr/testify/require"
 
 	dto "github.com/prometheus/client_model/go"
 )
 
+var errTest = errors.New("non-nil error")
+
 func TestMultiGathererEmptyGather(t *testing.T) {
 	require := require.New(t)
 
@@ -25,7 +29,7 @@ func TestMultiGathererDuplicatedPrefix(t *testing.T) {
 	require := require.New(t)
 
 	g := NewMultiGatherer()
-	og := NewOptionalGatherer()
+	og := prometheus.NewRegistry()
 
 	require.NoError(g.Register("", og))
 

chains/linearizable_vm.go

@@ -29,7 +29,7 @@ type initializeOnLinearizeVM struct {
 	vmToInitialize common.VM
 	vmToLinearize  *linearizeOnInitializeVM
 
-	registerer   metrics.OptionalGatherer
+	registerer   metrics.MultiGatherer
 	ctx          *snow.Context
 	db           database.Database
 	genesisBytes []byte

chains/manager.go

@@ -434,7 +434,7 @@ func (m *manager) buildChain(chainParams ChainParameters, sb subnets.Subnet) (*c
 		return nil, fmt.Errorf("error while registering DAG metrics %w", err)
 	}
 
-	vmMetrics := metrics.NewOptionalGatherer()
+	vmMetrics := metrics.NewMultiGatherer()
 	vmNamespace := metric.AppendNamespace(chainNamespace, "vm")
 	if err := m.Metrics.Register(vmNamespace, vmMetrics); err != nil {
 		return nil, fmt.Errorf("error while registering vm's metrics %w", err)
@@ -642,17 +642,12 @@ func (m *manager) createAvalancheChain(
 		},
 	)
 
-	avalancheRegisterer := metrics.NewOptionalGatherer()
-	snowmanRegisterer := metrics.NewOptionalGatherer()
-
-	registerer := metrics.NewMultiGatherer()
-	if err := registerer.Register("avalanche", avalancheRegisterer); err != nil {
-		return nil, err
-	}
-	if err := registerer.Register("", snowmanRegisterer); err != nil {
+	avalancheRegisterer := metrics.NewMultiGatherer()
+	snowmanRegisterer := metrics.NewMultiGatherer()
+	if err := ctx.Context.Metrics.Register("avalanche", avalancheRegisterer); err != nil {
 		return nil, err
 	}
-	if err := ctx.Context.Metrics.Register(registerer); err != nil {
+	if err := ctx.Context.Metrics.Register("", snowmanRegisterer); err != nil {
 		return nil, err
 	}
 

snow/context.go

@@ -46,7 +46,7 @@ type Context struct {
 	Keystore     keystore.BlockchainKeystore
 	SharedMemory atomic.SharedMemory
 	BCLookup     ids.AliaserReader
-	Metrics      metrics.OptionalGatherer
+	Metrics      metrics.MultiGatherer
 
 	WarpSigner warp.Signer
 

snow/snowtest/snowtest.go

@@ -90,7 +90,7 @@ func Context(tb testing.TB, chainID ids.ID) *snow.Context {
 
 		Log:      logging.NoLog{},
 		BCLookup: aliaser,
-		Metrics:  metrics.NewOptionalGatherer(),
+		Metrics:  metrics.NewMultiGatherer(),
 
 		ValidatorState: validatorState,
 		ChainDataDir:   "",

vms/avm/vm.go

@@ -174,7 +174,7 @@ func (vm *VM) Initialize(
 	)
 
 	registerer := prometheus.NewRegistry()
-	if err := ctx.Metrics.Register(registerer); err != nil {
+	if err := ctx.Metrics.Register("", registerer); err != nil {
 		return err
 	}
 	vm.registerer = registerer

vms/metervm/block_vm.go

@@ -70,18 +70,14 @@ func (vm *blockVM) Initialize(
 		return err
 	}
 
-	optionalGatherer := metrics.NewOptionalGatherer()
 	multiGatherer := metrics.NewMultiGatherer()
-	if err := multiGatherer.Register("metervm", registerer); err != nil {
+	if err := chainCtx.Metrics.Register("metervm", registerer); err != nil {
 		return err
 	}
-	if err := multiGatherer.Register("", optionalGatherer); err != nil {
+	if err := chainCtx.Metrics.Register("", multiGatherer); err != nil {
 		return err
 	}
-	if err := chainCtx.Metrics.Register(multiGatherer); err != nil {
-		return err
-	}
-	chainCtx.Metrics = optionalGatherer
+	chainCtx.Metrics = multiGatherer
 
 	return vm.ChainVM.Initialize(ctx, chainCtx, db, genesisBytes, upgradeBytes, configBytes, toEngine, fxs, appSender)
 }

vms/metervm/vertex_vm.go

@@ -50,18 +50,14 @@ func (vm *vertexVM) Initialize(
 		return err
 	}
 
-	optionalGatherer := metrics.NewOptionalGatherer()
 	multiGatherer := metrics.NewMultiGatherer()
-	if err := multiGatherer.Register("metervm", registerer); err != nil {
+	if err := chainCtx.Metrics.Register("metervm", registerer); err != nil {
 		return err
 	}
-	if err := multiGatherer.Register("", optionalGatherer); err != nil {
+	if err := chainCtx.Metrics.Register("", multiGatherer); err != nil {
 		return err
 	}
-	if err := chainCtx.Metrics.Register(multiGatherer); err != nil {
-		return err
-	}
-	chainCtx.Metrics = optionalGatherer
+	chainCtx.Metrics = multiGatherer
 
 	return vm.LinearizableVMWithEngine.Initialize(
 		ctx,

vms/platformvm/vm.go

@@ -114,7 +114,7 @@ func (vm *VM) Initialize(
 	chainCtx.Log.Info("using VM execution config", zap.Reflect("config", execConfig))
 
 	registerer := prometheus.NewRegistry()
-	if err := chainCtx.Metrics.Register(registerer); err != nil {
+	if err := chainCtx.Metrics.Register("", registerer); err != nil {
 		return err
 	}
 

vms/proposervm/vm.go

@@ -132,20 +132,15 @@ func (vm *VM) Initialize(
 ) error {
 	// TODO: Add a helper for this metrics override, it is performed in multiple
 	//       places.
-	multiGatherer := metrics.NewMultiGatherer()
 	registerer := prometheus.NewRegistry()
-	if err := multiGatherer.Register("proposervm", registerer); err != nil {
+	if err := chainCtx.Metrics.Register("proposervm", registerer); err != nil {
 		return err
 	}
-
-	optionalGatherer := metrics.NewOptionalGatherer()
-	if err := multiGatherer.Register("", optionalGatherer); err != nil {
-		return err
-	}
-	if err := chainCtx.Metrics.Register(multiGatherer); err != nil {
+	multiGatherer := metrics.NewMultiGatherer()
+	if err := chainCtx.Metrics.Register("", multiGatherer); err != nil {
 		return err
 	}
-	chainCtx.Metrics = optionalGatherer
+	chainCtx.Metrics = multiGatherer
 
 	vm.ctx = chainCtx
 	vm.db = versiondb.New(prefixdb.New(dbPrefix, db))

vms/rpcchainvm/vm_client.go

@@ -18,7 +18,6 @@ import (
 	"google.golang.org/protobuf/types/known/emptypb"
 
 	"github.com/ava-labs/avalanchego/api/keystore/gkeystore"
-	"github.com/ava-labs/avalanchego/api/metrics"
 	"github.com/ava-labs/avalanchego/chains/atomic/gsharedmemory"
 	"github.com/ava-labs/avalanchego/database"
 	"github.com/ava-labs/avalanchego/database/rpcdb"
@@ -137,15 +136,14 @@ func (vm *VMClient) Initialize(
 
 	// Register metrics
 	registerer := prometheus.NewRegistry()
-	multiGatherer := metrics.NewMultiGatherer()
 	vm.grpcServerMetrics = grpc_prometheus.NewServerMetrics()
 	if err := registerer.Register(vm.grpcServerMetrics); err != nil {
 		return err
 	}
-	if err := multiGatherer.Register("rpcchainvm", registerer); err != nil {
+	if err := chainCtx.Metrics.Register("rpcchainvm", registerer); err != nil {
 		return err
 	}
-	if err := multiGatherer.Register("", vm); err != nil {
+	if err := chainCtx.Metrics.Register("", vm); err != nil {
 		return err
 	}
 
@@ -226,7 +224,7 @@ func (vm *VMClient) Initialize(
 		time:     time,
 	}
 
-	chainState, err := chain.NewMeteredState(
+	vm.State, err = chain.NewMeteredState(
 		registerer,
 		&chain.Config{
 			DecidedCacheSize:      decidedCacheSize,
@@ -241,12 +239,7 @@ func (vm *VMClient) Initialize(
 			BuildBlockWithContext: vm.buildBlockWithContext,
 		},
 	)
-	if err != nil {
-		return err
-	}
-	vm.State = chainState
-
-	return chainCtx.Metrics.Register(multiGatherer)
+	return err
 }
 
 func (vm *VMClient) newDBServer(db database.Database) *grpc.Server {

vms/rpcchainvm/vm_server.go

@@ -225,7 +225,7 @@ func (vm *VMServer) Initialize(ctx context.Context, req *vmpb.InitializeRequest)
 		Keystore:     keystoreClient,
 		SharedMemory: sharedMemoryClient,
 		BCLookup:     bcLookupClient,
-		Metrics:      metrics.NewOptionalGatherer(),
+		Metrics:      metrics.NewMultiGatherer(),
 
 		// Signs warp messages
 		WarpSigner: warpSignerClient,