fixed a memory bug

alob-mtc · Mar 18, 2023 · 993274e · 993274e
1 parent 4ec4ed1
commit 993274e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/eventloop.go b/eventloop.go
@@ -34,6 +34,7 @@ type eventLoop struct {
 	size         uint64
 	signal       chan struct{}
 	keepAlive    bool
+	sync         sync.Mutex
 }
 
 func (e *eventLoop) Await(currentP *Promise) (interface{}, error) {
@@ -99,7 +100,9 @@ func (e *eventLoop) Main(fn func()) {
 
 func (e *eventLoop) awaitAll() {
 	for {
+		e.sync.Lock()
 		n := len(e.promiseQueue)
+		e.sync.Unlock()
 		for i := n - 1; i >= 0; i-- {
 			p := e.promiseQueue[i]
 			if p.handler {

diff --git a/promise.go b/promise.go
@@ -19,7 +19,9 @@ func (e *eventLoop) newPromise(rev <-chan interface{}, errChan chan error) *Prom
 		defer close(e.signal)
 	}
 	currentP := &Promise{id: atomic.AddUint64(&e.size, 1), rev: rev, errChan: errChan, done: make(chan struct{}), err: make(chan struct{})}
+	e.sync.Lock()
 	e.promiseQueue = append(e.promiseQueue, currentP)
+	e.sync.Unlock()
 	return currentP
 }