From 386e4e25f3243f9ba02ea1e23b3aad235f9d0f8b Mon Sep 17 00:00:00 2001 From: Pavel Sviderski Date: Tue, 3 Dec 2024 18:33:48 +1000 Subject: [PATCH] fix context in PullImage --- internal/machine/docker/server.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/internal/machine/docker/server.go b/internal/machine/docker/server.go index 1133ebee..6807b445 100644 --- a/internal/machine/docker/server.go +++ b/internal/machine/docker/server.go @@ -81,9 +81,9 @@ func (s *Server) StartContainer(ctx context.Context, req *pb.StartContainerReque return &emptypb.Empty{}, nil } -func (s *Server) PullImage( - ctx context.Context, req *pb.PullImageRequest, stream grpc.ServerStreamingServer[pb.JSONMessage], -) error { +func (s *Server) PullImage(req *pb.PullImageRequest, stream grpc.ServerStreamingServer[pb.JSONMessage]) error { + ctx := stream.Context() + var opts image.PullOptions if len(req.Options) > 0 { if err := json.Unmarshal(req.Options, &opts); err != nil {