fix: propagate context cancellation after Wait to prevent incomplete artifacts

Signed-off-by: Zhao Chen <winters.zc@antgroup.com>

Author: aftersnow

pkg/backend/fetch.go

@@ -155,6 +155,9 @@ func (b *backend) Fetch(ctx context.Context, target string, cfg *config.Fetch) e
 	}
 
 	_ = g.Wait()
+	if ctx.Err() != nil {
+		return fmt.Errorf("fetch cancelled: %w", ctx.Err())
+	}
 	if err := errors.Join(errs...); err != nil {
 		return err
 	}

pkg/backend/fetch_by_d7y.go

@@ -155,6 +155,9 @@ func (b *backend) fetchByDragonfly(ctx context.Context, target string, cfg *conf
 	}
 
 	_ = g.Wait()
+	if ctx.Err() != nil {
+		return fmt.Errorf("fetch cancelled: %w", ctx.Err())
+	}
 	if err := errors.Join(errs...); err != nil {
 		return err
 	}

pkg/backend/processor/base.go

@@ -191,6 +191,10 @@ func (b *base) Process(ctx context.Context, builder build.Builder, workDir strin
 
 	_ = eg.Wait()
 
+	if ctx.Err() != nil {
+		return nil, fmt.Errorf("processing cancelled: %w", ctx.Err())
+	}
+
 	if err := errors.Join(errs...); err != nil {
 		return nil, err
 	}

pkg/backend/pull.go

@@ -150,6 +150,9 @@ func (b *backend) Pull(ctx context.Context, target string, cfg *config.Pull) err
 	}
 
 	_ = g.Wait()
+	if ctx.Err() != nil {
+		return fmt.Errorf("pull cancelled: %w", ctx.Err())
+	}
 	if len(errs) > 0 {
 		return fmt.Errorf("failed to pull blob to local: %w", errors.Join(errs...))
 	}

pkg/backend/pull_by_d7y.go

@@ -131,6 +131,9 @@ func (b *backend) pullByDragonfly(ctx context.Context, target string, cfg *confi
 	}
 
 	_ = g.Wait()
+	if ctx.Err() != nil {
+		return fmt.Errorf("pull cancelled: %w", ctx.Err())
+	}
 	if err := errors.Join(errs...); err != nil {
 		return err
 	}

pkg/backend/push.go

@@ -120,6 +120,9 @@ func (b *backend) Push(ctx context.Context, target string, cfg *config.Push) err
 	}
 
 	g.Wait()
+	if ctx.Err() != nil {
+		return fmt.Errorf("push cancelled: %w", ctx.Err())
+	}
 	if len(errs) > 0 {
 		return fmt.Errorf("failed to push blob to remote: %w", errors.Join(errs...))
 	}