fmsg: add to wg prior to enqueue

Adding after channel write is racy.

Signed-off-by: Ophestra Umiker <cat@ophivana.moe>

1 files changed, 3 insertions, 2 deletions

diff --git a/internal/fmsg/defer.go b/internal/fmsg/defer.go
index 0f3f5ad5..0161e27e 100644
--- a/internal/fmsg/defer.go
+++ b/internal/fmsg/defer.go
@@ -33,16 +33,17 @@ func dequeue() {
 // queue submits ops to msgbuf but drops messages
 // when the buffer is full and dequeue is withholding
 func queue(op dOp) {
+	queueSync.Add(1)
+
 	select {
 	case msgbuf <- op:
-		queueSync.Add(1)
 	default:
 		// send the op anyway if not withholding
 		// as dequeue will get to it eventually
 		if !wstate.Load() {
-			queueSync.Add(1)
 			msgbuf <- op
 		} else {
+			queueSync.Done()
 			// increment dropped message count
 			dropped.Add(1)
 		}