summaryrefslogtreecommitdiff
path: root/vendor/codeberg.org/gruf/go-store/v2/util/io.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/codeberg.org/gruf/go-store/v2/util/io.go')
-rw-r--r--vendor/codeberg.org/gruf/go-store/v2/util/io.go93
1 files changed, 14 insertions, 79 deletions
diff --git a/vendor/codeberg.org/gruf/go-store/v2/util/io.go b/vendor/codeberg.org/gruf/go-store/v2/util/io.go
index 3d62e8be6..c5135084a 100644
--- a/vendor/codeberg.org/gruf/go-store/v2/util/io.go
+++ b/vendor/codeberg.org/gruf/go-store/v2/util/io.go
@@ -5,102 +5,37 @@ import (
"io"
)
-// ReaderSize ...
+// ReaderSize defines a reader of known size in bytes.
type ReaderSize interface {
io.Reader
-
- // Size ...
Size() int64
}
-// ByteReaderSize ...
+// ByteReaderSize implements ReaderSize for an in-memory byte-slice.
type ByteReaderSize struct {
- bytes.Reader
+ br bytes.Reader
sz int64
}
-// NewByteReaderSize ...
+// NewByteReaderSize returns a new ByteReaderSize instance reset to slice b.
func NewByteReaderSize(b []byte) *ByteReaderSize {
- rs := ByteReaderSize{}
+ rs := new(ByteReaderSize)
rs.Reset(b)
- return &rs
+ return rs
+}
+
+// Read implements io.Reader.
+func (rs *ByteReaderSize) Read(b []byte) (int, error) {
+ return rs.br.Read(b)
}
-// Size implements ReaderSize.Size().
-func (rs ByteReaderSize) Size() int64 {
+// Size implements ReaderSize.
+func (rs *ByteReaderSize) Size() int64 {
return rs.sz
}
// Reset resets the ReaderSize to be reading from b.
func (rs *ByteReaderSize) Reset(b []byte) {
- rs.Reader.Reset(b)
+ rs.br.Reset(b)
rs.sz = int64(len(b))
}
-
-// NopReadCloser turns a supplied io.Reader into io.ReadCloser with a nop Close() implementation.
-func NopReadCloser(r io.Reader) io.ReadCloser {
- return &nopReadCloser{r}
-}
-
-// NopWriteCloser turns a supplied io.Writer into io.WriteCloser with a nop Close() implementation.
-func NopWriteCloser(w io.Writer) io.WriteCloser {
- return &nopWriteCloser{w}
-}
-
-// ReadCloserWithCallback adds a customizable callback to be called upon Close() of a supplied io.ReadCloser.
-// Note that the callback will never be called more than once, after execution this will remove the func reference.
-func ReadCloserWithCallback(rc io.ReadCloser, cb func()) io.ReadCloser {
- return &callbackReadCloser{
- ReadCloser: rc,
- callback: cb,
- }
-}
-
-// WriteCloserWithCallback adds a customizable callback to be called upon Close() of a supplied io.WriteCloser.
-// Note that the callback will never be called more than once, after execution this will remove the func reference.
-func WriteCloserWithCallback(wc io.WriteCloser, cb func()) io.WriteCloser {
- return &callbackWriteCloser{
- WriteCloser: wc,
- callback: cb,
- }
-}
-
-// nopReadCloser turns an io.Reader -> io.ReadCloser with a nop Close().
-type nopReadCloser struct{ io.Reader }
-
-func (r *nopReadCloser) Close() error { return nil }
-
-// nopWriteCloser turns an io.Writer -> io.WriteCloser with a nop Close().
-type nopWriteCloser struct{ io.Writer }
-
-func (w nopWriteCloser) Close() error { return nil }
-
-// callbackReadCloser allows adding our own custom callback to an io.ReadCloser.
-type callbackReadCloser struct {
- io.ReadCloser
- callback func()
-}
-
-func (c *callbackReadCloser) Close() error {
- if c.callback != nil {
- cb := c.callback
- c.callback = nil
- defer cb()
- }
- return c.ReadCloser.Close()
-}
-
-// callbackWriteCloser allows adding our own custom callback to an io.WriteCloser.
-type callbackWriteCloser struct {
- io.WriteCloser
- callback func()
-}
-
-func (c *callbackWriteCloser) Close() error {
- if c.callback != nil {
- cb := c.callback
- c.callback = nil
- defer cb()
- }
- return c.WriteCloser.Close()
-}