aboutsummaryrefslogtreecommitdiff
path: root/objectstore/chain/chain.go
diff options
context:
space:
mode:
authorGravatar Runxi Yu2026-02-21 01:47:11 +0800
committerGravatar Runxi Yu2026-02-21 01:47:11 +0800
commite936d246576e8bf714dbf8666e9349a5f22be87e (patch)
tree2e97a53b076627958f2babbb0bdd732dafcaa381 /objectstore/chain/chain.go
parentobjectdb/chain: Chain belongs separately from objectdb (diff)
signatureNo signature
objectstore: Rename from objectdb
Diffstat (limited to 'objectstore/chain/chain.go')
-rw-r--r--objectstore/chain/chain.go128
1 files changed, 128 insertions, 0 deletions
diff --git a/objectstore/chain/chain.go b/objectstore/chain/chain.go
new file mode 100644
index 00000000..f11e3f67
--- /dev/null
+++ b/objectstore/chain/chain.go
@@ -0,0 +1,128 @@
+// Package chain provides an ordered object database chain implementation.
+package chain
+
+import (
+ "errors"
+ "fmt"
+ "io"
+
+ "codeberg.org/lindenii/furgit/objectid"
+ "codeberg.org/lindenii/furgit/objectstore"
+ "codeberg.org/lindenii/furgit/objecttype"
+)
+
+// Chain queries multiple object databases in order.
+type Chain struct {
+ backends []objectstore.ObjectStore
+}
+
+// New creates an ordered object database chain.
+func New(backends ...objectstore.ObjectStore) *Chain {
+ return &Chain{
+ backends: append([]objectstore.ObjectStore(nil), backends...),
+ }
+}
+
+// ReadBytesFull reads a full serialized object from the first backend that has it.
+func (chain *Chain) ReadBytesFull(id objectid.ObjectID) ([]byte, error) {
+ for i, backend := range chain.backends {
+ if backend == nil {
+ continue
+ }
+ full, err := backend.ReadBytesFull(id)
+ if err == nil {
+ return full, nil
+ }
+ if errors.Is(err, objectstore.ErrObjectNotFound) {
+ continue
+ }
+ return nil, fmt.Errorf("objectstore: backend %d read bytes full: %w", i, err)
+ }
+ return nil, objectstore.ErrObjectNotFound
+}
+
+// ReadBytesContent reads an object's type and content bytes from the first backend that has it.
+func (chain *Chain) ReadBytesContent(id objectid.ObjectID) (objecttype.Type, []byte, error) {
+ for i, backend := range chain.backends {
+ if backend == nil {
+ continue
+ }
+ ty, content, err := backend.ReadBytesContent(id)
+ if err == nil {
+ return ty, content, nil
+ }
+ if errors.Is(err, objectstore.ErrObjectNotFound) {
+ continue
+ }
+ return objecttype.TypeInvalid, nil, fmt.Errorf("objectstore: backend %d read bytes content: %w", i, err)
+ }
+ return objecttype.TypeInvalid, nil, objectstore.ErrObjectNotFound
+}
+
+// ReadReaderFull reads a full serialized object stream from the first backend that has it.
+func (chain *Chain) ReadReaderFull(id objectid.ObjectID) (io.ReadCloser, error) {
+ for i, backend := range chain.backends {
+ if backend == nil {
+ continue
+ }
+ reader, err := backend.ReadReaderFull(id)
+ if err == nil {
+ 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 and content stream from the first backend that has it.
+func (chain *Chain) ReadReaderContent(id objectid.ObjectID) (objecttype.Type, io.ReadCloser, error) {
+ for i, backend := range chain.backends {
+ if backend == nil {
+ continue
+ }
+ ty, reader, err := backend.ReadReaderContent(id)
+ if err == nil {
+ return ty, reader, nil
+ }
+ if errors.Is(err, objectstore.ErrObjectNotFound) {
+ continue
+ }
+ return objecttype.TypeInvalid, nil, fmt.Errorf("objectstore: backend %d read reader content: %w", i, err)
+ }
+ return objecttype.TypeInvalid, nil, objectstore.ErrObjectNotFound
+}
+
+// ReadHeader reads object header data from the first backend that has it.
+func (chain *Chain) ReadHeader(id objectid.ObjectID) (objecttype.Type, int64, error) {
+ for i, backend := range chain.backends {
+ if backend == nil {
+ continue
+ }
+ ty, size, err := backend.ReadHeader(id)
+ if err == nil {
+ return ty, size, nil
+ }
+ if errors.Is(err, objectstore.ErrObjectNotFound) {
+ continue
+ }
+ return objecttype.TypeInvalid, 0, fmt.Errorf("objectstore: backend %d read header: %w", i, err)
+ }
+ return objecttype.TypeInvalid, 0, objectstore.ErrObjectNotFound
+}
+
+// Close closes all backends and joins close errors.
+func (chain *Chain) Close() error {
+ var errs []error
+ for _, backend := range chain.backends {
+ if backend == nil {
+ continue
+ }
+ if err := backend.Close(); err != nil {
+ errs = append(errs, err)
+ }
+ }
+ return errors.Join(errs...)
+}