buildkit/worker/cacheresult.go

101 lines
2.4 KiB
Go
Raw Normal View History

package worker
import (
"context"
"strings"
"time"
"github.com/moby/buildkit/cache"
"github.com/moby/buildkit/solver"
"github.com/pkg/errors"
)
func NewCacheResultStorage(wc *Controller) solver.CacheResultStorage {
return &cacheResultStorage{
wc: wc,
}
}
type cacheResultStorage struct {
wc *Controller
}
func (s *cacheResultStorage) Save(res solver.Result, createdAt time.Time) (solver.CacheResult, error) {
ref, ok := res.Sys().(*WorkerRef)
if !ok {
return solver.CacheResult{}, errors.Errorf("invalid result: %T", res.Sys())
}
if ref.ImmutableRef != nil {
if !cache.HasCachePolicyRetain(ref.ImmutableRef) {
if err := cache.CachePolicyRetain(ref.ImmutableRef); err != nil {
return solver.CacheResult{}, err
}
ref.ImmutableRef.Metadata().Commit()
}
}
return solver.CacheResult{ID: ref.ID(), CreatedAt: createdAt}, nil
}
func (s *cacheResultStorage) Load(ctx context.Context, res solver.CacheResult) (solver.Result, error) {
return s.load(res.ID, false)
}
func (s *cacheResultStorage) getWorkerRef(id string) (Worker, string, error) {
workerID, refID, err := parseWorkerRef(id)
if err != nil {
return nil, "", err
}
w, err := s.wc.Get(workerID)
if err != nil {
return nil, "", err
}
return w, refID, nil
}
func (s *cacheResultStorage) load(id string, hidden bool) (solver.Result, error) {
w, refID, err := s.getWorkerRef(id)
if err != nil {
return nil, err
}
if refID == "" {
return NewWorkerRefResult(nil, w), nil
}
ref, err := w.LoadRef(refID, hidden)
if err != nil {
return nil, err
}
return NewWorkerRefResult(ref, w), nil
}
func (s *cacheResultStorage) LoadRemote(ctx context.Context, res solver.CacheResult) (*solver.Remote, error) {
w, refID, err := s.getWorkerRef(res.ID)
if err != nil {
return nil, err
}
ref, err := w.LoadRef(refID, true)
if err != nil {
return nil, err
}
defer ref.Release(context.TODO())
remote, err := w.GetRemote(ctx, ref, false)
if err != nil {
return nil, nil // ignore error. loadRemote is best effort
}
return remote, nil
}
func (s *cacheResultStorage) Exists(id string) bool {
ref, err := s.load(id, true)
if err != nil {
return false
}
ref.Release(context.TODO())
return true
}
func parseWorkerRef(id string) (string, string, error) {
parts := strings.Split(id, "::")
if len(parts) != 2 {
return "", "", errors.Errorf("invalid workerref id: %s", id)
}
return parts[0], parts[1], nil
}