fix(tailnet): track agent names for http debug

coadler · coadler · commit b3dedc2ab7cd · 2023-07-26T17:55:35.000Z
These were previously not passed through.
diff --git a/enterprise/tailnet/coordinator.go b/enterprise/tailnet/coordinator.go
@@ -158,7 +158,7 @@ func (c *haCoordinator) ServeClient(conn net.Conn, id, agentID uuid.UUID) error
 	defer cancel()
 	logger := c.clientLogger(id, agentID)
 
-	tc := agpl.NewTrackedConn(ctx, cancel, conn, id, logger, 0)
+	tc := agpl.NewTrackedConn(ctx, cancel, conn, id, logger, "", 0)
 	defer tc.Close()
 
 	c.addClient(id, tc)
@@ -301,7 +301,7 @@ func (c *haCoordinator) ServeAgent(conn net.Conn, id uuid.UUID, name string) err
 	}
 	// This uniquely identifies a connection that belongs to this goroutine.
 	unique := uuid.New()
-	tc := agpl.NewTrackedConn(ctx, cancel, conn, unique, logger, overwrites)
+	tc := agpl.NewTrackedConn(ctx, cancel, conn, unique, logger, name, overwrites)
 
 	// Publish all nodes on this instance that want to connect to this agent.
 	nodes := c.nodesSubscribedToAgent(id)
diff --git a/enterprise/tailnet/pgcoord.go b/enterprise/tailnet/pgcoord.go
@@ -131,12 +131,6 @@ func (c *pgCoord) ServeMultiAgent(id uuid.UUID) agpl.MultiAgentConn {
 	panic("not implemented") // TODO: Implement
 }
 
-func (*pgCoord) ServeHTTPDebug(w http.ResponseWriter, _ *http.Request) {
-	// TODO(spikecurtis) I'd like to hold off implementing this until after the rest of this is code reviewed.
-	w.WriteHeader(http.StatusOK)
-	_, _ = w.Write([]byte("Coder Enterprise PostgreSQL distributed tailnet coordinator"))
-}
-
 func (c *pgCoord) Node(id uuid.UUID) *agpl.Node {
 	// In production, we only ever get this request for an agent.
 	// We're going to directly query the database, since we would only have the agent mapping stored locally if we had
@@ -167,7 +161,7 @@ func (c *pgCoord) ServeClient(conn net.Conn, id uuid.UUID, agent uuid.UUID) erro
 				slog.Error(err))
 		}
 	}()
-	cIO := newConnIO(c.ctx, c.logger, c.bindings, conn, id, agent)
+	cIO := newConnIO(c.ctx, c.logger, c.bindings, conn, id, agent, "")
 	if err := sendCtx(c.ctx, c.newConnections, cIO); err != nil {
 		// can only be a context error, no need to log here.
 		return err
@@ -186,7 +180,7 @@ func (c *pgCoord) ServeAgent(conn net.Conn, id uuid.UUID, name string) error {
 		}
 	}()
 	logger := c.logger.With(slog.F("name", name))
-	cIO := newConnIO(c.ctx, logger, c.bindings, conn, uuid.Nil, id)
+	cIO := newConnIO(c.ctx, logger, c.bindings, conn, uuid.Nil, id, name)
 	if err := sendCtx(c.ctx, c.newConnections, cIO); err != nil {
 		// can only be a context error, no need to log here.
 		return err
@@ -217,8 +211,12 @@ type connIO struct {
 	bindings chan<- binding
 }
 
-func newConnIO(
-	pCtx context.Context, logger slog.Logger, bindings chan<- binding, conn net.Conn, client, agent uuid.UUID,
+func newConnIO(pCtx context.Context,
+	logger slog.Logger,
+	bindings chan<- binding,
+	conn net.Conn,
+	client, agent uuid.UUID,
+	name string,
 ) *connIO {
 	ctx, cancel := context.WithCancel(pCtx)
 	id := agent
@@ -235,7 +233,7 @@ func newConnIO(
 		client:   client,
 		agent:    agent,
 		decoder:  json.NewDecoder(conn),
-		updates:  agpl.NewTrackedConn(ctx, cancel, conn, id, logger, 0),
+		updates:  agpl.NewTrackedConn(ctx, cancel, conn, id, logger, name, 0),
 		bindings: bindings,
 	}
 	go c.recvLoop()
@@ -1291,3 +1289,9 @@ func (h *heartbeats) cleanup() {
 	}
 	h.logger.Debug(h.ctx, "cleaned up old coordinators")
 }
+
+func (*pgCoord) ServeHTTPDebug(w http.ResponseWriter, _ *http.Request) {
+	// TODO(spikecurtis) I'd like to hold off implementing this until after the rest of this is code reviewed.
+	w.WriteHeader(http.StatusOK)
+	_, _ = w.Write([]byte("Coder Enterprise PostgreSQL distributed tailnet coordinator"))
+}
diff --git a/tailnet/coordinator.go b/tailnet/coordinator.go
@@ -262,7 +262,7 @@ func (c *coordinator) ServeClient(conn net.Conn, id, agentID uuid.UUID) error {
 	logger := c.core.clientLogger(id, agentID)
 	logger.Debug(ctx, "coordinating client")
 
-	tc := NewTrackedConn(ctx, cancel, conn, id, logger, 0)
+	tc := NewTrackedConn(ctx, cancel, conn, id, logger, "", 0)
 	defer tc.Close()
 
 	c.core.addClient(id, tc)
@@ -507,7 +507,7 @@ func (c *core) initAndTrackAgent(ctx context.Context, cancel func(), conn net.Co
 		overwrites = oldAgentSocket.Overwrites() + 1
 		_ = oldAgentSocket.Close()
 	}
-	tc := NewTrackedConn(ctx, cancel, conn, unique, logger, overwrites)
+	tc := NewTrackedConn(ctx, cancel, conn, unique, logger, name, overwrites)
 	c.agentNameCache.Add(id, name)
 
 	sockets, ok := c.agentToConnectionSockets[id]
diff --git a/tailnet/trackedconn.go b/tailnet/trackedconn.go
@@ -35,7 +35,7 @@ type TrackedConn struct {
 	overwrites int64
 }
 
-func NewTrackedConn(ctx context.Context, cancel func(), conn net.Conn, id uuid.UUID, logger slog.Logger, overwrites int64) *TrackedConn {
+func NewTrackedConn(ctx context.Context, cancel func(), conn net.Conn, id uuid.UUID, logger slog.Logger, name string, overwrites int64) *TrackedConn {
 	// buffer updates so they don't block, since we hold the
 	// coordinator mutex while queuing.  Node updates don't
 	// come quickly, so 512 should be plenty for all but

Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,7 @@ type TrackedConn struct {`
`35`	`35`	`overwrites int64`
`36`	`36`	`}`
`37`	`37`
`38`		`-func NewTrackedConn(ctx context.Context, cancel func(), conn net.Conn, id uuid.UUID, logger slog.Logger, overwrites int64) *TrackedConn {`
	`38`	`+func NewTrackedConn(ctx context.Context, cancel func(), conn net.Conn, id uuid.UUID, logger slog.Logger, name string, overwrites int64) *TrackedConn {`
`39`	`39`	`// buffer updates so they don't block, since we hold the`
`40`	`40`	`// coordinator mutex while queuing. Node updates don't`
`41`	`41`	`// come quickly, so 512 should be plenty for all but`