historysegment: add testnet&mainnet hard code;

bnb-chain · Dec 8, 2023 · 39bf075 · 39bf075
1 parent 080828f
commit 39bf075
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 15 deletions.
diff --git a/cmd/geth/chaincmd.go b/cmd/geth/chaincmd.go
@@ -736,19 +736,18 @@ func exportSegment(ctx *cli.Context) error {
 		boundStartBlock      = params.BoundStartBlock
 		historySegmentLength = params.HistorySegmentLength
 	)
-	// TODO(0xbundler): for testing
-	//switch genesisHash {
-	//case params.BSCGenesisHash, params.ChapelGenesisHash, params.RialtoGenesisHash:
-	//	boundStartBlock = params.BoundStartBlock
-	//	historySegmentLength = params.HistorySegmentLength
-	//default:
-	if ctx.IsSet(utils.BoundStartBlockFlag.Name) {
-		boundStartBlock = ctx.Uint64(utils.BoundStartBlockFlag.Name)
-	}
-	if ctx.IsSet(utils.HistorySegmentLengthFlag.Name) {
-		historySegmentLength = ctx.Uint64(utils.HistorySegmentLengthFlag.Name)
-	}
-	//}
+	switch genesisHash {
+	case params.BSCGenesisHash, params.ChapelGenesisHash, params.RialtoGenesisHash:
+		boundStartBlock = params.BoundStartBlock
+		historySegmentLength = params.HistorySegmentLength
+	default:
+		if ctx.IsSet(utils.BoundStartBlockFlag.Name) {
+			boundStartBlock = ctx.Uint64(utils.BoundStartBlockFlag.Name)
+		}
+		if ctx.IsSet(utils.HistorySegmentLengthFlag.Name) {
+			historySegmentLength = ctx.Uint64(utils.HistorySegmentLengthFlag.Name)
+		}
+	}
 	if boundStartBlock == 0 || historySegmentLength == 0 {
 		return fmt.Errorf("wrong params, boundStartBlock: %v, historySegmentLength: %v", boundStartBlock, historySegmentLength)
 	}