Skip to content

Commit 68a9306

Browse files
TheFox0x7TheFox0x7
committed
add context to DataAsync
1 parent f4fb209 commit 68a9306

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

49 files changed

+110
-102
lines changed

‎modules/actions/workflows.go‎

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ package actions
55

66
import (
77
"bytes"
8+
"context"
89
"io"
910
"strings"
1011

@@ -69,8 +70,8 @@ func ListWorkflows(commit *git.Commit) (git.Entries, error) {
6970
return ret, nil
7071
}
7172

72-
func GetContentFromEntry(entry *git.TreeEntry) ([]byte, error) {
73-
f, err := entry.Blob().DataAsync()
73+
func GetContentFromEntry(ctx context.Context, entry *git.TreeEntry) ([]byte, error) {
74+
f, err := entry.Blob().DataAsync(ctx)
7475
if err != nil {
7576
return nil, err
7677
}
@@ -96,6 +97,7 @@ func GetEventsFromContent(content []byte) ([]*jobparser.Event, error) {
9697
}
9798

9899
func DetectWorkflows(
100+
ctx context.Context,
99101
gitRepo *git.Repository,
100102
commit *git.Commit,
101103
triggedEvent webhook_module.HookEventType,
@@ -110,7 +112,7 @@ func DetectWorkflows(
110112
workflows := make([]*DetectedWorkflow, 0, len(entries))
111113
schedules := make([]*DetectedWorkflow, 0, len(entries))
112114
for _, entry := range entries {
113-
content, err := GetContentFromEntry(entry)
115+
content, err := GetContentFromEntry(ctx, entry)
114116
if err != nil {
115117
return nil, nil, err
116118
}
@@ -146,15 +148,15 @@ func DetectWorkflows(
146148
return workflows, schedules, nil
147149
}
148150

149-
func DetectScheduledWorkflows(gitRepo *git.Repository, commit *git.Commit) ([]*DetectedWorkflow, error) {
151+
func DetectScheduledWorkflows(ctx context.Context, gitRepo *git.Repository, commit *git.Commit) ([]*DetectedWorkflow, error) {
150152
entries, err := ListWorkflows(commit)
151153
if err != nil {
152154
return nil, err
153155
}
154156

155157
wfs := make([]*DetectedWorkflow, 0, len(entries))
156158
for _, entry := range entries {
157-
content, err := GetContentFromEntry(entry)
159+
content, err := GetContentFromEntry(ctx, entry)
158160
if err != nil {
159161
return nil, err
160162
}

‎modules/fileicon/basic.go‎

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,18 +4,19 @@
44
package fileicon
55

66
import (
7+
"context"
78
"html/template"
89

910
"code.gitea.io/gitea/modules/git"
1011
"code.gitea.io/gitea/modules/svg"
1112
)
1213

13-
func BasicThemeIcon(entry *git.TreeEntry) template.HTML {
14+
func BasicThemeIcon(ctx context.Context, entry *git.TreeEntry) template.HTML {
1415
svgName := "octicon-file"
1516
switch {
1617
case entry.IsLink():
1718
svgName = "octicon-file-symlink-file"
18-
if te, err := entry.FollowLink(); err == nil && te.IsDir() {
19+
if te, err := entry.FollowLink(ctx); err == nil && te.IsDir() {
1920
svgName = "octicon-file-directory-symlink"
2021
}
2122
case entry.IsDir():

‎modules/fileicon/material.go‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,11 +87,11 @@ func (m *MaterialIconProvider) renderFileIconSVG(ctx reqctx.RequestContext, name
8787

8888
func (m *MaterialIconProvider) FileIcon(ctx reqctx.RequestContext, entry *git.TreeEntry) template.HTML {
8989
if m.rules == nil {
90-
return BasicThemeIcon(entry)
90+
return BasicThemeIcon(ctx, entry)
9191
}
9292

9393
if entry.IsLink() {
94-
if te, err := entry.FollowLink(); err == nil && te.IsDir() {
94+
if te, err := entry.FollowLink(ctx); err == nil && te.IsDir() {
9595
// keep the old "octicon-xxx" class name to make some "theme plugin selector" could still work
9696
return svg.RenderHTML("material-folder-symlink", 16, "octicon-file-directory-symlink")
9797
}

‎modules/git/blame.go‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -132,7 +132,7 @@ func (r *BlameReader) Close() error {
132132
func CreateBlameReader(ctx context.Context, objectFormat ObjectFormat, repoPath string, commit *Commit, file string, bypassBlameIgnore bool) (*BlameReader, error) {
133133
var ignoreRevsFile *string
134134
if DefaultFeatures().CheckVersionAtLeast("2.23") && !bypassBlameIgnore {
135-
ignoreRevsFile = tryCreateBlameIgnoreRevsFile(commit)
135+
ignoreRevsFile = tryCreateBlameIgnoreRevsFile(ctx, commit)
136136
}
137137

138138
cmd := NewCommandNoGlobals("blame", "--porcelain")
@@ -180,13 +180,13 @@ func CreateBlameReader(ctx context.Context, objectFormat ObjectFormat, repoPath
180180
}, nil
181181
}
182182

183-
func tryCreateBlameIgnoreRevsFile(commit *Commit) *string {
183+
func tryCreateBlameIgnoreRevsFile(ctx context.Context, commit *Commit) *string {
184184
entry, err := commit.GetTreeEntryByPath(".git-blame-ignore-revs")
185185
if err != nil {
186186
return nil
187187
}
188188

189-
r, err := entry.Blob().DataAsync()
189+
r, err := entry.Blob().DataAsync(ctx)
190190
if err != nil {
191191
return nil
192192
}

‎modules/git/blob.go‎

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ package git
66

77
import (
88
"bytes"
9+
"context"
910
"encoding/base64"
1011
"errors"
1112
"io"
@@ -22,11 +23,11 @@ func (b *Blob) Name() string {
2223
}
2324

2425
// GetBlobContent Gets the limited content of the blob as raw text
25-
func (b *Blob) GetBlobContent(limit int64) (string, error) {
26+
func (b *Blob) GetBlobContent(ctx context.Context, limit int64) (string, error) {
2627
if limit <= 0 {
2728
return "", nil
2829
}
29-
dataRc, err := b.DataAsync()
30+
dataRc, err := b.DataAsync(ctx)
3031
if err != nil {
3132
return "", err
3233
}
@@ -37,8 +38,8 @@ func (b *Blob) GetBlobContent(limit int64) (string, error) {
3738

3839
// GetBlobLineCount gets line count of the blob.
3940
// It will also try to write the content to w if it's not nil, then we could pre-fetch the content without reading it again.
40-
func (b *Blob) GetBlobLineCount(w io.Writer) (int, error) {
41-
reader, err := b.DataAsync()
41+
func (b *Blob) GetBlobLineCount(ctx context.Context, w io.Writer) (int, error) {
42+
reader, err := b.DataAsync(ctx)
4243
if err != nil {
4344
return 0, err
4445
}
@@ -64,8 +65,8 @@ func (b *Blob) GetBlobLineCount(w io.Writer) (int, error) {
6465
}
6566

6667
// GetBlobContentBase64 Reads the content of the blob with a base64 encode and returns the encoded string
67-
func (b *Blob) GetBlobContentBase64() (string, error) {
68-
dataRc, err := b.DataAsync()
68+
func (b *Blob) GetBlobContentBase64(ctx context.Context) (string, error) {
69+
dataRc, err := b.DataAsync(ctx)
6970
if err != nil {
7071
return "", err
7172
}
@@ -93,8 +94,8 @@ func (b *Blob) GetBlobContentBase64() (string, error) {
9394
}
9495

9596
// GuessContentType guesses the content type of the blob.
96-
func (b *Blob) GuessContentType() (typesniffer.SniffedType, error) {
97-
r, err := b.DataAsync()
97+
func (b *Blob) GuessContentType(ctx context.Context) (typesniffer.SniffedType, error) {
98+
r, err := b.DataAsync(ctx)
9899
if err != nil {
99100
return typesniffer.SniffedType{}, err
100101
}

‎modules/git/blob_gogit.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ type Blob struct {
2323

2424
// DataAsync gets a ReadCloser for the contents of a blob without reading it all.
2525
// Calling the Close function on the result will discard all unread output.
26-
func (b *Blob) DataAsync() (io.ReadCloser, error) {
26+
func (b *Blob) DataAsync(_ context.Context) (io.ReadCloser, error) {
2727
return b.gogitEncodedObj.Reader()
2828
}
2929

‎modules/git/blob_nogogit.go‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@ type Blob struct {
2626

2727
// DataAsync gets a ReadCloser for the contents of a blob without reading it all.
2828
// Calling the Close function on the result will discard all unread output.
29-
func (b *Blob) DataAsync() (io.ReadCloser, error) {
30-
wr, rd, cancel, err := b.repo.CatFileBatch(b.repo.Ctx)
29+
func (b *Blob) DataAsync(ctx context.Context) (io.ReadCloser, error) {
30+
wr, rd, cancel, err := b.repo.CatFileBatch(ctx)
3131
if err != nil {
3232
return nil, err
3333
}

‎modules/git/blob_test.go‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ func TestBlob_Data(t *testing.T) {
2323
testBlob, err := repo.GetBlob("6c493ff740f9380390d5c9ddef4af18697ac9375")
2424
assert.NoError(t, err)
2525

26-
r, err := testBlob.DataAsync()
26+
r, err := testBlob.DataAsync(t.Context())
2727
assert.NoError(t, err)
2828
require.NotNil(t, r)
2929

@@ -48,7 +48,7 @@ func Benchmark_Blob_Data(b *testing.B) {
4848
}
4949

5050
for b.Loop() {
51-
r, err := testBlob.DataAsync()
51+
r, err := testBlob.DataAsync(b.Context())
5252
if err != nil {
5353
b.Fatal(err)
5454
}

‎modules/git/commit.go‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -328,13 +328,13 @@ func (c *Commit) HasFile(filename string) (bool, error) {
328328
}
329329

330330
// GetFileContent reads a file content as a string or returns false if this was not possible
331-
func (c *Commit) GetFileContent(filename string, limit int) (string, error) {
331+
func (c *Commit) GetFileContent(ctx context.Context, filename string, limit int) (string, error) {
332332
entry, err := c.GetTreeEntryByPath(filename)
333333
if err != nil {
334334
return "", err
335335
}
336336

337-
r, err := entry.Blob().DataAsync()
337+
r, err := entry.Blob().DataAsync(ctx)
338338
if err != nil {
339339
return "", err
340340
}

‎modules/git/commit_info_gogit.go‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ func (tes Entries) GetCommitsInfo(ctx context.Context, commit *Commit, treePath
8080
} else {
8181
fullPath = entry.Name()
8282
}
83-
if subModule, err := commit.GetSubModule(fullPath); err != nil {
83+
if subModule, err := commit.GetSubModule(ctx, fullPath); err != nil {
8484
return nil, nil, err
8585
} else if subModule != nil {
8686
subModuleURL = subModule.URL

0 commit comments

Comments
 (0)