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
|
package file
import (
"context"
"github.com/moby/buildkit/cache"
"github.com/moby/buildkit/session"
"github.com/moby/buildkit/snapshot"
"github.com/moby/buildkit/solver/llbsolver/ops/fileoptypes"
"github.com/moby/buildkit/util/bklog"
"github.com/pkg/errors"
)
func NewRefManager(cm cache.Manager, name string) *RefManager {
return &RefManager{cm: cm, desc: name}
}
type RefManager struct {
cm cache.Manager
desc string
}
func (rm *RefManager) Prepare(ctx context.Context, ref fileoptypes.Ref, readonly bool, g session.Group) (_ fileoptypes.Mount, rerr error) {
ir, ok := ref.(cache.ImmutableRef)
if !ok && ref != nil {
return nil, errors.Errorf("invalid ref type: %T", ref)
}
if ir != nil && readonly {
m, err := ir.Mount(ctx, readonly, g)
if err != nil {
return nil, err
}
return &Mount{m: m, readonly: readonly}, nil
}
desc := "fileop target"
if d := rm.desc; d != "" {
desc = d
}
mr, err := rm.cm.New(ctx, ir, g, cache.WithDescription(desc), cache.CachePolicyRetain)
if err != nil {
return nil, err
}
defer func() {
if rerr != nil {
ctx := context.WithoutCancel(ctx)
if err := mr.SetCachePolicyDefault(); err != nil {
bklog.G(ctx).Errorf("failed to reset FileOp mutable ref cachepolicy: %v", err)
}
mr.Release(ctx)
}
}()
m, err := mr.Mount(ctx, readonly, g)
if err != nil {
return nil, err
}
return &Mount{m: m, mr: mr, readonly: readonly}, nil
}
func (rm *RefManager) Commit(ctx context.Context, mount fileoptypes.Mount) (fileoptypes.Ref, error) {
m, ok := mount.(*Mount)
if !ok {
return nil, errors.Errorf("invalid mount type %T", mount)
}
if m.mr == nil {
return nil, errors.Errorf("invalid mount without active ref for commit")
}
defer func() {
m.mr = nil
}()
return m.mr.Commit(ctx)
}
type Mount struct {
m snapshot.Mountable
mr cache.MutableRef
readonly bool
}
func (m *Mount) Mountable() snapshot.Mountable {
return m.m
}
func (m *Mount) Release(ctx context.Context) error {
if m.mr != nil {
return m.mr.Release(ctx)
}
return nil
}
func (m *Mount) IsFileOpMount() {}
func (m *Mount) Readonly() bool {
return m.readonly
}
|
