Skip to content

Commit b65f52f

Browse files
authored
Merge pull request #5982 from thaJeztah/context_unexport_limitreader
cli/context/store: un-export LimitedReader
2 parents 58fba25 + ce61ea0 commit b65f52f

3 files changed

Lines changed: 9 additions & 9 deletions

File tree

cli/context/store/io_utils.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,14 @@ import (
55
"io"
66
)
77

8-
// LimitedReader is a fork of io.LimitedReader to override Read.
9-
type LimitedReader struct {
8+
// limitedReader is a fork of [io.LimitedReader] to override Read.
9+
type limitedReader struct {
1010
R io.Reader
1111
N int64 // max bytes remaining
1212
}
1313

14-
// Read is a fork of io.LimitedReader.Read that returns an error when limit exceeded.
15-
func (l *LimitedReader) Read(p []byte) (n int, err error) {
14+
// Read is a fork of [io.LimitedReader.Read] that returns an error when limit exceeded.
15+
func (l *limitedReader) Read(p []byte) (n int, err error) {
1616
if l.N < 0 {
1717
return 0, errors.New("read exceeds the defined limit")
1818
}

cli/context/store/io_utils_test.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,10 +15,10 @@ func TestLimitReaderReadAll(t *testing.T) {
1515
assert.NilError(t, err)
1616

1717
r = strings.NewReader("Test")
18-
_, err = io.ReadAll(&LimitedReader{R: r, N: 4})
18+
_, err = io.ReadAll(&limitedReader{R: r, N: 4})
1919
assert.NilError(t, err)
2020

2121
r = strings.NewReader("Test")
22-
_, err = io.ReadAll(&LimitedReader{R: r, N: 2})
22+
_, err = io.ReadAll(&limitedReader{R: r, N: 2})
2323
assert.Error(t, err, "read exceeds the defined limit")
2424
}

cli/context/store/store.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -356,7 +356,7 @@ func isValidFilePath(p string) error {
356356
}
357357

358358
func importTar(name string, s Writer, reader io.Reader) error {
359-
tr := tar.NewReader(&LimitedReader{R: reader, N: maxAllowedFileSizeToImport})
359+
tr := tar.NewReader(&limitedReader{R: reader, N: maxAllowedFileSizeToImport})
360360
tlsData := ContextTLSData{
361361
Endpoints: map[string]EndpointTLSData{},
362362
}
@@ -406,7 +406,7 @@ func importTar(name string, s Writer, reader io.Reader) error {
406406
}
407407

408408
func importZip(name string, s Writer, reader io.Reader) error {
409-
body, err := io.ReadAll(&LimitedReader{R: reader, N: maxAllowedFileSizeToImport})
409+
body, err := io.ReadAll(&limitedReader{R: reader, N: maxAllowedFileSizeToImport})
410410
if err != nil {
411411
return err
412412
}
@@ -434,7 +434,7 @@ func importZip(name string, s Writer, reader io.Reader) error {
434434
return err
435435
}
436436

437-
data, err := io.ReadAll(&LimitedReader{R: f, N: maxAllowedFileSizeToImport})
437+
data, err := io.ReadAll(&limitedReader{R: f, N: maxAllowedFileSizeToImport})
438438
defer f.Close()
439439
if err != nil {
440440
return err

0 commit comments

Comments
 (0)