fixup! chore: refactor agent stats streaming

coadler · coadler · commit 3575be846869 · 2022-11-17T19:19:40.000-06:00
diff --git a/agent/agent_test.go b/agent/agent_test.go
@@ -73,7 +73,7 @@ func TestAgent(t *testing.T) {
 			var s *codersdk.AgentStats
 			require.Eventuallyf(t, func() bool {
 				var ok bool
-				s, ok = (<-stats)
+				s, ok = <-stats
 				return ok && s.NumConns > 0 && s.RxBytes > 0 && s.TxBytes > 0
 			}, testutil.WaitLong, testutil.IntervalFast,
 				"never saw stats: %+v", s,
@@ -102,7 +102,7 @@ func TestAgent(t *testing.T) {
 			var s *codersdk.AgentStats
 			require.Eventuallyf(t, func() bool {
 				var ok bool
-				s, ok = (<-stats)
+				s, ok = <-stats
 				return ok && s.NumConns > 0 && s.RxBytes > 0 && s.TxBytes > 0
 			}, testutil.WaitLong, testutil.IntervalFast,
 				"never saw stats: %+v", s,
diff --git a/codersdk/workspaceagents.go b/codersdk/workspaceagents.go
@@ -626,7 +626,9 @@ func (c *Client) AgentReportStats(
 			for r := retry.New(100*time.Millisecond, time.Minute); r.Wait(ctx); {
 				resp, err := sendStats(ctx)
 				if err != nil {
-					log.Error(ctx, "report stats", slog.Error(err))
+					if !xerrors.Is(err, context.Canceled) {
+						log.Error(ctx, "report stats", slog.Error(err))
+					}
 					continue
 				}
 
