-
Notifications
You must be signed in to change notification settings - Fork 1.1k
/
result.go
94 lines (80 loc) · 2.3 KB
/
result.go
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
package llbsolver
import (
"bytes"
"context"
"path"
"github.com/moby/buildkit/cache/contenthash"
"github.com/moby/buildkit/session"
"github.com/moby/buildkit/solver"
"github.com/moby/buildkit/util/compression"
"github.com/moby/buildkit/worker"
digest "github.com/opencontainers/go-digest"
"github.com/pkg/errors"
"golang.org/x/sync/errgroup"
)
type Selector struct {
Path string
Wildcard bool
FollowLinks bool
IncludePatterns []string
ExcludePatterns []string
}
func (sel Selector) HasWildcardOrFilters() bool {
return sel.Wildcard || len(sel.IncludePatterns) != 0 || len(sel.ExcludePatterns) != 0
}
func UnlazyResultFunc(ctx context.Context, res solver.Result, g session.Group) error {
ref, ok := res.Sys().(*worker.WorkerRef)
if !ok {
return errors.Errorf("invalid reference: %T", res)
}
if ref.ImmutableRef == nil {
return nil
}
return ref.ImmutableRef.Extract(ctx, g)
}
func NewContentHashFunc(selectors []Selector) solver.ResultBasedCacheFunc {
return func(ctx context.Context, res solver.Result, s session.Group) (digest.Digest, error) {
ref, ok := res.Sys().(*worker.WorkerRef)
if !ok {
return "", errors.Errorf("invalid reference: %T", res)
}
if len(selectors) == 0 {
selectors = []Selector{{}}
}
dgsts := make([][]byte, len(selectors))
eg, ctx := errgroup.WithContext(ctx)
for i, sel := range selectors {
i, sel := i, sel
eg.Go(func() error {
dgst, err := contenthash.Checksum(
ctx, ref.ImmutableRef, path.Join("/", sel.Path),
contenthash.ChecksumOpts{
Wildcard: sel.Wildcard,
FollowLinks: sel.FollowLinks,
IncludePatterns: sel.IncludePatterns,
ExcludePatterns: sel.ExcludePatterns,
},
s,
)
if err != nil {
return err
}
dgsts[i] = []byte(dgst)
return nil
})
}
if err := eg.Wait(); err != nil {
return "", err
}
return digest.FromBytes(bytes.Join(dgsts, []byte{0})), nil
}
}
func workerRefConverter(g session.Group) func(ctx context.Context, res solver.Result) (*solver.Remote, error) {
return func(ctx context.Context, res solver.Result) (*solver.Remote, error) {
ref, ok := res.Sys().(*worker.WorkerRef)
if !ok {
return nil, errors.Errorf("invalid result: %T", res.Sys())
}
return ref.GetRemote(ctx, true, compression.Default, false, g)
}
}