mirror of
https://github.com/genuinetools/reg.git
synced 2024-09-12 06:11:03 -04:00
136 lines
3.7 KiB
Go
136 lines
3.7 KiB
Go
package build // import "github.com/docker/docker/api/server/backend/build"
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/docker/distribution/reference"
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/backend"
|
|
"github.com/docker/docker/builder"
|
|
buildkit "github.com/docker/docker/builder/builder-next"
|
|
"github.com/docker/docker/builder/fscache"
|
|
"github.com/docker/docker/image"
|
|
"github.com/docker/docker/pkg/stringid"
|
|
"github.com/pkg/errors"
|
|
"golang.org/x/sync/errgroup"
|
|
)
|
|
|
|
// ImageComponent provides an interface for working with images
|
|
type ImageComponent interface {
|
|
SquashImage(from string, to string) (string, error)
|
|
TagImageWithReference(image.ID, reference.Named) error
|
|
}
|
|
|
|
// Builder defines interface for running a build
|
|
type Builder interface {
|
|
Build(context.Context, backend.BuildConfig) (*builder.Result, error)
|
|
}
|
|
|
|
// Backend provides build functionality to the API router
|
|
type Backend struct {
|
|
builder Builder
|
|
fsCache *fscache.FSCache
|
|
imageComponent ImageComponent
|
|
buildkit *buildkit.Builder
|
|
}
|
|
|
|
// NewBackend creates a new build backend from components
|
|
func NewBackend(components ImageComponent, builder Builder, fsCache *fscache.FSCache, buildkit *buildkit.Builder) (*Backend, error) {
|
|
return &Backend{imageComponent: components, builder: builder, fsCache: fsCache, buildkit: buildkit}, nil
|
|
}
|
|
|
|
// Build builds an image from a Source
|
|
func (b *Backend) Build(ctx context.Context, config backend.BuildConfig) (string, error) {
|
|
options := config.Options
|
|
useBuildKit := options.Version == types.BuilderBuildKit
|
|
|
|
tagger, err := NewTagger(b.imageComponent, config.ProgressWriter.StdoutFormatter, options.Tags)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
var build *builder.Result
|
|
if useBuildKit {
|
|
build, err = b.buildkit.Build(ctx, config)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
} else {
|
|
build, err = b.builder.Build(ctx, config)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
|
|
if build == nil {
|
|
return "", nil
|
|
}
|
|
|
|
var imageID = build.ImageID
|
|
if options.Squash {
|
|
if imageID, err = squashBuild(build, b.imageComponent); err != nil {
|
|
return "", err
|
|
}
|
|
if config.ProgressWriter.AuxFormatter != nil {
|
|
if err = config.ProgressWriter.AuxFormatter.Emit("moby.image.id", types.BuildResult{ID: imageID}); err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
}
|
|
|
|
if !useBuildKit {
|
|
stdout := config.ProgressWriter.StdoutFormatter
|
|
fmt.Fprintf(stdout, "Successfully built %s\n", stringid.TruncateID(imageID))
|
|
err = tagger.TagImages(image.ID(imageID))
|
|
}
|
|
return imageID, err
|
|
}
|
|
|
|
// PruneCache removes all cached build sources
|
|
func (b *Backend) PruneCache(ctx context.Context) (*types.BuildCachePruneReport, error) {
|
|
eg, ctx := errgroup.WithContext(ctx)
|
|
|
|
var fsCacheSize uint64
|
|
eg.Go(func() error {
|
|
var err error
|
|
fsCacheSize, err = b.fsCache.Prune(ctx)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to prune fscache")
|
|
}
|
|
return nil
|
|
})
|
|
|
|
var buildCacheSize int64
|
|
eg.Go(func() error {
|
|
var err error
|
|
buildCacheSize, err = b.buildkit.Prune(ctx)
|
|
if err != nil {
|
|
return errors.Wrap(err, "failed to prune build cache")
|
|
}
|
|
return nil
|
|
})
|
|
|
|
if err := eg.Wait(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &types.BuildCachePruneReport{SpaceReclaimed: fsCacheSize + uint64(buildCacheSize)}, nil
|
|
}
|
|
|
|
// Cancel cancels the build by ID
|
|
func (b *Backend) Cancel(ctx context.Context, id string) error {
|
|
return b.buildkit.Cancel(ctx, id)
|
|
}
|
|
|
|
func squashBuild(build *builder.Result, imageComponent ImageComponent) (string, error) {
|
|
var fromID string
|
|
if build.FromImage != nil {
|
|
fromID = build.FromImage.ImageID()
|
|
}
|
|
imageID, err := imageComponent.SquashImage(build.ImageID, fromID)
|
|
if err != nil {
|
|
return "", errors.Wrap(err, "error squashing image")
|
|
}
|
|
return imageID, nil
|
|
}
|