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
|
package opsutils
import (
"bytes"
"context"
"path"
"github.com/moby/buildkit/cache/contenthash"
"github.com/moby/buildkit/session"
"github.com/moby/buildkit/solver"
"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 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 errors.Wrapf(err, "failed to calculate checksum of ref %s", ref.ID())
}
dgsts[i] = []byte(dgst)
return nil
})
}
if err := eg.Wait(); err != nil {
return "", err
}
return digest.FromBytes(bytes.Join(dgsts, []byte{0})), nil
}
}
|
