mirror of
https://github.com/moby/buildkit.git
synced 2026-08-09 01:01:08 +00:00
108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
package worker
|
|
|
|
import (
|
|
stderrors "errors"
|
|
|
|
"github.com/containerd/containerd/v2/pkg/filters"
|
|
"github.com/moby/buildkit/cache"
|
|
"github.com/moby/buildkit/client"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// Controller holds worker instances.
|
|
// Currently, only local workers are supported.
|
|
type Controller struct {
|
|
// TODO: define worker interface and support remote ones
|
|
workers []Worker
|
|
}
|
|
|
|
func (c *Controller) Close() error {
|
|
var errs []error
|
|
for _, w := range c.workers {
|
|
if err := w.Close(); err != nil {
|
|
errs = append(errs, err)
|
|
}
|
|
}
|
|
return stderrors.Join(errs...)
|
|
}
|
|
|
|
// Add adds a local worker.
|
|
// The first worker becomes the default.
|
|
//
|
|
// Add is not thread-safe.
|
|
func (c *Controller) Add(w Worker) error {
|
|
c.workers = append(c.workers, w)
|
|
return nil
|
|
}
|
|
|
|
// List lists workers
|
|
func (c *Controller) List(filterStrings ...string) ([]Worker, error) {
|
|
filter, err := filters.ParseAll(filterStrings...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var workers []Worker
|
|
for _, w := range c.workers {
|
|
if filter.Match(adaptWorker(w)) {
|
|
workers = append(workers, w)
|
|
}
|
|
}
|
|
return workers, nil
|
|
}
|
|
|
|
// GetDefault returns the default local worker
|
|
func (c *Controller) GetDefault() (Worker, error) {
|
|
if len(c.workers) == 0 {
|
|
return nil, errors.Errorf("no default worker")
|
|
}
|
|
return c.workers[0], nil
|
|
}
|
|
|
|
func (c *Controller) Get(id string) (Worker, error) {
|
|
for _, w := range c.workers {
|
|
if w.ID() == id {
|
|
return w, nil
|
|
}
|
|
}
|
|
return nil, errors.Errorf("worker %s not found", id)
|
|
}
|
|
|
|
// TODO: add Get(Constraint) (*Worker, error)
|
|
|
|
// WorkerInfos returns slice of WorkerInfo.
|
|
// The first item is the default worker.
|
|
func (c *Controller) WorkerInfos() []client.WorkerInfo {
|
|
out := make([]client.WorkerInfo, 0, len(c.workers))
|
|
for _, w := range c.workers {
|
|
out = append(out, client.WorkerInfo{
|
|
ID: w.ID(),
|
|
Labels: w.Labels(),
|
|
Platforms: w.Platforms(false),
|
|
BuildkitVersion: w.BuildkitVersion(),
|
|
})
|
|
}
|
|
return out
|
|
}
|
|
|
|
func (c *Controller) Infos() Infos {
|
|
return &infosController{c: c}
|
|
}
|
|
|
|
type infosController struct {
|
|
c *Controller
|
|
}
|
|
|
|
var _ Infos = &infosController{}
|
|
|
|
func (c *infosController) DefaultCacheManager() (cache.Manager, error) {
|
|
w, err := c.c.GetDefault()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return w.CacheManager(), nil
|
|
}
|
|
|
|
func (c *infosController) WorkerInfos() []client.WorkerInfo {
|
|
return c.c.WorkerInfos()
|
|
}
|