dimgigov
6bfc5b3a3c
CI / test (push) Has been cancelled
CI / verify (push) Has been cancelled
Clients CI / build-server (push) Has been cancelled
Clients CI / test-python (push) Has been cancelled
Clients CI / test-javascript (push) Has been cancelled
Clients CI / test-nim (push) Has been cancelled
Clients CI / test-rust (push) Has been cancelled
- Pure Nim wire protocol client (async + sync, no C FFI) - Query Builder: fluent API, SQL generator, execution, transactions - Schema Builder: create/alter/drop tables, column operations - Connection pool with aging and timeout handling - ORM integration via compile-time DB_BARADB switch - Tests: test_open, test_query - Documentation: README.md, PLAN_BARADB.md
99 lines
2.6 KiB
Nim
99 lines
2.6 KiB
Nim
discard """
|
|
cmd: "nim c -d:reset $file"
|
|
"""
|
|
|
|
import std/asyncdispatch
|
|
import std/deques
|
|
import std/json
|
|
import std/tables
|
|
import std/unittest
|
|
import ../../src/allographer/query_builder/libs/surreal/surreal_lib
|
|
import ../../src/allographer/query_builder/libs/surreal/surreal_rdb
|
|
import ../../src/allographer/query_builder/log
|
|
import ../../src/allographer/query_builder/models/surreal/surreal_exec
|
|
import ../../src/allographer/query_builder/models/surreal/surreal_types
|
|
|
|
|
|
proc newTestSurreal(): SurrealConnections =
|
|
let pools = Connections(
|
|
conns: @[
|
|
Connection(
|
|
conn: SurrealConn(),
|
|
isBusy: false,
|
|
createdAt: 0,
|
|
)
|
|
],
|
|
timeout: 30,
|
|
waiters: initDeque[Future[void]](),
|
|
preparedCache: initTable[string, SurrealPreparedEntry](),
|
|
)
|
|
result = SurrealConnections(
|
|
log: LogSetting(
|
|
shouldDisplayLog: false,
|
|
shouldOutputLogFile: false,
|
|
logDir: "",
|
|
),
|
|
pools: pools,
|
|
)
|
|
|
|
|
|
suite "SurrealDB prepared statement":
|
|
test "dbFormatPrepared":
|
|
var args = newJArray()
|
|
args.add(%*"user:alice")
|
|
args.add(%*"2026-04-03T00:00:00Z")
|
|
args.add(newJNull())
|
|
let sql = dbFormatPrepared(
|
|
questionToDaller("""SELECT * FROM "user" WHERE "id" = ? AND "submit_at" >= ? AND "address" IS ?"""),
|
|
args
|
|
)
|
|
check sql ==
|
|
"""LET $a = user:alice; LET $b = <datetime>"2026-04-03T00:00:00Z"; LET $c = NONE; SELECT * FROM "user" WHERE "id" = $a AND "submit_at" >= $b AND "address" IS $c"""
|
|
|
|
|
|
test "prepare cache lifecycle":
|
|
let rdb = newTestSurreal()
|
|
let sql = """SELECT * FROM "user" WHERE "id" = ?"""
|
|
|
|
let stmt1 = rdb.prepare(sql)
|
|
check stmt1.nArgs == 1
|
|
check stmt1.entry.normalizedSql == questionToDaller(sql)
|
|
check stmt1.entry.refCount == 1
|
|
check rdb.pools.preparedCache.len == 1
|
|
|
|
let stmt2 = rdb.prepare(sql)
|
|
check stmt2.entry == stmt1.entry
|
|
check stmt1.entry.refCount == 2
|
|
|
|
waitFor stmt1.close()
|
|
check stmt1.isClosed
|
|
check stmt1.entry.refCount == 1
|
|
|
|
waitFor stmt2.close()
|
|
check stmt2.isClosed
|
|
check stmt2.entry.refCount == 0
|
|
|
|
let stmt4 = rdb.prepare(sql)
|
|
check rdb.pools.preparedCache.len == 1
|
|
waitFor rdb.flushStmt(stmt4)
|
|
check stmt4.isClosed
|
|
check rdb.pools.preparedCache.len == 0
|
|
|
|
let stmt3 = rdb.prepare(sql)
|
|
check stmt3.entry != stmt1.entry
|
|
waitFor stmt3.close()
|
|
waitFor rdb.clearStmtCache()
|
|
check rdb.pools.preparedCache.len == 0
|
|
|
|
|
|
test "withConn context":
|
|
let rdb = newTestSurreal()
|
|
|
|
waitFor rdb.withConn(
|
|
proc(ctx: SurrealPreparedContext): Future[void] {.async.} =
|
|
check ctx.owner == rdb
|
|
check ctx.connI == 0
|
|
)
|
|
|
|
check not rdb.pools.conns[0].isBusy
|