From c68d6f7c987ebb5c027980b76cac475ab7682d8a Mon Sep 17 00:00:00 2001
From: galaio <12880651+galaio@users.noreply.github.com>
Date: Wed, 14 Aug 2024 11:28:04 +0800
Subject: [PATCH] txdag: support reset txdag reader when SetHead; (#31)

* txdag: support reset txdag reader when SetHead;

* txdag: clean some useless logs;

---------

Co-authored-by: galaio <galaio@users.noreply.github.com>
---
 core/blockchain.go      | 48 ++++++++++++++++++++++++++++++++---------
 core/blockchain_test.go | 14 ++++++++++++
 2 files changed, 52 insertions(+), 10 deletions(-)

diff --git a/core/blockchain.go b/core/blockchain.go
index 9f4bd3da3f..6d56d27523 100644
--- a/core/blockchain.go
+++ b/core/blockchain.go
@@ -886,6 +886,10 @@ func (bc *BlockChain) setHeadBeyondRoot(head uint64, time uint64, root common.Ha
 	bc.miningStateCache.Purge()
 	bc.futureBlocks.Purge()
 
+	if bc.txDAGReader != nil {
+		bc.txDAGReader.Reset(head)
+	}
+
 	// Clear safe block, finalized block if needed
 	if safe := bc.CurrentSafeBlock(); safe != nil && head < safe.Number.Uint64() {
 		log.Warn("SetHead invalidated safe block")
@@ -1992,7 +1996,7 @@ func (bc *BlockChain) insertChain(chain types.Blocks, setHead bool) (int, error)
 			return it.index, err
 		}
 
-		if bc.enableTxDAG {
+		if bc.enableTxDAG && !bc.parallelExecution {
 			// compare input TxDAG when it enable in consensus
 			dag, err := statedb.ResolveTxDAG(len(block.Transactions()), []common.Address{block.Coinbase(), params.OptimismBaseFeeRecipient, params.OptimismL1FeeRecipient})
 			if err == nil {
@@ -2917,9 +2921,10 @@ func writeTxDAGToFile(writeHandle *os.File, item TxDAGOutputItem) error {
 	return err
 }
 
-var TxDAGCacheSize = 10000
+var TxDAGCacheSize = uint64(10000)
 
 type TxDAGFileReader struct {
+	output  string
 	file    *os.File
 	scanner *bufio.Scanner
 	cache   map[uint64]types.TxDAG
@@ -2928,16 +2933,12 @@ type TxDAGFileReader struct {
 }
 
 func NewTxDAGFileReader(output string) (*TxDAGFileReader, error) {
-	file, err := os.Open(output)
+	reader := &TxDAGFileReader{output: output}
+	err := reader.openFile(output)
 	if err != nil {
 		return nil, err
 	}
-	scanner := bufio.NewScanner(file)
-	scanner.Buffer(make([]byte, 5*1024*1024), 5*1024*1024)
-	return &TxDAGFileReader{
-		file:    file,
-		scanner: scanner,
-	}, nil
+	return reader, nil
 }
 
 func (t *TxDAGFileReader) Close() {
@@ -2946,6 +2947,18 @@ func (t *TxDAGFileReader) Close() {
 	t.closeFile()
 }
 
+func (t *TxDAGFileReader) openFile(output string) error {
+	file, err := os.Open(output)
+	if err != nil {
+		return err
+	}
+	scanner := bufio.NewScanner(file)
+	scanner.Buffer(make([]byte, 5*1024*1024), 5*1024*1024)
+	t.file = file
+	t.scanner = scanner
+	return nil
+}
+
 func (t *TxDAGFileReader) closeFile() {
 	if t.scanner != nil {
 		t.scanner = nil
@@ -2991,7 +3004,7 @@ func (t *TxDAGFileReader) TxDAG(expect uint64) types.TxDAG {
 		}
 		t.cache[num] = dag
 		t.latest = num
-		if len(t.cache) >= TxDAGCacheSize {
+		if uint64(len(t.cache)) >= TxDAGCacheSize {
 			break
 		}
 	}
@@ -3005,6 +3018,21 @@ func (t *TxDAGFileReader) TxDAG(expect uint64) types.TxDAG {
 	return t.cache[expect]
 }
 
+func (t *TxDAGFileReader) Reset(number uint64) error {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+	if t.latest-TxDAGCacheSize <= number {
+		return nil
+	}
+	t.closeFile()
+	if err := t.openFile(t.output); err != nil {
+		return err
+	}
+	t.latest = 0
+	t.cache = nil
+	return nil
+}
+
 func readTxDAGItemFromLine(line string) (uint64, types.TxDAG, error) {
 	tokens := strings.Split(line, ",")
 	if len(tokens) != 2 {
diff --git a/core/blockchain_test.go b/core/blockchain_test.go
index a285fdad1c..2f000aba67 100644
--- a/core/blockchain_test.go
+++ b/core/blockchain_test.go
@@ -4789,6 +4789,20 @@ func TestTxDAGFile_LargeRead(t *testing.T) {
 	for i := uint64(0); i < totalSize; i++ {
 		require.Equal(t, except[i], reader.TxDAG(i), i)
 	}
+
+	// test reset to genesis
+	err = reader.Reset(0)
+	require.NoError(t, err)
+	for i := uint64(0); i < totalSize; i++ {
+		require.Equal(t, except[i], reader.TxDAG(i), i)
+	}
+
+	// test reset skip
+	err = reader.Reset(totalSize - TxDAGCacheSize)
+	require.NoError(t, err)
+	for i := totalSize - TxDAGCacheSize; i < totalSize; i++ {
+		require.Equal(t, except[i], reader.TxDAG(i), i)
+	}
 }
 
 func makeEmptyPlainTxDAG(cnt int, flags ...uint8) *types.PlainTxDAG {