logos-storage/nim-datastore
1
0
Fork 0
mirror of https://github.com/logos-storage/nim-datastore.git synced 2026-01-04 06:33:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

back to ref TaskCtx

Browse Source
This commit is contained in:
Jaremy Creechley 2023-09-18 14:05:21 -07:00
parent 5d259aca1d
commit 4ab54f97c9
No known key found for this signature in database
GPG Key ID: 4E66FB67B21D3300
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
datastore/threads/threadproxyds.nim
View File

@ -52,21 +52,23 @@ type
queryLock: AsyncLock # global query lock, this is only really \
# needed for the fsds, but it is expensive!
template addrOf*[T](ctx: ptr TaskCtx[T]): ptr TaskCtx[T] =
ctx
proc addrOf*[T](ctx: ref TaskCtx[T]): ptr TaskCtx[T] =
cast[ptr TaskCtx[T]](ctx)
proc new*[T](
ctx: typedesc[TaskCtx[T]],
ds: Datastore,
signal: ThreadSignalPtr,
): ptr TaskCtx[T] =
let res = cast[ptr TaskCtx[T]](allocShared0(sizeof(TaskCtx[T])))
res[].ds = unsafeAddr ds
): ref TaskCtx[T] =
# let res = cast[ptr TaskCtx[T]](allocShared0(sizeof(TaskCtx[T])))
let res = (ref TaskCtx[T])()
res.ds = unsafeAddr ds
res.signal = signal
res
template withLocks(
self: ThreadDatastore,
ctx: ptr TaskCtx,
ctx: ref TaskCtx,
key: ?Key = Key.none,
fut: Future[void],
body: untyped) =
@ -89,7 +91,7 @@ template withLocks(
template dispatchTask(
self: ThreadDatastore,
ctx: ptr TaskCtx,
ctx: ref TaskCtx,
key: ?Key = Key.none,
runTask: proc): untyped =
@ -110,6 +112,7 @@ template dispatchTask(
# but for now it'd at least be better to leak than possibly
# corrupt memory since it's easier to detect and fix leaks
warn "request was cancelled while thread task is running", exc = exc.msg
GC_ref(ctx)
ctx.cancelled.store(true, moAcquireRelease)
await ctx.signal.fire()
raise exc