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
|
package loose
import (
"fmt"
"io"
"codeberg.org/lindenii/furgit/objectheader"
"codeberg.org/lindenii/furgit/objectid"
"codeberg.org/lindenii/furgit/objecttype"
)
// WriteReaderContent writes one loose object from typed content bytes read from src.
// src must provide exactly size bytes.
// size is required because loose object headers are "type size\0content", so the
// header must be emitted before streaming content without buffering.
func (store *Store) WriteReaderContent(ty objecttype.Type, size int64, src io.Reader) (objectid.ObjectID, error) {
if size < 0 {
return objectid.ObjectID{}, fmt.Errorf("objectstore/loose: negative content size: %d", size)
}
header, ok := objectheader.Encode(ty, size)
if !ok {
return objectid.ObjectID{}, fmt.Errorf("objectstore/loose: failed to encode object header for type %v", ty)
}
writer, err := store.newStreamWriter(false)
if err != nil {
return objectid.ObjectID{}, err
}
writer.headerDone = true
writer.expectedContentLeft = size
if err := writer.writeRawChunk(header); err != nil {
_ = writer.Close()
_ = store.root.Remove(writer.tmpRelPath)
return objectid.ObjectID{}, err
}
return writeReaderIntoStreamWriter(writer, src)
}
// WriteReaderFull writes one loose object from raw bytes "type size\0content"
// read from src.
func (store *Store) WriteReaderFull(src io.Reader) (objectid.ObjectID, error) {
writer, err := store.newStreamWriter(true)
if err != nil {
return objectid.ObjectID{}, err
}
return writeReaderIntoStreamWriter(writer, src)
}
// writeReaderIntoStreamWriter copies src into writer and publishes the object.
func writeReaderIntoStreamWriter(writer *streamWriter, src io.Reader) (objectid.ObjectID, error) {
if _, err := io.Copy(writer, src); err != nil {
_ = writer.Close()
_ = writer.store.root.Remove(writer.tmpRelPath)
return objectid.ObjectID{}, err
}
if err := writer.Close(); err != nil {
_ = writer.store.root.Remove(writer.tmpRelPath)
return objectid.ObjectID{}, err
}
id, err := writer.finalize()
if err != nil {
_ = writer.store.root.Remove(writer.tmpRelPath)
return objectid.ObjectID{}, err
}
return id, nil
}
|
