certusone
/
solana_exporter
mirror of https://github.com/certusone/solana_exporter.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

added rpc.Provider interface

This commit is contained in:
Matt Johnstone 2024-06-11 22:23:10 +02:00
parent eef3cea353
commit 89d209f188
No known key found for this signature in database
GPG Key ID: 7D96C656728409F9
3 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
cmd/solana_exporter/exporter.go
View File

@ -25,7 +25,7 @@ func init() {
}
type solanaCollector struct {
rpcClient *rpc.Client
rpcClient rpc.Provider
totalValidatorsDesc *prometheus.Desc
validatorActivatedStake *prometheus.Desc
@ -35,9 +35,9 @@ type solanaCollector struct {
solanaVersion *prometheus.Desc
}
func NewSolanaCollector(rpcAddr string) *solanaCollector {
func createSolanaCollector(provider rpc.Provider) *solanaCollector {
return &solanaCollector{
rpcClient: rpc.NewRPCClient(rpcAddr),
rpcClient: provider,
totalValidatorsDesc: prometheus.NewDesc(
"solana_active_validators",
"Total number of active validators by state",
@ -65,6 +65,10 @@ func NewSolanaCollector(rpcAddr string) *solanaCollector {
}
}
func NewSolanaCollector(rpcAddr string) *solanaCollector {
return createSolanaCollector(rpc.NewRPCClient(rpcAddr))
}
func (c *solanaCollector) Describe(ch chan<- *prometheus.Desc) {
ch <- c.totalValidatorsDesc
ch <- c.solanaVersion

2
cmd/solana_exporter/slots.go
View File

@ -164,7 +164,7 @@ func getEpochBounds(info *rpc.EpochInfo) (int64, int64, int64) {
return info.Epoch, firstSlot, lastSlot
}
func updateCounters(c *rpc.Client, epoch, firstSlot int64, lastSlotOpt *int64) (int64, error) {
func updateCounters(c rpc.Provider, epoch, firstSlot int64, lastSlotOpt *int64) (int64, error) {
ctx, cancel := context.WithTimeout(context.Background(), httpTimeout)
defer cancel()

8
pkg/rpc/client.go
View File

@ -35,6 +35,14 @@ type (
Commitment string
)
type Provider interface {
GetBlockProduction(ctx context.Context, firstSlot *int64, lastSlot *int64) (BlockProduction, error)
GetEpochInfo(ctx context.Context, commitment Commitment) (*EpochInfo, error)
GetSlot(ctx context.Context) (int64, error)
GetVoteAccounts(ctx context.Context, params []interface{}) (*GetVoteAccountsResponse, error)
GetVersion(ctx context.Context) (*string, error)
}
func (c Commitment) MarshalJSON() ([]byte, error) {
return json.Marshal(map[string]string{"commitment": string(c)})
}