From 4a5afcb09243eaff657ba7995612f087244fe1d5 Mon Sep 17 00:00:00 2001 From: Mathias Fredriksson Date: Fri, 7 Oct 2022 20:56:14 +0300 Subject: [PATCH] chore: Rename context in `cli/agent` Rename context from common `ctx` to `retryCtx` to avoid later re-use. Also kind of a bug-fix since client post was using `cmd.Context()`. --- cli/agent.go | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/cli/agent.go b/cli/agent.go index ff127325be2a5..de1bd87352aea 100644 --- a/cli/agent.go +++ b/cli/agent.go @@ -168,15 +168,15 @@ func workspaceAgent() *cobra.Command { } } - ctx, cancelFunc := context.WithTimeout(cmd.Context(), time.Hour) - defer cancelFunc() - for retry.New(100*time.Millisecond, 5*time.Second).Wait(ctx) { - err := client.PostWorkspaceAgentVersion(cmd.Context(), version) + retryCtx, cancelRetry := context.WithTimeout(cmd.Context(), time.Hour) + defer cancelRetry() + for retrier := retry.New(100*time.Millisecond, 5*time.Second); retrier.Wait(retryCtx); { + err := client.PostWorkspaceAgentVersion(retryCtx, version) if err != nil { - logger.Warn(cmd.Context(), "post agent version: %w", slog.Error(err), slog.F("version", version)) + logger.Warn(retryCtx, "post agent version: %w", slog.Error(err), slog.F("version", version)) continue } - logger.Info(ctx, "updated agent version", slog.F("version", version)) + logger.Info(retryCtx, "updated agent version", slog.F("version", version)) break }