1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
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.Store
}
// New creates an ordered object database chain.
func New(backends ...objectstore.Store) *Chain {
return &Chain{
backends: append([]objectstore.Store(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, declared content length, and content stream from the first backend that has it.
func (chain *Chain) ReadReaderContent(id objectid.ObjectID) (objecttype.Type, int64, io.ReadCloser, error) {
for i, backend := range chain.backends {
if backend == nil {
continue
}
ty, size, reader, err := backend.ReadReaderContent(id)
if err == nil {
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
}
// 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...)
}
|
