add tier chances to report

lavanet · Dec 1, 2024 · 17166c6 · 17166c6
1 parent 3e7dd86
commit 17166c6
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 12 deletions.
diff --git a/protocol/metrics/consumer_optimizer_qos_client.go b/protocol/metrics/consumer_optimizer_qos_client.go
@@ -41,21 +41,23 @@ type OptimizerQoSReport struct {
 	LatencyScore      float64
 	GenericScore      float64
 	EntryIndex        int
+	TierChances       map[int]float64
 }
 
 type optimizerQoSReportToSend struct {
-	Timestamp         time.Time `json:"timestamp"`
-	SyncScore         float64   `json:"sync_score"`
-	AvailabilityScore float64   `json:"availability_score"`
-	LatencyScore      float64   `json:"latency_score"`
-	GenericScore      float64   `json:"generic_score"`
-	ProviderAddress   string    `json:"provider"`
-	ConsumerOrigin    string    `json:"consumer"`
-	ChainId           string    `json:"chain_id"`
-	NodeErrorRate     float64   `json:"node_error_rate"`
-	Epoch             uint64    `json:"epoch"`
-	ProviderStake     int64     `json:"provider_stake"`
-	EntryIndex        int       `json:"entry_index"`
+	Timestamp         time.Time       `json:"timestamp"`
+	SyncScore         float64         `json:"sync_score"`
+	AvailabilityScore float64         `json:"availability_score"`
+	LatencyScore      float64         `json:"latency_score"`
+	GenericScore      float64         `json:"generic_score"`
+	ProviderAddress   string          `json:"provider"`
+	ConsumerOrigin    string          `json:"consumer"`
+	ChainId           string          `json:"chain_id"`
+	NodeErrorRate     float64         `json:"node_error_rate"`
+	Epoch             uint64          `json:"epoch"`
+	ProviderStake     int64           `json:"provider_stake"`
+	EntryIndex        int             `json:"entry_index"`
+	TierChances       map[int]float64 `json:"tier_chances"`
 }
 
 func (oqosr optimizerQoSReportToSend) String() string {

diff --git a/protocol/provideroptimizer/provider_optimizer.go b/protocol/provideroptimizer/provider_optimizer.go
@@ -202,9 +202,11 @@ func (po *ProviderOptimizer) CalculateQoSScoresForMetrics(allAddresses []string,
 	reports := []*metrics.OptimizerQoSReport{}
 
 	rawScores := selectionTier.GetRawScores()
+	tierChances := selectionTier.ShiftTierChance(po.OptimizerNumTiers, map[int]float64{0: ATierChance})
 	for idx, entry := range rawScores {
 		qosReport := providersScores[entry.Address]
 		qosReport.EntryIndex = idx
+		qosReport.TierChances = tierChances
 		reports = append(reports, qosReport)
 	}
 

diff --git a/protocol/provideroptimizer/provider_optimizer_refactor.go b/protocol/provideroptimizer/provider_optimizer_refactor.go
@@ -168,10 +168,12 @@ func (po *ProviderOptimizer_Refactor) CalculateQoSScoresForMetrics(allAddresses
 	selectionTier, _, providersScores := po.CalculateSelectionTiers_Refactor(allAddresses, ignoredProviders, cu, requestedBlock)
 	reports := []*metrics.OptimizerQoSReport{}
 
+	tierChances := selectionTier.ShiftTierChance(po.OptimizerNumTiers, map[int]float64{0: ATierChance_Refactor})
 	rawScores := selectionTier.GetRawScores()
 	for idx, entry := range rawScores {
 		qosReport := providersScores[entry.Address]
 		qosReport.EntryIndex = idx
+		qosReport.TierChances = tierChances
 		reports = append(reports, qosReport)
 	}