You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
112 lines
2.4 KiB
Go
112 lines
2.4 KiB
Go
package driver
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
"github.com/docker/buildx/store"
|
|
"github.com/docker/buildx/util/progress"
|
|
clitypes "github.com/docker/cli/cli/config/types"
|
|
controlapi "github.com/moby/buildkit/api/services/control"
|
|
"github.com/moby/buildkit/client"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
var ErrNotRunning = errors.Errorf("driver not running")
|
|
var ErrNotConnecting = errors.Errorf("driver not connecting")
|
|
|
|
type Status int
|
|
|
|
const (
|
|
Inactive Status = iota
|
|
Starting
|
|
Running
|
|
Stopping
|
|
Stopped
|
|
)
|
|
|
|
func (s Status) String() string {
|
|
switch s {
|
|
case Inactive:
|
|
return "inactive"
|
|
case Starting:
|
|
return "starting"
|
|
case Running:
|
|
return "running"
|
|
case Stopping:
|
|
return "stopping"
|
|
case Stopped:
|
|
return "stopped"
|
|
}
|
|
return "unknown"
|
|
}
|
|
|
|
type Info struct {
|
|
Status Status
|
|
// DynamicNodes must be empty if the actual nodes are statically listed in the store
|
|
DynamicNodes []store.Node
|
|
}
|
|
|
|
type Auth interface {
|
|
GetAuthConfig(registryHostname string) (clitypes.AuthConfig, error)
|
|
}
|
|
|
|
type Driver interface {
|
|
Factory() Factory
|
|
Bootstrap(context.Context, progress.Logger) error
|
|
Info(context.Context) (*Info, error)
|
|
Version(context.Context) (string, error)
|
|
Stop(ctx context.Context, force bool) error
|
|
Rm(ctx context.Context, force, rmVolume, rmDaemon bool) error
|
|
Client(ctx context.Context) (*client.Client, error)
|
|
Features(ctx context.Context) map[Feature]bool
|
|
IsMobyDriver() bool
|
|
Config() InitConfig
|
|
}
|
|
|
|
func Boot(ctx, clientContext context.Context, d *DriverHandle, pw progress.Writer) (*client.Client, error) {
|
|
try := 0
|
|
for {
|
|
info, err := d.Info(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
try++
|
|
if info.Status != Running {
|
|
if try > 2 {
|
|
return nil, errors.Errorf("failed to bootstrap %T driver in attempts", d)
|
|
}
|
|
if err := d.Bootstrap(ctx, pw.Write); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
c, err := d.Client(clientContext)
|
|
if err != nil {
|
|
if errors.Cause(err) == ErrNotRunning && try <= 2 {
|
|
continue
|
|
}
|
|
return nil, err
|
|
}
|
|
return c, nil
|
|
}
|
|
}
|
|
|
|
func historyAPISupported(ctx context.Context, c *client.Client) bool {
|
|
cl, err := c.ControlClient().ListenBuildHistory(ctx, &controlapi.BuildHistoryRequest{
|
|
ActiveOnly: true,
|
|
Ref: "buildx-test-history-api-feature", // dummy ref to check if the server supports the API
|
|
EarlyExit: true,
|
|
})
|
|
if err != nil {
|
|
return false
|
|
}
|
|
for {
|
|
_, err := cl.Recv()
|
|
if errors.Is(err, io.EOF) {
|
|
return true
|
|
} else if err != nil {
|
|
return false
|
|
}
|
|
}
|
|
}
|