nimbus-eth1/premix/persist.nim

108 lines
2.8 KiB
Nim
Raw Normal View History

2019-01-06 15:21:34 +00:00
# use this module to quickly populate db with data from geth/parity
import
2019-02-05 19:15:50 +00:00
eth/[common, rlp], stint, byteutils, nimcrypto,
chronicles, downloader, configuration,
2019-01-09 12:10:58 +00:00
../nimbus/errors
2019-01-06 15:21:34 +00:00
import
2019-02-05 19:15:50 +00:00
eth/trie/[hexary, db, trie_defs],
2019-01-06 15:21:34 +00:00
../nimbus/db/[storage_types, db_chain, select_backend],
2019-01-08 11:29:56 +00:00
../nimbus/[genesis, utils],
2019-01-06 15:21:34 +00:00
../nimbus/p2p/chain
const
manualCommit = nimbus_db_backend == "lmdb"
template persistToDb(db: ChainDB, body: untyped) =
when manualCommit:
if not db.txBegin(): assert(false)
body
when manualCommit:
if not db.txCommit(): assert(false)
proc main() =
# 97 block with uncles
# 46147 block with first transaction
# 46400 block with transaction
# 46402 block with first contract: failed
# 47205 block with first success contract
# 48712 block with 5 transactions
# 48915 block with contract
# 49018 first problematic block
# 49439 first block with contract call
# 52029 first block with receipts logs
# 66407 failed transaction
2019-01-10 08:23:18 +00:00
let conf = getConfiguration()
2019-01-06 15:21:34 +00:00
let db = newChainDb(conf.dataDir)
let trieDB = trieDB db
let chainDB = newBaseChainDB(trieDB, false)
# move head to block number ...
2019-01-08 11:29:56 +00:00
if conf.head != 0.u256:
var parentBlock = requestBlock(conf.head)
chainDB.setHead(parentBlock.header)
2019-01-06 15:21:34 +00:00
if canonicalHeadHashKey().toOpenArray notin trieDB:
persistToDb(db):
initializeEmptyDb(chainDB)
assert(canonicalHeadHashKey().toOpenArray in trieDB)
var head = chainDB.getCanonicalHead()
var blockNumber = head.blockNumber + 1
var chain = newChain(chainDB)
2019-01-08 11:29:56 +00:00
let numBlocksToCommit = conf.numCommits
2019-01-06 15:21:34 +00:00
var headers = newSeqOfCap[BlockHeader](numBlocksToCommit)
var bodies = newSeqOfCap[BlockBody](numBlocksToCommit)
var one = 1.u256
var numBlocks = 0
2019-01-08 11:29:56 +00:00
var counter = 0
2019-03-11 13:21:09 +00:00
while true:
2019-01-06 15:21:34 +00:00
var thisBlock = requestBlock(blockNumber)
headers.add thisBlock.header
bodies.add thisBlock.body
2019-03-11 13:21:09 +00:00
info "REQUEST HEADER", blockNumber=blockNumber, txs=thisBlock.body.transactions.len
2019-01-06 15:21:34 +00:00
inc numBlocks
blockNumber += one
if numBlocks == numBlocksToCommit:
persistToDb(db):
2019-01-08 11:29:56 +00:00
if chain.persistBlocks(headers, bodies) != ValidationResult.OK:
2019-01-09 12:10:58 +00:00
raise newException(ValidationError, "Error when validating blocks")
2019-01-06 15:21:34 +00:00
numBlocks = 0
headers.setLen(0)
bodies.setLen(0)
2019-01-08 11:29:56 +00:00
inc counter
if conf.maxBlocks != 0 and counter >= conf.maxBlocks:
break
2019-01-06 15:21:34 +00:00
if numBlocks > 0:
persistToDb(db):
2019-01-09 12:10:58 +00:00
if chain.persistBlocks(headers, bodies) != ValidationResult.OK:
raise newException(ValidationError, "Error when validating blocks")
2019-01-06 15:21:34 +00:00
2019-01-08 11:29:56 +00:00
when isMainModule:
var message: string
## Processing command line arguments
if processArguments(message) != Success:
echo message
quit(QuitFailure)
else:
if len(message) > 0:
echo message
quit(QuitSuccess)
2019-01-09 12:10:58 +00:00
try:
main()
except:
echo getCurrentExceptionMsg()