From a74a2c7819af9c14707067f43d8162193f49f478 Mon Sep 17 00:00:00 2001
From: Rob Watson <rob@netflux.io>
Date: Sun, 13 Apr 2025 05:23:16 +0200
Subject: [PATCH] fixup! refactor(container): restart handling

---
 internal/container/container.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/internal/container/container.go b/internal/container/container.go
index 698230e..0e240a7 100644
--- a/internal/container/container.go
+++ b/internal/container/container.go
@@ -243,6 +243,7 @@ func (a *Client) RunContainer(ctx context.Context, params RunContainerParams) (<
 
 		a.runContainerLoop(
 			ctx,
+			cancel,
 			createResp.ID,
 			params.ContainerConfig.Image,
 			params.NetworkCountConfig,
@@ -329,6 +330,7 @@ func (a *Client) pullImageIfNeeded(ctx context.Context, imageName string, contai
 // when the container exits.
 func (a *Client) runContainerLoop(
 	ctx context.Context,
+	cancel context.CancelFunc,
 	containerID string,
 	imageName string,
 	networkCountConfig NetworkCountConfig,
@@ -336,6 +338,8 @@ func (a *Client) runContainerLoop(
 	stateC chan<- domain.Container,
 	errC chan<- error,
 ) {
+	defer cancel()
+
 	type containerWaitResponse struct {
 		container.WaitResponse