95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
package llbsolver
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"path"
|
|
|
|
cacheconfig "github.com/moby/buildkit/cache/config"
|
|
"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 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 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
|
|
}
|
|
}
|
|
|
|
func workerRefResolver(refCfg cacheconfig.RefConfig, all bool, 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.GetRemotes(ctx, true, refCfg, all, g)
|
|
}
|
|
}
|