diff options
| author |  Runxi Yu | 2026-02-21 18:18:43 +0800 |
|---|---|---|
| committer | Runxi Yu | 2026-02-21 18:18:43 +0800 |
| commit | 1fedf54fa79d5bdf67971d89924653e3bac7179e (patch) | |
| tree | ff93400f559bf98cf03dac3623ce86cf981006a6 /objectstore/packed/read_reader.go | |
| parent | iolimit: Add ExpectLengthReader (diff) | |
| signature | No signature | |
objectstore/{loose,packed}: Use iolimit v0.1.23
Diffstat (limited to 'objectstore/packed/read_reader.go')
| -rw-r--r-- | objectstore/packed/read_reader.go | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/objectstore/packed/read_reader.go b/objectstore/packed/read_reader.go index 4f40792d..bcafe975 100644 --- a/objectstore/packed/read_reader.go +++ b/objectstore/packed/read_reader.go @@ -5,6 +5,7 @@ import ( "fmt" "io" + "codeberg.org/lindenii/furgit/internal/iolimit" "codeberg.org/lindenii/furgit/objectheader" "codeberg.org/lindenii/furgit/objectid" "codeberg.org/lindenii/furgit/objecttype" @@ -45,7 +46,7 @@ func (store *Store) ReadReaderContent(id objectid.ObjectID) (objecttype.Type, in return objecttype.TypeInvalid, 0, nil, err } return meta.ty, meta.size, &readCloser{ - reader: io.LimitReader(zr, meta.size), + reader: iolimit.ExpectLengthReader(zr, meta.size), closer: zr, }, nil } @@ -80,7 +81,7 @@ func (store *Store) ReadReaderFull(id objectid.ObjectID) (io.ReadCloser, error) return nil, err } return &readCloser{ - reader: io.MultiReader(bytes.NewReader(header), io.LimitReader(zr, meta.size)), + reader: io.MultiReader(bytes.NewReader(header), iolimit.ExpectLengthReader(zr, meta.size)), closer: zr, }, nil } |