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

add manual mode

Browse Source
This commit is contained in:
Jaremy Creechley 2023-09-14 15:03:50 -07:00
parent 7974dea0aa
commit 185f1248e2
No known key found for this signature in database
GPG Key ID: 4E66FB67B21D3300
2 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
datastore/threads/sharedptr.nim
View File

@ -47,7 +47,7 @@ proc echoed*(vals: varargs[string, `$`]) =
type
SharedPtr*[T] = object
## Shared ownership reference counting pointer.
container*: ptr tuple[value: T, cnt: int]
container*: ptr tuple[value: T, cnt: int, manual: bool]
proc incr*[T](a: var SharedPtr[T]) =
if a.container != nil:
@ -75,19 +75,19 @@ proc release*[T](x: var SharedPtr[T]) =
x.container = nil
proc `=destroy`*[T](x: var SharedPtr[T]) =
if x.container != nil:
if x.container != nil and not x.container.manual:
echoed "SharedPtr: destroy: ", x.container.pointer.repr, " cnt: ", x.container.cnt, " tp: ", $(typeof(T))
decr(x)
decr(x)
proc `=dup`*[T](src: SharedPtr[T]): SharedPtr[T] =
if src.container != nil:
if src.container != nil and not src.container.manual:
echoed "SharedPtr: dup: ", src.container.pointer.repr, " cnt: ", src.container.cnt, " tp: ", $(typeof(T))
discard atomicAddFetch(src.cnt, 1, ATOMIC_RELAXED)
result.container = src.container
result.cnt = src.cnt
proc `=copy`*[T](dest: var SharedPtr[T], src: SharedPtr[T]) =
if src.container != nil:
if src.container != nil and not src.container.manual:
# echo "SharedPtr: copy: ", src.container.pointer.repr
discard atomicAddFetch(addr src.container.cnt, 1, ATOMIC_RELAXED)
echoed "SharedPtr: copy:src: ", src.container.pointer.repr, " cnt: ", src.container.cnt, " tp: ", $(typeof(T))
@ -96,21 +96,24 @@ proc `=copy`*[T](dest: var SharedPtr[T], src: SharedPtr[T]) =
`=destroy`(dest)
dest.container = src.container
proc newSharedPtr*[T](val: sink Isolated[T]): SharedPtr[T] {.nodestroy.} =
proc newSharedPtr*[T](val: sink Isolated[T], manualCount: Natural = 0): SharedPtr[T] {.nodestroy.} =
## Returns a shared pointer which shares,
## ownership of the object by reference counting.
result.container = cast[typeof(result.container)](allocShared(sizeof(result.container[])))
result.container.cnt = 1
result.container.cnt = max(1, manualCount)
result.container.manual = manualCount > 0
result.container.value = extract val
echoed "SharedPtr: alloc: ", result.container.pointer.repr, " cnt: ", result.container.cnt, " tp: ", $(typeof(T))
template newSharedPtr*[T](val: T): SharedPtr[T] =
newSharedPtr(isolate(val))
proc newSharedPtr*[T](t: typedesc[T]): SharedPtr[T] =
proc newSharedPtr*[T](t: typedesc[T], manualCount: Natural = 0): SharedPtr[T] =
## Returns a shared pointer. It is not initialized,
## ownership of the object by reference counting.
result.container = cast[typeof(result.container)](allocShared0(sizeof(result.container[])))
result.container.cnt = 1
result.container.cnt = max(1, manualCount)
result.container.manual = manualCount > 0
echoed "SharedPtr: alloc: ", result.container.pointer.repr, " cnt: ", result.container.cnt, " tp: ", $(typeof(T))
proc isNil*[T](p: SharedPtr[T]): bool {.inline.} =

4
tests/datastore/testsharedptr.nim
View File

@ -67,10 +67,10 @@ suite "Share buffer test":
check $a1 == "nil\"\""
check a1.isNil
# check $a2 == "(value: 0, cnt: 2)"
check split($(a2),'"')[1] == "(value: 0, cnt: 2)"
check split($(a2),'"')[1] == "(value: 0, cnt: 2, manual: false)"
check not a2.isNil
check a2[] == 0
check split($(a3),'"')[1] == "(value: 0, cnt: 2)"
check split($(a3),'"')[1] == "(value: 0, cnt: 2, manual: false)"
check not a3.isNil
check a3[] == 0