fix: race condition on async

async.go

@@ -10,6 +10,7 @@ type Async struct {
 	data     any
 	done     *flag
 	pending  chan bool
+	notify   *flag
 	listener func(as *Async)
 	err      error
 }
@@ -19,6 +20,7 @@ func newAsync(hdl Handler, cmd Command) *Async {
 		hdl:     hdl,
 		cmd:     cmd,
 		done:    newFlag(),
+		notify:  newFlag(),
 		pending: make(chan bool, 1),
 	}
 }
@@ -42,34 +44,40 @@ func (as *Async) await() {
 
 func (as *Async) notifyDone() {
 	if as.done.enable() {
-		as.pending <- true
 		as.notifyListener()
+		as.pending <- true
 	}
 }
 
 func (as *Async) fail(err error) {
+	as.Lock()
 	as.err = err
 	as.notifyDone()
+	as.Unlock()
 }
 
 func (as *Async) success(data any) {
+	as.Lock()
 	as.data = data
 	as.notifyDone()
+	as.Unlock()
 }
 
 func (as *Async) setListener(listener func(as *Async)) {
 	as.Lock()
-	as.listener = listener
-	as.Unlock()
 	if as.done.enabled() {
-		as.notifyListener()
+		listener(as)
+		return
+	}
+	if as.notify.enable() {
+		as.listener = listener
 	}
+	as.Unlock()
+
 }
 
 func (as *Async) notifyListener() {
-	as.Lock()
-	if as.listener != nil {
+	if as.notify.disable() {
 		as.listener(as)
 	}
-	as.Unlock()
 }

async_list.go

@@ -1,6 +1,9 @@
 package command
 
-import "errors"
+import (
+	"errors"
+	"fmt"
+)
 
 // Async is the struct returned from async commands.
 type AsyncList struct {
@@ -48,15 +51,16 @@ func (asl *AsyncList) AwaitIterator() (<-chan AsyncResult, error) {
 	return results, nil
 }
 
-func (asl *AsyncList) generateListener(i int, asyncResults chan<- AsyncResult, processed *counter, total uint32) func(as *Async) {
+func (asl *AsyncList) generateListener(i int, results chan<- AsyncResult, processed *counter, total uint32) func(as *Async) {
 	return func(as *Async) {
-		asyncResults <- AsyncResult{
+		results <- AsyncResult{
 			Index: i,
 			Data:  as.data,
 			Err:   as.err,
 		}
 		if processed.increment() == total {
-			close(asyncResults)
+			close(results)
+			fmt.Println("CHANNEL CLOSED")
 		}
 	}
 }