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
|
package loose
import (
"errors"
"io/fs"
"path/filepath"
objectid "codeberg.org/lindenii/furgit/object/id"
)
// Close flushes and closes the underlying zlib stream and temp file.
//
// Repeated calls to Close are undefined behavior.
func (writer *streamWriter) Close() error {
errZlib := writer.zw.Close()
errSync := writer.file.Sync()
errFile := writer.file.Close()
writer.closed = true
writer.file = nil
return errors.Join(errZlib, errSync, errFile)
}
// finalize validates write completeness and atomically publishes the object.
// Publication is no-clobber: it links tmpRelPath to the object path and treats
// existing destination objects as success.
func (writer *streamWriter) finalize() (objectid.ObjectID, error) {
writer.finalized = true
var zero objectid.ObjectID
if !writer.closed {
err := writer.Close()
if err != nil {
return zero, err
}
}
if writer.fullMode && !writer.headerDone {
return zero, errors.New("objectstorer/loose: missing full object header")
}
if writer.expectedContentLeft != 0 {
return zero, errors.New("objectstorer/loose: object content shorter than declared size")
}
idBytes := writer.hash.Sum(nil)
id, err := objectid.FromBytes(writer.store.algo, idBytes)
if err != nil {
return zero, err
}
relPath, err := writer.store.objectPath(id)
if err != nil {
return zero, err
}
dir := filepath.Dir(relPath)
err = writer.store.root.MkdirAll(dir, 0o755)
if err != nil {
return zero, err
}
cleanup := true
defer func() {
if cleanup {
_ = writer.store.root.Remove(writer.tmpRelPath)
}
}()
err = writer.store.root.Link(writer.tmpRelPath, relPath)
if err != nil {
if errors.Is(err, fs.ErrExist) {
cleanup = false
_ = writer.store.root.Remove(writer.tmpRelPath)
return id, nil
}
return zero, err
}
cleanup = false
return id, nil
}
|
