From eb1bf2377be15deb593a3e2426558d92c1973107 Mon Sep 17 00:00:00 2001 From: zeripath Date: Fri, 28 Aug 2020 07:55:12 +0100 Subject: [PATCH] Set context for running CreateArchive in to that of the request (#12555) Set the context for CreateArchive to that of the request to ensure that archives are only built for as long as a request is requesting them Fix #11551 Signed-off-by: Andrew Thornton Co-authored-by: Lauris BH --- modules/git/commit_archive.go | 5 +++-- routers/repo/repo.go | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/modules/git/commit_archive.go b/modules/git/commit_archive.go index c7d1d06c4680..d075ba09115f 100644 --- a/modules/git/commit_archive.go +++ b/modules/git/commit_archive.go @@ -6,6 +6,7 @@ package git import ( + "context" "fmt" "path/filepath" "strings" @@ -39,7 +40,7 @@ type CreateArchiveOpts struct { } // CreateArchive create archive content to the target path -func (c *Commit) CreateArchive(target string, opts CreateArchiveOpts) error { +func (c *Commit) CreateArchive(ctx context.Context, target string, opts CreateArchiveOpts) error { if opts.Format.String() == "unknown" { return fmt.Errorf("unknown format: %v", opts.Format) } @@ -58,6 +59,6 @@ func (c *Commit) CreateArchive(target string, opts CreateArchiveOpts) error { c.ID.String(), ) - _, err := NewCommand(args...).RunInDir(c.repo.Path) + _, err := NewCommandContext(ctx, args...).RunInDir(c.repo.Path) return err } diff --git a/routers/repo/repo.go b/routers/repo/repo.go index 71df2d0cb75b..5fc081a6f615 100644 --- a/routers/repo/repo.go +++ b/routers/repo/repo.go @@ -524,7 +524,7 @@ func Download(ctx *context.Context) { archivePath = path.Join(archivePath, base.ShortSha(commit.ID.String())+ext) if !com.IsFile(archivePath) { - if err := commit.CreateArchive(archivePath, git.CreateArchiveOpts{ + if err := commit.CreateArchive(ctx.Req.Context(), archivePath, git.CreateArchiveOpts{ Format: archiveType, Prefix: setting.Repository.PrefixArchiveFiles, }); err != nil {