test a tiny buffer for udpate channels to avoid stalling

Signed-off-by: Kristoffer Dalby <[email protected]>
kradalby · Nov 27, 2023 · 57f3dfc · 57f3dfc
1 parent c6fc3e3
commit 57f3dfc
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/hscontrol/poll.go b/hscontrol/poll.go
@@ -284,7 +284,11 @@ func (h *Headscale) handlePoll(
 	h.pollNetMapStreamWG.Add(1)
 	defer h.pollNetMapStreamWG.Done()
 
-	updateChan := make(chan types.StateUpdate)
+	// Use a buffered channel in case a node is not fully ready
+	// to receive a message to make sure we dont block the entire
+	// notifier.
+	// 12 is arbitrarily chosen.
+	updateChan := make(chan types.StateUpdate, 12)
 	defer closeChanWithLog(updateChan, node.Hostname, "updateChan")
 
 	// Register the node's update channel