99 lines
3.6 KiB
Nim
99 lines
3.6 KiB
Nim
# nimbus-eth1
|
|
# Copyright (c) 2023-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.
|
|
|
|
## Rocks DB internal driver descriptor
|
|
## ===================================
|
|
|
|
{.push raises: [].}
|
|
|
|
import
|
|
std/os,
|
|
eth/common,
|
|
rocksdb,
|
|
stew/[endians2, keyed_queue],
|
|
../../aristo_desc,
|
|
../init_common
|
|
|
|
type
|
|
RdbWriteEventCb* =
|
|
proc(session: WriteBatchRef): bool {.gcsafe, raises: [].}
|
|
## Call back closure function that passes the the write session handle
|
|
## to a guest peer right after it was opened. The guest may store any
|
|
## data on its own column family and return `true` if that worked
|
|
## all right. Then the `Aristo` handler will stor its own columns and
|
|
## finalise the write session.
|
|
##
|
|
## In case of an error when `false` is returned, `Aristo` will abort the
|
|
## write session and return a session error.
|
|
|
|
RdbInst* = object
|
|
admCol*: ColFamilyReadWrite ## Admin column family handler
|
|
vtxCol*: ColFamilyReadWrite ## Vertex column family handler
|
|
keyCol*: ColFamilyReadWrite ## Hash key column family handler
|
|
session*: WriteBatchRef ## For batched `put()`
|
|
|
|
# Note that the key type `VertexID` for LRU caches requires that there is
|
|
# strictly no vertex ID re-use.
|
|
#
|
|
# Otherwise, in some fringe cases one might remove a vertex with key
|
|
# `(root1,vid)` and insert another vertex with key `(root2,vid)` while
|
|
# re-using the vertex ID `vid`. Without knowledge of `root1` and `root2`,
|
|
# the LRU cache will return the same vertex for `(root2,vid)` also for
|
|
# `(root1,vid)`.
|
|
#
|
|
# The other alternaive would be to use the key type `RootedVertexID` which
|
|
# is less memory and time efficient (the latter one due to internal LRU
|
|
# handling of the longer key.)
|
|
#
|
|
rdKeyLru*: KeyedQueue[VertexID,HashKey] ## Read cache
|
|
rdVtxLru*: KeyedQueue[VertexID,VertexRef] ## Read cache
|
|
|
|
basePath*: string ## Database directory
|
|
trgWriteEvent*: RdbWriteEventCb ## Database piggiback call back handler
|
|
|
|
AristoCFs* = enum
|
|
## Column family symbols/handles and names used on the database
|
|
AdmCF = "AriAdm" ## Admin column family name
|
|
VtxCF = "AriVtx" ## Vertex column family name
|
|
KeyCF = "AriKey" ## Hash key column family name
|
|
|
|
const
|
|
BaseFolder* = "nimbus" ## Same as for Legacy DB
|
|
DataFolder* = "aristo" ## Legacy DB has "data"
|
|
RdKeyLruMaxSize* = 80000
|
|
## Max size of read cache for keys - ~4 levels of MPT
|
|
RdVtxLruMaxSize* = 80000
|
|
## Max size of read cache for vertex IDs - ~4 levels of MPT
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# Public functions
|
|
# ------------------------------------------------------------------------------
|
|
|
|
template logTxt*(info: static[string]): static[string] =
|
|
"RocksDB/" & info
|
|
|
|
template baseDb*(rdb: RdbInst): RocksDbReadWriteRef =
|
|
rdb.admCol.db
|
|
|
|
|
|
func baseDir*(rdb: RdbInst): string =
|
|
rdb.basePath / BaseFolder
|
|
|
|
func dataDir*(rdb: RdbInst): string =
|
|
rdb.baseDir / DataFolder
|
|
|
|
|
|
template toOpenArray*(xid: AdminTabID): openArray[byte] =
|
|
xid.uint64.toBytesBE.toOpenArray(0,7)
|
|
|
|
# ------------------------------------------------------------------------------
|
|
# End
|
|
# ------------------------------------------------------------------------------
|