mirror of https://github.com/waku-org/nwaku.git
112 lines
3.3 KiB
Nim
112 lines
3.3 KiB
Nim
|
|
when (NimMajor, NimMinor) < (1, 4):
|
|
{.push raises: [Defect].}
|
|
else:
|
|
{.push raises: [].}
|
|
|
|
import
|
|
stew/results,
|
|
chronicles,
|
|
chronos
|
|
import
|
|
../driver,
|
|
../../common/databases/dburl,
|
|
../../common/databases/db_sqlite,
|
|
./sqlite_driver,
|
|
./sqlite_driver/migrations as archive_driver_sqlite_migrations,
|
|
./queue_driver
|
|
|
|
export
|
|
sqlite_driver,
|
|
queue_driver
|
|
|
|
when defined(postgres):
|
|
import ./postgres_driver ## This import adds dependency with an external libpq library
|
|
export postgres_driver
|
|
|
|
proc new*(T: type ArchiveDriver,
|
|
url: string,
|
|
vacuum: bool,
|
|
migrate: bool,
|
|
maxNumConn: int,
|
|
onErrAction: OnErrHandler):
|
|
Result[T, string] =
|
|
## url - string that defines the database
|
|
## vacuum - if true, a cleanup operation will be applied to the database
|
|
## migrate - if true, the database schema will be updated
|
|
## maxNumConn - defines the maximum number of connections to handle simultaneously (Postgres)
|
|
## onErrAction - called if, e.g., the connection with db got lost
|
|
|
|
let dbUrlValidationRes = dburl.validateDbUrl(url)
|
|
if dbUrlValidationRes.isErr():
|
|
return err("DbUrl failure in ArchiveDriver.new: " &
|
|
dbUrlValidationRes.error)
|
|
|
|
let engineRes = dburl.getDbEngine(url)
|
|
if engineRes.isErr():
|
|
return err("error getting db engine in setupWakuArchiveDriver: " &
|
|
engineRes.error)
|
|
|
|
let engine = engineRes.get()
|
|
|
|
case engine
|
|
of "sqlite":
|
|
let pathRes = dburl.getDbPath(url)
|
|
if pathRes.isErr():
|
|
return err("error get path in setupWakuArchiveDriver: " & pathRes.error)
|
|
|
|
let dbRes = SqliteDatabase.new(pathRes.get())
|
|
if dbRes.isErr():
|
|
return err("error in setupWakuArchiveDriver: " & dbRes.error)
|
|
|
|
let db = dbRes.get()
|
|
|
|
# SQLite vacuum
|
|
let (pageSize, pageCount, freelistCount) = ? db.gatherSqlitePageStats()
|
|
debug "sqlite database page stats", pageSize = pageSize,
|
|
pages = pageCount,
|
|
freePages = freelistCount
|
|
|
|
if vacuum and (pageCount > 0 and freelistCount > 0):
|
|
? db.performSqliteVacuum()
|
|
|
|
# Database migration
|
|
if migrate:
|
|
? archive_driver_sqlite_migrations.migrate(db)
|
|
|
|
debug "setting up sqlite waku archive driver"
|
|
let res = SqliteDriver.new(db)
|
|
if res.isErr():
|
|
return err("failed to init sqlite archive driver: " & res.error)
|
|
|
|
return ok(res.get())
|
|
|
|
of "postgres":
|
|
when defined(postgres):
|
|
let res = PostgresDriver.new(dbUrl = url,
|
|
maxConnections = maxNumConn,
|
|
onErrAction = onErrAction)
|
|
if res.isErr():
|
|
return err("failed to init postgres archive driver: " & res.error)
|
|
|
|
let driver = res.get()
|
|
|
|
try:
|
|
# The table should exist beforehand.
|
|
let newTableRes = waitFor driver.createMessageTable()
|
|
if newTableRes.isErr():
|
|
return err("error creating table: " & newTableRes.error)
|
|
except CatchableError:
|
|
return err("exception creating table: " & getCurrentExceptionMsg())
|
|
|
|
return ok(driver)
|
|
|
|
else:
|
|
return err("Postgres has been configured but not been compiled. Check compiler definitions.")
|
|
|
|
else:
|
|
debug "setting up in-memory waku archive driver"
|
|
let driver = QueueDriver.new() # Defaults to a capacity of 25.000 messages
|
|
return ok(driver)
|
|
|