mirror of
https://github.com/logos-storage/nim-datastore.git
synced 2026-01-05 07:03:12 +00:00
69 lines
2.2 KiB
Nim
69 lines
2.2 KiB
Nim
import std/atomics
|
|
import std/options
|
|
|
|
import pkg/questionable/results
|
|
import pkg/results
|
|
|
|
import ../types
|
|
import ../query
|
|
import ../key
|
|
|
|
import ./databuffer
|
|
|
|
type
|
|
ErrorEnum* {.pure.} = enum
|
|
DatastoreErr,
|
|
DatastoreKeyNotFoundErr,
|
|
QueryEndedErr,
|
|
CatchableErr
|
|
|
|
ThreadTypes* = void | bool | SomeInteger | DataBuffer | tuple | Atomic
|
|
ThreadResErr* = (ErrorEnum, DataBuffer)
|
|
ThreadQueryRes* = (DataBuffer, DataBuffer)
|
|
ThreadResult*[T: ThreadTypes] = Result[T, ThreadResErr]
|
|
|
|
DbKey* = object
|
|
data: DataBuffer
|
|
DbVal* = object
|
|
data: DataBuffer
|
|
|
|
proc toDb*(key: Key): DbKey {.inline, raises: [].} =
|
|
let id: string = key.id()
|
|
let db = DataBuffer.new(id.len()+1) # include room for null for cstring compat
|
|
db.setData(id)
|
|
DbKey(data: db)
|
|
|
|
proc toKey*(key: DbKey): Key {.inline, raises: [].} =
|
|
Key.init(key.data).expect("expected valid key here for but got `" & $key.data & "`")
|
|
|
|
proc toDb*(value: sink seq[byte]): DbVal {.inline, raises: [].} =
|
|
DbVal(data: DataBuffer.new(value))
|
|
|
|
proc toValue*(value: DbVal): seq[byte] {.inline, raises: [].} =
|
|
value.data.toSeq()
|
|
|
|
template toOpenArray*(x: DbKey): openArray[char] =
|
|
x.data.toOpenArray(char)
|
|
template toOpenArray*(x: DbVal): openArray[byte] =
|
|
x.data.toOpenArray(byte)
|
|
|
|
converter toThreadErr*(e: ref CatchableError): ThreadResErr {.inline, raises: [].} =
|
|
if e of DatastoreKeyNotFound: (ErrorEnum.DatastoreKeyNotFoundErr, DataBuffer.new(e.msg))
|
|
elif e of QueryEndedError: (ErrorEnum.QueryEndedErr, DataBuffer.new(e.msg))
|
|
elif e of DatastoreError: (DatastoreErr, DataBuffer.new(e.msg))
|
|
elif e of CatchableError: (CatchableErr, DataBuffer.new(e.msg))
|
|
else: raise (ref Defect)(msg: e.msg)
|
|
|
|
converter toExc*(e: ThreadResErr): ref CatchableError =
|
|
case e[0]:
|
|
of ErrorEnum.DatastoreKeyNotFoundErr: (ref DatastoreKeyNotFound)(msg: $e[1])
|
|
of ErrorEnum.QueryEndedErr: (ref QueryEndedError)(msg: $e[1])
|
|
of ErrorEnum.DatastoreErr: (ref DatastoreError)(msg: $e[1])
|
|
of ErrorEnum.CatchableErr: (ref CatchableError)(msg: $e[1])
|
|
|
|
converter toQueryResponse*(r: ThreadQueryRes): QueryResponse =
|
|
if not r[0].isNil and r[0].len > 0 and key =? Key.init($r[0]):
|
|
(key.some, @(r[1]))
|
|
else:
|
|
(Key.none, EmptyBytes)
|