nim-datastore/datastore/threadbackend.nim

95 lines
1.9 KiB
Nim
Raw Normal View History

2023-08-24 16:02:13 -07:00
2023-08-24 17:33:32 -07:00
import pkg/chronos/threadsync
import stew/results
2023-08-24 16:02:13 -07:00
import pkg/upraises
2023-08-24 16:18:31 -07:00
import pkg/taskpools
2023-08-24 16:02:13 -07:00
import ./key
import ./query
import ./datastore
import ./databuffer
2023-08-24 16:29:25 -07:00
import fsds
2023-08-24 16:02:13 -07:00
export key, query
push: {.upraises: [].}
type
2023-08-24 16:29:25 -07:00
ThreadBackendKind* {.pure.} = enum
FSBackend
SQliteBackend
ThreadBackend* = object
case kind*: ThreadBackendKind
of FSBackend:
root: StringBuffer
depth: int
caseSensitive: bool
ignoreProtected: bool
of SQliteBackend:
discard
2023-08-24 16:18:31 -07:00
2023-08-24 17:33:32 -07:00
ThreadDatastore* = ptr object
2023-08-24 16:18:31 -07:00
tp: Taskpool
2023-08-24 16:32:57 -07:00
var backendDatastore {.threadvar.}: Datastore
2023-08-24 16:18:31 -07:00
2023-08-24 16:29:25 -07:00
proc startupDatastore(backend: ThreadBackend): bool =
## starts up a FS instance on a give thread
case backend.kind:
of FSBackend:
let res = FSDatastore.new(
root = backend.root.toString(),
depth = backend.depth,
caseSensitive = backend.caseSensitive,
ignoreProtected = backend.ignoreProtected)
if res.isOk:
2023-08-24 16:32:57 -07:00
backendDatastore = res.get()
2023-08-24 16:29:25 -07:00
else:
discard
2023-08-24 16:02:13 -07:00
proc get*(
self: ThreadDatastore,
2023-08-24 17:33:32 -07:00
signal: ThreadSignalPtr,
2023-08-24 16:02:13 -07:00
key: KeyBuffer
2023-08-24 17:33:32 -07:00
): Result[DataBuffer, CatchableErrorBuffer] =
2023-08-24 16:02:13 -07:00
2023-08-24 17:33:32 -07:00
return ok(DataBuffer.new())
2023-08-24 16:02:13 -07:00
proc put*(
self: ThreadDatastore,
2023-08-24 17:33:32 -07:00
signal: ThreadSignalPtr,
2023-08-24 16:02:13 -07:00
key: KeyBuffer,
data: DataBuffer
2023-08-24 17:33:32 -07:00
): Result[void, CatchableErrorBuffer] =
2023-08-24 16:02:13 -07:00
2023-08-24 17:33:32 -07:00
return ok()
2023-08-24 16:02:13 -07:00
proc close*(
2023-08-24 17:33:32 -07:00
self: ThreadDatastore,
signal: ThreadSignalPtr,
): Result[void, CatchableErrorBuffer] =
try:
self[].tp.shutdown()
return ok()
except Exception as exc:
return err(exc.toBuffer())
2023-08-24 16:02:13 -07:00
func new*[S: ref Datastore](
T: typedesc[ThreadDatastore],
2023-08-24 17:33:32 -07:00
signal: ThreadSignalPtr,
2023-08-24 16:32:57 -07:00
backend: ThreadBackend,
2023-08-24 17:33:32 -07:00
): Result[ThreadDatastore, CatchableErrorBuffer] =
2023-08-24 16:02:13 -07:00
var self = T()
2023-08-24 16:18:31 -07:00
self.tp = Taskpool.new(num_threads = 1) ##\
## Default to one thread, multiple threads \
## will require more work
let pending = self.tp.spawn startupDatastore(backend)
sync pending
2023-08-24 16:02:13 -07:00
2023-08-24 17:33:32 -07:00
ok self