objectstore/mix: Separate

1 files changed, 53 insertions, 0 deletions

diff --git a/objectstore/mix/reader.go b/objectstore/mix/reader.go
new file mode 100644
index 00000000..33811611
--- /dev/null
+++ b/objectstore/mix/reader.go
@@ -0,0 +1,53 @@
+package mix
+
+import (
+	"errors"
+	"fmt"
+	"io"
+
+	"codeberg.org/lindenii/furgit/objectid"
+	"codeberg.org/lindenii/furgit/objectstore"
+	"codeberg.org/lindenii/furgit/objecttype"
+)
+
+// ReadReaderFull reads a full serialized object stream from one backend that
+// has it.
+func (mix *Mix) ReadReaderFull(id objectid.ObjectID) (io.ReadCloser, error) {
+	for i, backend := 0, mix.firstBackend(); backend != nil; i, backend = i+1, mix.nextBackend(backend) {
+		reader, err := backend.ReadReaderFull(id)
+		if err == nil {
+			mix.touchBackend(backend)
+
+			return reader, nil
+		}
+
+		if errors.Is(err, objectstore.ErrObjectNotFound) {
+			continue
+		}
+
+		return nil, fmt.Errorf("objectstore: backend %d read reader full: %w", i, err)
+	}
+
+	return nil, objectstore.ErrObjectNotFound
+}
+
+// ReadReaderContent reads an object's type, declared content length, and
+// content stream from one backend that has it.
+func (mix *Mix) ReadReaderContent(id objectid.ObjectID) (objecttype.Type, int64, io.ReadCloser, error) {
+	for i, backend := 0, mix.firstBackend(); backend != nil; i, backend = i+1, mix.nextBackend(backend) {
+		ty, size, reader, err := backend.ReadReaderContent(id)
+		if err == nil {
+			mix.touchBackend(backend)
+
+			return ty, size, reader, nil
+		}
+
+		if errors.Is(err, objectstore.ErrObjectNotFound) {
+			continue
+		}
+
+		return objecttype.TypeInvalid, 0, nil, fmt.Errorf("objectstore: backend %d read reader content: %w", i, err)
+	}
+
+	return objecttype.TypeInvalid, 0, nil, objectstore.ErrObjectNotFound
+}