Support io.LimitedReader in object size check

thorfour · thorfour · commit 88a1fdc840f3 · 2024-10-29T10:44:35.000-05:00
Signed-off-by: thorfour &lt;me@thor-hansen.com&gt;
diff --git a/objstore.go b/objstore.go
@@ -211,6 +211,8 @@ func TryToGetSize(r io.Reader) (int64, error) {
 		return f.Size(), nil
 	case ObjectSizer:
 		return f.ObjectSize()
+	case *io.LimitedReader:
+		return f.N, nil
 	}
 	return 0, errors.Errorf("unsupported type of io.Reader: %T", r)
 }

Original file line number	Diff line number	Diff line change
`@@ -211,6 +211,8 @@ func TryToGetSize(r io.Reader) (int64, error) {`
`211`	`211`	`return f.Size(), nil`
`212`	`212`	`case ObjectSizer:`
`213`	`213`	`return f.ObjectSize()`
	`214`	`+ case *io.LimitedReader:`
	`215`	`+ return f.N, nil`
`214`	`216`	`}`
`215`	`217`	`return 0, errors.Errorf("unsupported type of io.Reader: %T", r)`
`216`	`218`	`}`