diff options
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 } |