Dev to Main (#28)

* fix go mod tidy * fix go mod tidy (#21) * Dev to Main (#23) * fix go mod tidy * fix go mod tidy (#21) * fix iterator * lazy cow and mutations: * removed old chan * revert decrement iter * fix mutate chn bug * trying fix race
absolutelightning · May 28, 2024 · 8e9b3e7 · 8e9b3e7
1 parent 6a1f106
commit 8e9b3e7
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 4 deletions.
diff --git a/node_16.go b/node_16.go
@@ -225,7 +225,8 @@ func (n *Node16[T]) getRefCount() int32 {
 }
 
 func (n *Node16[T]) processLazyRef() {
-	atomic.AddInt32(&n.refCount, n.lazyRefCount)
+	lazyRefCount := atomic.LoadInt32(&n.lazyRefCount)
+	atomic.AddInt32(&n.refCount, lazyRefCount)
 	for i := 0; i < 16; i++ {
 		if n.children[i] != nil {
 			n.children[i].incrementLazyRefCount(n.lazyRefCount)

diff --git a/node_256.go b/node_256.go
@@ -219,7 +219,8 @@ func (n *Node256[T]) getRefCount() int32 {
 }
 
 func (n *Node256[T]) processLazyRef() {
-	atomic.AddInt32(&n.refCount, n.lazyRefCount)
+	lazyRefCount := atomic.LoadInt32(&n.lazyRefCount)
+	atomic.AddInt32(&n.refCount, lazyRefCount)
 	for i := 0; i < 256; i++ {
 		if n.children[i] != nil {
 			n.children[i].incrementLazyRefCount(n.lazyRefCount)

diff --git a/node_4.go b/node_4.go
@@ -225,7 +225,8 @@ func (n *Node4[T]) getRefCount() int32 {
 }
 
 func (n *Node4[T]) processLazyRef() {
-	atomic.AddInt32(&n.refCount, n.lazyRefCount)
+	lazyRefCount := atomic.LoadInt32(&n.lazyRefCount)
+	atomic.AddInt32(&n.refCount, lazyRefCount)
 	for i := 0; i < 4; i++ {
 		if n.children[i] != nil {
 			n.children[i].incrementLazyRefCount(n.lazyRefCount)

diff --git a/node_48.go b/node_48.go
@@ -232,7 +232,8 @@ func (n *Node48[T]) setRefCount(val int32) {
 }
 
 func (n *Node48[T]) processLazyRef() {
-	atomic.AddInt32(&n.refCount, n.lazyRefCount)
+	lazyRefCount := atomic.LoadInt32(&n.lazyRefCount)
+	atomic.AddInt32(&n.refCount, lazyRefCount)
 	for i := 0; i < 48; i++ {
 		if n.children[i] != nil {
 			n.children[i].incrementLazyRefCount(n.lazyRefCount)