fixup! use dedicated channels for querier subscribe and closing conns

coadler · coadler · commit 036094fbf27a · 2023-09-20T15:38:08.000Z
diff --git a/enterprise/tailnet/pgcoord.go b/enterprise/tailnet/pgcoord.go
@@ -267,6 +267,7 @@ func (c *pgCoord) ServeClient(conn net.Conn, id uuid.UUID, agent uuid.UUID) erro
 	if err := c.addSubscription(cIO, agent); err != nil {
 		return err
 	}
+	defer func() { _ = c.removeSubscription(cIO, agent) }()
 
 	<-cIO.ctx.Done()
 	return nil

Original file line number	Diff line number	Diff line change
`@@ -267,6 +267,7 @@ func (c *pgCoord) ServeClient(conn net.Conn, id uuid.UUID, agent uuid.UUID) erro`
`267`	`267`	`if err := c.addSubscription(cIO, agent); err != nil {`
`268`	`268`	`return err`
`269`	`269`	`}`
	`270`	`+ defer func() { _ = c.removeSubscription(cIO, agent) }()`
`270`	`271`
`271`	`272`	`<-cIO.ctx.Done()`
`272`	`273`	`return nil`