Update buckets withdrawals on API

This commit is contained in:
ToniRamirezM
2021-01-11 15:06:36 +01:00
parent ef8a6d1397
commit 0bb69c3f3e
3 changed files with 71 additions and 1 deletions

View File

@@ -109,6 +109,24 @@ func (a *API) UpdateNetworkInfo(
a.status.Network.LastBatch = lastBatch
a.status.Network.CurrentSlot = currentSlot
a.status.Network.NextForgers = nextForgers
// Update buckets withdrawals
bucketsUpdate, err := a.h.GetBucketUpdates()
if tracerr.Unwrap(err) == sql.ErrNoRows {
bucketsUpdate = nil
} else if err != nil {
return tracerr.Wrap(err)
}
for i, bucketParams := range a.status.Rollup.Buckets {
for _, bucketUpdate := range bucketsUpdate {
if bucketUpdate.NumBucket == i {
bucketParams.Withdrawals = bucketUpdate.Withdrawals
a.status.Rollup.Buckets[i] = bucketParams
break
}
}
}
a.status.Unlock()
return nil
}