55 lines
1.8 KiB
Nim
55 lines
1.8 KiB
Nim
# Nimbus
|
|
# Copyright (c) 2021-2024 Status Research & Development GmbH
|
|
# Licensed under either of
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or
|
|
# http://www.apache.org/licenses/LICENSE-2.0)
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT) or
|
|
# http://opensource.org/licenses/MIT)
|
|
# at your option. This file may not be copied, modified, or distributed except
|
|
# according to those terms.
|
|
|
|
import results, eth/common, ../../nimbus/db/era1_db
|
|
|
|
var noisy* = false
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Public undump
|
|
# ------------------------------------------------------------------------------
|
|
|
|
iterator undumpBlocksEra1*(
|
|
dir: string,
|
|
least = low(uint64), # First block to extract
|
|
stopAfter = high(uint64), # Last block to extract
|
|
): (seq[BlockHeader], seq[BlockBody]) =
|
|
let db = Era1DbRef.init(dir, "mainnet").expect("Era files present")
|
|
defer:
|
|
db.dispose()
|
|
|
|
# TODO it would be a lot more natural for this iterator to return 1 block at
|
|
# a time and let the consumer do the chunking
|
|
const blocksPerYield = 192
|
|
var tmp =
|
|
(newSeqOfCap[BlockHeader](blocksPerYield), newSeqOfCap[BlockBody](blocksPerYield))
|
|
|
|
for i in 0 ..< stopAfter:
|
|
var bck = db.getBlockTuple(least + i).valueOr:
|
|
doAssert i > 0, "expected at least one block"
|
|
break
|
|
|
|
tmp[0].add move(bck[0])
|
|
tmp[1].add move(bck[1])
|
|
|
|
# Genesis block requires a chunk of its own, for compatibility with current
|
|
# test setup (a bit weird, that...)
|
|
if tmp[0].len mod blocksPerYield == 0 or tmp[0][0].blockNumber == 0:
|
|
yield tmp
|
|
tmp[0].setLen(0)
|
|
tmp[1].setLen(0)
|
|
|
|
if tmp[0].len > 0:
|
|
yield tmp
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# End
|
|
# ------------------------------------------------------------------------------
|