Function WaitForReinitLoop

codersdk/agentsdk/agentsdk.go:835–863

WaitForReinitLoop polls the /reinit SSE endpoint in a retry loop and forwards received reinitialization events to the returned channel. The channel is closed when ctx is canceled or the server returns 409 Conflict (indicating the workspace is not a prebuilt workspace or the claim build failed perman

(ctx context.Context, logger slog.Logger, client *Client)

// claim build failed permanently). The caller should select on both the
// channel and ctx.Done().
func WaitForReinitLoop(ctx context.Context, logger slog.Logger, client *Client) <-chan ReinitializationEvent {
	reinitEvents := make(chan ReinitializationEvent)

	go func() {
		defer close(reinitEvents)
		for retrier := retry.New(100*time.Millisecond, 10*time.Second); retrier.Wait(ctx); {
			logger.Debug(ctx, "waiting for agent reinitialization instructions")
			reinitEvent, err := client.WaitForReinit(ctx)
			if err != nil {
				var sdkErr *codersdk.Error
				if errors.As(err, &sdkErr) && sdkErr.StatusCode() == http.StatusConflict {
					logger.Info(ctx, "received terminal 409, stopping reinit polling",
						slog.Error(sdkErr))
					return
				}
				logger.Error(ctx, "failed to wait for agent reinitialization instructions", slog.Error(err))
				continue
			}
			retrier.Reset()
			select {
			case <-ctx.Done():
				return
			case reinitEvents <- *reinitEvent:
			}
		}
	}()

	return reinitEvents
}

func NewSSEAgentReinitTransmitter(logger slog.Logger, rw http.ResponseWriter, r *http.Request) *SSEAgentReinitTransmitter {
	return &SSEAgentReinitTransmitter{logger: logger, rw: rw, r: r}