feat: accelerate garbage collection

Author: singchia

go.mod

@@ -2,7 +2,4 @@ module github.com/singchia/yafsm
 
 go 1.15
 
-require (
-	github.com/singchia/go-hammer v0.0.0-20211110114452-df62bc907676
-	gonum.org/v1/gonum v0.9.3
-)
+require github.com/jumboframes/armorigo v0.1.0

go.sum

@@ -72,14 +72,18 @@ func (pq *PrioQueue) Available() int {
 }
 
 func (pq *PrioQueue) PrioPush(prio int, data interface{}) error {
+	pq.mutex.RLock()
+	if !pq.ok {
+		pq.mutex.RUnlock()
+		return errors.New("queue closed")
+	}
 	select {
 	case pq.ch <- struct{}{}:
 	default:
 		return errors.New("queue full")
 	}
 
 	queue := (*prioQueue)(nil)
-	pq.mutex.RLock()
 	for elem := pq.queues.Front(); elem != nil; elem = elem.Next() {
 		value, _ := elem.Value.(*prioQueue)
 		if value.prio == prio {
@@ -203,7 +207,6 @@ func (pq *PrioQueue) PopSync() interface{} {
 		}
 		pq.mutex.RUnlock()
 	}
-	return nil
 }
 
 func (pq *PrioQueue) Close() {

pkg/prioqueue/prioqueue.go

@@ -98,7 +98,11 @@ func (fsm *FSM) Close() {
 	for k, _ := range fsm.states {
 		delete(fsm.states, k)
 	}
-	for k, _ := range fsm.events {
+	for k, v := range fsm.events {
+		for elem := v.Front(); elem != nil; elem = elem.Next() {
+			event := elem.Value.(*Event)
+			event.handlers, event.From, event.To = nil, nil, nil
+		}
 		delete(fsm.events, k)
 	}
 	fsm.pq.Close()
@@ -108,10 +112,16 @@ func (fsm *FSM) emit(ctx context.Context) {
 	for {
 		select {
 		case <-ctx.Done():
+			fsm.pq = nil
+			fsm.events = nil
+			fsm.states = nil
 			return
 
 		default:
 			data := fsm.pq.PopSync()
+			if data == nil {
+				continue
+			}
 			switch ec := data.(type) {
 			case *eventchan:
 				et := (*Event)(nil)