update block hashes

Author: colinlyguo

coordinator/internal/logic/provertask/chunk_prover_task.go

@@ -10,6 +10,7 @@ import (
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promauto"
 	"github.com/scroll-tech/da-codec/encoding"
+	"github.com/scroll-tech/go-ethereum/common"
 	"github.com/scroll-tech/go-ethereum/log"
 	"github.com/scroll-tech/go-ethereum/params"
 	"gorm.io/gorm"
@@ -196,15 +197,30 @@ func (cp *ChunkProverTask) hardForkName(ctx *gin.Context, chunkTask *orm.Chunk)
 }
 
 func (cp *ChunkProverTask) formatProverTask(ctx context.Context, task *orm.ProverTask, hardForkName string) (*coordinatorType.GetTaskSchema, error) {
-	// Get block hashes.
-	blockHashes, dbErr := cp.blockOrm.GetL2BlockHashesByChunkHash(ctx, task.TaskID)
-	if dbErr != nil || len(blockHashes) == 0 {
+	dbChunk, err := cp.chunkOrm.GetChunkByHash(ctx, task.TaskID)
+	if err != nil {
+		return nil, fmt.Errorf("failed to fetch chunk by hash:%s err:%w", task.TaskID, err)
+	}
+
+	blocks, dbErr := cp.blockOrm.GetL2BlocksInRange(ctx, dbChunk.StartBlockNumber+1, dbChunk.EndBlockNumber+1)
+	if dbErr != nil || len(blocks) == 0 {
 		return nil, fmt.Errorf("failed to fetch block hashes of a chunk, chunk hash:%s err:%w", task.TaskID, dbErr)
 	}
 
+	if len(blocks) != int(dbChunk.EndBlockNumber-dbChunk.StartBlockNumber+1) {
+		return nil, fmt.Errorf("failed to fetch all block hashes of a chunk, chunk hash:%s, expected block number:%d, actual block number:%d",
+			task.TaskID, dbChunk.EndBlockNumber-dbChunk.StartBlockNumber+1, len(blocks))
+	}
+
+	var blockHashes []common.Hash
+	for _, block := range blocks {
+		blockHashes = append(blockHashes, block.Header.Hash())
+	}
+
 	taskDetail := message.ChunkTaskDetail{
 		BlockHashes: blockHashes,
 	}
+
 	blockHashesBytes, err := json.Marshal(taskDetail)
 	if err != nil {
 		return nil, fmt.Errorf("failed to marshal block hashes hash:%s, err:%w", task.TaskID, err)