mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
Fix parsing resources from compose file for stack deploy.
Signed-off-by: Daniel Nephin <dnephin@docker.com>
This commit is contained in:
parent
43cc971d76
commit
f27a297595
2 changed files with 37 additions and 6 deletions
|
@ -723,10 +723,14 @@ func convertUpdateConfig(source *composetypes.UpdateConfig) *swarm.UpdateConfig
|
||||||
|
|
||||||
func convertResources(source composetypes.Resources) (*swarm.ResourceRequirements, error) {
|
func convertResources(source composetypes.Resources) (*swarm.ResourceRequirements, error) {
|
||||||
resources := &swarm.ResourceRequirements{}
|
resources := &swarm.ResourceRequirements{}
|
||||||
|
var err error
|
||||||
if source.Limits != nil {
|
if source.Limits != nil {
|
||||||
cpus, err := opts.ParseCPUs(source.Limits.NanoCPUs)
|
var cpus int64
|
||||||
if err != nil {
|
if source.Limits.NanoCPUs != "" {
|
||||||
return nil, err
|
cpus, err = opts.ParseCPUs(source.Limits.NanoCPUs)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
}
|
}
|
||||||
resources.Limits = &swarm.Resources{
|
resources.Limits = &swarm.Resources{
|
||||||
NanoCPUs: cpus,
|
NanoCPUs: cpus,
|
||||||
|
@ -734,9 +738,12 @@ func convertResources(source composetypes.Resources) (*swarm.ResourceRequirement
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if source.Reservations != nil {
|
if source.Reservations != nil {
|
||||||
cpus, err := opts.ParseCPUs(source.Reservations.NanoCPUs)
|
var cpus int64
|
||||||
if err != nil {
|
if source.Reservations.NanoCPUs != "" {
|
||||||
return nil, err
|
cpus, err = opts.ParseCPUs(source.Reservations.NanoCPUs)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
}
|
}
|
||||||
resources.Reservations = &swarm.Resources{
|
resources.Reservations = &swarm.Resources{
|
||||||
NanoCPUs: cpus,
|
NanoCPUs: cpus,
|
||||||
|
|
|
@ -5,6 +5,7 @@ import (
|
||||||
|
|
||||||
composetypes "github.com/aanand/compose-file/types"
|
composetypes "github.com/aanand/compose-file/types"
|
||||||
"github.com/docker/docker/api/types/mount"
|
"github.com/docker/docker/api/types/mount"
|
||||||
|
"github.com/docker/docker/api/types/swarm"
|
||||||
"github.com/docker/docker/pkg/testutil/assert"
|
"github.com/docker/docker/pkg/testutil/assert"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -28,3 +29,26 @@ func TestConvertVolumeToMountInvalidFormat(t *testing.T) {
|
||||||
assert.Error(t, err, "invalid volume: "+vol)
|
assert.Error(t, err, "invalid volume: "+vol)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestConvertResourcesOnlyMemory(t *testing.T) {
|
||||||
|
source := composetypes.Resources{
|
||||||
|
Limits: &composetypes.Resource{
|
||||||
|
MemoryBytes: composetypes.UnitBytes(300000000),
|
||||||
|
},
|
||||||
|
Reservations: &composetypes.Resource{
|
||||||
|
MemoryBytes: composetypes.UnitBytes(200000000),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
resources, err := convertResources(source)
|
||||||
|
assert.NilError(t, err)
|
||||||
|
|
||||||
|
expected := &swarm.ResourceRequirements{
|
||||||
|
Limits: &swarm.Resources{
|
||||||
|
MemoryBytes: 300000000,
|
||||||
|
},
|
||||||
|
Reservations: &swarm.Resources{
|
||||||
|
MemoryBytes: 200000000,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
assert.DeepEqual(t, resources, expected)
|
||||||
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue