Lock on close wait

kylecarbs · kylecarbs · commit 55f72083b09f · 2022-03-07T01:00:05.000Z
diff --git a/agent/agent.go b/agent/agent.go
@@ -280,7 +280,9 @@ func (s *server) run(ctx context.Context) {
 			s.run(ctx)
 			return
 		}
+		s.closeMutex.Lock()
 		s.connCloseWait.Add(1)
+		s.closeMutex.Unlock()
 		go s.handlePeerConn(ctx, conn)
 	}
 }

Original file line number	Diff line number	Diff line change
`@@ -280,7 +280,9 @@ func (s *server) run(ctx context.Context) {`
`280`	`280`	`s.run(ctx)`
`281`	`281`	`return`
`282`	`282`	`}`
	`283`	`+ s.closeMutex.Lock()`
`283`	`284`	`s.connCloseWait.Add(1)`
	`285`	`+ s.closeMutex.Unlock()`
`284`	`286`	`go s.handlePeerConn(ctx, conn)`
`285`	`287`	`}`
`286`	`288`	`}`