mirror of
https://github.com/logos-storage/nim-datastore.git
synced 2026-01-11 18:13:06 +00:00
90 lines
2.2 KiB
Nim
90 lines
2.2 KiB
Nim
import std/times
|
|
import std/options
|
|
|
|
import pkg/chronos
|
|
import pkg/questionable
|
|
import pkg/questionable/results
|
|
import pkg/sqlite3_abi
|
|
from pkg/stew/results as stewResults import isErr
|
|
import pkg/upraises
|
|
|
|
import std/sequtils
|
|
import ../datastore
|
|
import ./threads/backend
|
|
import ./threads/sqlbackend
|
|
|
|
import pkg/chronos
|
|
import pkg/taskpools
|
|
|
|
import ./threads/sqlbackend
|
|
import ./threads/threadproxyds
|
|
import ./datastore
|
|
|
|
export datastore, Taskpool
|
|
|
|
push: {.upraises: [].}
|
|
|
|
type
|
|
SQLiteDatastore* = ref object of Datastore
|
|
db: SQLiteBackend[KeyId, DataBuffer]
|
|
|
|
proc path*(self: SQLiteDatastore): string =
|
|
self.db.path()
|
|
|
|
proc readOnly*(self: SQLiteDatastore): bool =
|
|
self.db.readOnly()
|
|
|
|
method has*(self: SQLiteDatastore,
|
|
key: Key): Future[?!bool] {.async.} =
|
|
return self.db.has(KeyId.new key.id())
|
|
|
|
method delete*(self: SQLiteDatastore,
|
|
key: Key): Future[?!void] {.async.} =
|
|
return self.db.delete(KeyId.new key.id())
|
|
|
|
method delete*(self: SQLiteDatastore,
|
|
keys: seq[Key]): Future[?!void] {.async.} =
|
|
let dkeys = keys.mapIt(KeyId.new it.id())
|
|
return self.db.delete(dkeys)
|
|
|
|
method get*(self: SQLiteDatastore,
|
|
key: Key): Future[?!seq[byte]] {.async.} =
|
|
self.db.get(KeyId.new key.id()).map() do(d: DataBuffer) -> seq[byte]:
|
|
d.toSeq()
|
|
|
|
method put*(self: SQLiteDatastore,
|
|
key: Key,
|
|
data: seq[byte]): Future[?!void] {.async.} =
|
|
self.db.put(KeyId.new key.id(), DataBuffer.new data)
|
|
|
|
method put*(self: SQLiteDatastore,
|
|
batch: seq[BatchEntry]): Future[?!void] {.async.} =
|
|
var dbatch: seq[tuple[key: KeyId, data: DataBuffer]]
|
|
for entry in batch:
|
|
dbatch.add((KeyId.new entry.key.id(), DataBuffer.new entry.data))
|
|
self.db.put(dbatch)
|
|
|
|
method close*(self: SQLiteDatastore): Future[?!void] {.async.} =
|
|
self.db.close()
|
|
|
|
method queryIter*(self: SQLiteDatastore,
|
|
query: Query
|
|
): ?!(iterator(): ?!QueryResponse) =
|
|
|
|
|
|
proc new*(
|
|
T: type SQLiteDatastore,
|
|
path: string,
|
|
readOnly = false): ?!SQLiteDatastore =
|
|
|
|
success SQLiteDatastore(
|
|
db: ? newSQLiteBackend[KeyId, DataBuffer](path, readOnly))
|
|
|
|
proc new*(
|
|
T: type SQLiteDatastore,
|
|
db: SQLiteBackend[KeyId, DataBuffer]): ?!T =
|
|
|
|
success T(
|
|
db: db,
|
|
readOnly: db.readOnly)
|