Populate PruningPointBlueWork with the actual blue work of the prunin…

…g point.
stasatdaglabs · Sep 5, 2021 · f2a9829 · f2a9829
1 parent c6a90c5
commit f2a9829
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/app/protocol/flows/blockrelay/ibd_with_headers_proof.go b/app/protocol/flows/blockrelay/ibd_with_headers_proof.go
@@ -122,7 +122,7 @@ func (flow *handleRelayInvsFlow) downloadHeadersAndPruningUTXOSet(highHash *exte
 	if err != nil {
 		return err
 	}
-	pruningPointHeader, err := flow.Domain().Consensus().GetBlockHeader(pruningPointHash)
+	pruningPointHeader, err := flow.Domain().StagingConsensus().GetBlockHeader(pruningPointHash)
 	if err != nil {
 		return err
 	}

diff --git a/domain/consensus/consensus.go b/domain/consensus/consensus.go
@@ -727,9 +727,19 @@ func (s *consensus) BuildPruningPointProof() (*externalapi.PruningPointProof, er
 	s.lock.Lock()
 	defer s.lock.Unlock()
 
+	stagingArea := model.NewStagingArea()
+	pruningPointHash, err := s.pruningStore.PruningPoint(s.databaseContext, stagingArea)
+	if err != nil {
+		return nil, err
+	}
+	pruningPointHeader, err := s.blockHeaderStore.BlockHeader(s.databaseContext, stagingArea, pruningPointHash)
+	if err != nil {
+		return nil, err
+	}
+
 	return &externalapi.PruningPointProof{
 		Headers:              []externalapi.BlockHeader{},
-		PruningPointBlueWork: big.NewInt(0),
+		PruningPointBlueWork: pruningPointHeader.BlueWork(),
 	}, nil
 }