github.com/celestiaorg/celestia-node@v0.15.0-beta.1/nodebuilder/share/opts.go (about)

     1  package share
     2  
     3  import (
     4  	"github.com/celestiaorg/celestia-node/share/eds"
     5  	"github.com/celestiaorg/celestia-node/share/getters"
     6  	disc "github.com/celestiaorg/celestia-node/share/p2p/discovery"
     7  	"github.com/celestiaorg/celestia-node/share/p2p/peers"
     8  	"github.com/celestiaorg/celestia-node/share/p2p/shrexeds"
     9  	"github.com/celestiaorg/celestia-node/share/p2p/shrexnd"
    10  )
    11  
    12  // WithPeerManagerMetrics is a utility function to turn on peer manager metrics and that is
    13  // expected to be "invoked" by the fx lifecycle.
    14  func WithPeerManagerMetrics(m *peers.Manager) error {
    15  	return m.WithMetrics()
    16  }
    17  
    18  // WithDiscoveryMetrics is a utility function to turn on discovery metrics and that is expected to
    19  // be "invoked" by the fx lifecycle.
    20  func WithDiscoveryMetrics(d *disc.Discovery) error {
    21  	return d.WithMetrics()
    22  }
    23  
    24  func WithShrexClientMetrics(edsClient *shrexeds.Client, ndClient *shrexnd.Client) error {
    25  	err := edsClient.WithMetrics()
    26  	if err != nil {
    27  		return err
    28  	}
    29  
    30  	return ndClient.WithMetrics()
    31  }
    32  
    33  func WithShrexServerMetrics(edsServer *shrexeds.Server, ndServer *shrexnd.Server) error {
    34  	err := edsServer.WithMetrics()
    35  	if err != nil {
    36  		return err
    37  	}
    38  
    39  	return ndServer.WithMetrics()
    40  }
    41  
    42  func WithShrexGetterMetrics(sg *getters.ShrexGetter) error {
    43  	return sg.WithMetrics()
    44  }
    45  
    46  func WithStoreMetrics(s *eds.Store) error {
    47  	return s.WithMetrics()
    48  }