diff --git a/client/service_create.go b/client/service_create.go index 90082d150f..73fc68fb33 100644 --- a/client/service_create.go +++ b/client/service_create.go @@ -6,6 +6,7 @@ import ( "github.com/docker/distribution/reference" "github.com/docker/docker/api/types" + registrytypes "github.com/docker/docker/api/types/registry" "github.com/docker/docker/api/types/swarm" "github.com/opencontainers/go-digest" "golang.org/x/net/context" @@ -33,6 +34,8 @@ func (cli *Client) ServiceCreate(ctx context.Context, service swarm.ServiceSpec, if img != "" { service.TaskTemplate.ContainerSpec.Image = img } + // add platforms that are compatible with the service + service.TaskTemplate.Placement = updateServicePlatforms(service.TaskTemplate.Placement, distributionInspect) } } var response types.ServiceCreateResponse @@ -71,6 +74,22 @@ func imageWithDigestString(image string, dgst digest.Digest) string { return "" } +// updateServicePlatforms updates the Platforms in swarm.Placement to list +// all compatible platforms for the service, as found in distributionInspect +// and returns a pointer to the new or updated swarm.Placement struct +func updateServicePlatforms(placement *swarm.Placement, distributionInspect registrytypes.DistributionInspect) *swarm.Placement { + if placement == nil { + placement = &swarm.Placement{} + } + for _, p := range distributionInspect.Platforms { + placement.Platforms = append(placement.Platforms, swarm.Platform{ + Architecture: p.Architecture, + OS: p.OS, + }) + } + return placement +} + // digestWarning constructs a formatted warning string using the // image name that could not be pinned by digest. The formatting // is hardcoded, but could me made smarter in the future diff --git a/client/service_create_test.go b/client/service_create_test.go index 1e07382870..2ece62b993 100644 --- a/client/service_create_test.go +++ b/client/service_create_test.go @@ -4,13 +4,16 @@ import ( "bytes" "encoding/json" "fmt" + "io" "io/ioutil" "net/http" "strings" "testing" "github.com/docker/docker/api/types" + registrytypes "github.com/docker/docker/api/types/registry" "github.com/docker/docker/api/types/swarm" + "github.com/opencontainers/image-spec/specs-go/v1" "golang.org/x/net/context" ) @@ -55,3 +58,66 @@ func TestServiceCreate(t *testing.T) { t.Fatalf("expected `service_id`, got %s", r.ID) } } + +func TestServiceCreateCompatiblePlatforms(t *testing.T) { + var ( + platforms []v1.Platform + distributionInspectBody io.ReadCloser + distributionInspect registrytypes.DistributionInspect + ) + + client := &Client{ + client: newMockClient(func(req *http.Request) (*http.Response, error) { + if strings.HasPrefix(req.URL.Path, "/services/create") { + // check if the /distribution endpoint returned correct output + err := json.NewDecoder(distributionInspectBody).Decode(&distributionInspect) + if err != nil { + return nil, err + } + if len(distributionInspect.Platforms) == 0 || distributionInspect.Platforms[0].Architecture != platforms[0].Architecture || distributionInspect.Platforms[0].OS != platforms[0].OS { + return nil, fmt.Errorf("received incorrect platform information from registry") + } + + b, err := json.Marshal(types.ServiceCreateResponse{ + ID: "service_" + platforms[0].Architecture, + }) + if err != nil { + return nil, err + } + return &http.Response{ + StatusCode: http.StatusOK, + Body: ioutil.NopCloser(bytes.NewReader(b)), + }, nil + } else if strings.HasPrefix(req.URL.Path, "/distribution/") { + platforms = []v1.Platform{ + { + Architecture: "amd64", + OS: "linux", + }, + } + b, err := json.Marshal(registrytypes.DistributionInspect{ + Descriptor: v1.Descriptor{}, + Platforms: platforms, + }) + if err != nil { + return nil, err + } + distributionInspectBody = ioutil.NopCloser(bytes.NewReader(b)) + return &http.Response{ + StatusCode: http.StatusOK, + Body: ioutil.NopCloser(bytes.NewReader(b)), + }, nil + } else { + return nil, fmt.Errorf("unexpected URL '%s'", req.URL.Path) + } + }), + } + + r, err := client.ServiceCreate(context.Background(), swarm.ServiceSpec{}, types.ServiceCreateOptions{QueryRegistry: true}) + if err != nil { + t.Fatal(err) + } + if r.ID != "service_amd64" { + t.Fatalf("expected `service_amd64`, got %s", r.ID) + } +} diff --git a/client/service_update.go b/client/service_update.go index 8c7776aad7..0cb35a1991 100644 --- a/client/service_update.go +++ b/client/service_update.go @@ -46,6 +46,8 @@ func (cli *Client) ServiceUpdate(ctx context.Context, serviceID string, version if img != "" { service.TaskTemplate.ContainerSpec.Image = img } + // add platforms that are compatible with the service + service.TaskTemplate.Placement = updateServicePlatforms(service.TaskTemplate.Placement, distributionInspect) } }