add aggregated_price metric

Author: Richard Patel

main.go

@@ -4,7 +4,6 @@ import (
 	"context"
 	"flag"
 	"fmt"
-	"math"
 	"net/http"
 	"strings"
 	"time"
@@ -120,35 +119,13 @@ func main() {
 	})
 
 	// Send price updates to Prometheus.
+	prices := priceScraper{
+		productKeys: productKeys.pubkeys,
+		publishKeys: publishKeys.pubkeys,
+	}
 	group.Go(func() error {
 		for update := range updates {
-			// Filter by product key.
-			var knownProduct bool
-			for _, product := range productKeys.pubkeys {
-				if product == update.Product {
-					knownProduct = true
-					break
-				}
-			}
-			if !knownProduct {
-				continue
-			}
-			decimals := math.Pow10(int(update.Exponent))
-			// Update price of each publisher.
-			for _, publisher := range publishKeys.pubkeys {
-				comp := update.GetComponent(&publisher)
-				if comp != nil {
-					metrics.PublisherSlot.
-						WithLabelValues(update.Product.String(), publisher.String()).
-						Set(float64(comp.Latest.PubSlot))
-					metrics.PublisherPrice.
-						WithLabelValues(update.Product.String(), publisher.String()).
-						Set(float64(comp.Latest.Price) * decimals)
-					metrics.PublisherConf.
-						WithLabelValues(update.Product.String(), publisher.String()).
-						Set(float64(comp.Latest.Conf) * decimals)
-				}
-			}
+			prices.onUpdate(update)
 		}
 		return nil
 	})

metrics/metrics.go

@@ -55,6 +55,18 @@ var (
 		Name:      "publish_account_balance",
 		Help:      "SOL balance of Pyth publish account in lamports",
 	}, []string{labelPublisher})
+	AggPrice = factory.NewGaugeVec(prometheus.GaugeOpts{
+		Namespace: promNamespace,
+		Subsystem: promSubsystemOracle,
+		Name:      "aggregated_price",
+		Help:      "Last aggregated price of Pyth product",
+	}, []string{labelProduct})
+	AggConf = factory.NewGaugeVec(prometheus.GaugeOpts{
+		Namespace: promNamespace,
+		Subsystem: promSubsystemOracle,
+		Name:      "aggregated_conf_amount",
+		Help:      "Last aggregated conf of Pyth product",
+	}, []string{labelProduct})
 	PublisherPrice = factory.NewGaugeVec(prometheus.GaugeOpts{
 		Namespace: promNamespace,
 		Subsystem: promSubsystemOracle,
@@ -78,11 +90,11 @@ var (
 	metricTxFeesTotal = factory.NewCounterVec(prometheus.CounterOpts{
 		Namespace: promNamespace,
 		Name:      "tx_fees_total",
-		Help:      "Cumulative amount of SOL in lamports spent on Pyth publishing",
+		Help:      "Approximate amount of SOL in lamports spent on Pyth publishing",
 	}, []string{labelPublisher})
 	metricTxCount = factory.NewCounterVec(prometheus.CounterOpts{
 		Namespace: promNamespace,
 		Name:      "txs_total",
-		Help:      "Number of Pyth transactions sent",
+		Help:      "Approximate number of Pyth transactions sent",
 	}, []string{labelPublisher, labelTxStatus})
 )

price.go

@@ -0,0 +1,64 @@
+package main
+
+import (
+	"math"
+
+	"github.com/gagliardetto/solana-go"
+	"go.blockdaemon.com/pyth_exporter/metrics"
+	"go.blockdaemon.com/pyth_exporter/pyth"
+)
+
+type priceScraper struct {
+	productKeys []solana.PublicKey
+	publishKeys []solana.PublicKey
+}
+
+func (p *priceScraper) onUpdate(update pyth.PriceAccountUpdate) {
+	if !p.isInteresting(update) {
+		return
+	}
+	decimals := math.Pow10(int(update.Exponent))
+	p.aggregate(&update.Product, &update.Agg, decimals)
+	// Update price of each publisher.
+	for _, publisher := range p.publishKeys {
+		comp := update.GetComponent(&publisher)
+		if comp != nil {
+			p.component(&update.Product, &publisher, comp, decimals)
+		}
+	}
+}
+
+func (p *priceScraper) isInteresting(update pyth.PriceAccountUpdate) bool {
+	for _, product := range p.productKeys {
+		if product == update.Product {
+			return true
+		}
+	}
+	return false
+}
+
+// aggregate exports price as aggregated by the smart contract.
+func (p *priceScraper) aggregate(product *solana.PublicKey, agg *pyth.PriceInfo, decimals float64) {
+	productStr := product.String()
+	metrics.AggPrice.
+		WithLabelValues(productStr).
+		Set(float64(agg.Price) * decimals)
+	metrics.AggConf.
+		WithLabelValues(productStr).
+		Set(float64(agg.Conf) * decimals)
+}
+
+// component exports a price component (i.e. a price value published by an individual Pyth publisher).
+func (p *priceScraper) component(product *solana.PublicKey, publisher *solana.PublicKey, comp *pyth.PriceComp, decimals float64) {
+	productStr := product.String()
+	publisherStr := publisher.String()
+	metrics.PublisherSlot.
+		WithLabelValues(productStr, publisherStr).
+		Set(float64(comp.Latest.PubSlot))
+	metrics.PublisherPrice.
+		WithLabelValues(productStr, publisherStr).
+		Set(float64(comp.Latest.Price) * decimals)
+	metrics.PublisherConf.
+		WithLabelValues(productStr, publisherStr).
+		Set(float64(comp.Latest.Conf) * decimals)
+}