Browse code

Add support for host port PublishMode in services

Add api/cli support for adding host port PublishMode in services.

Signed-off-by: Jana Radhakrishnan <mrjana@docker.com>

Jana Radhakrishnan authored on 2016/11/11 05:13:26
Showing 15 changed files
... ...
@@ -31,8 +31,23 @@ type PortConfig struct {
31 31
 	TargetPort uint32 `json:",omitempty"`
32 32
 	// PublishedPort is the port on the swarm hosts
33 33
 	PublishedPort uint32 `json:",omitempty"`
34
+	// PublishMode is the mode in which port is published
35
+	PublishMode PortConfigPublishMode `json:",omitempty"`
34 36
 }
35 37
 
38
+// PortConfigPublishMode represents the mode in which the port is to
39
+// be published.
40
+type PortConfigPublishMode string
41
+
42
+const (
43
+	// PortConfigPublishModeIngress is used for ports published
44
+	// for ingress load balancing using routing mesh.
45
+	PortConfigPublishModeIngress PortConfigPublishMode = "ingress"
46
+	// PortConfigPublishModeHost is used for ports published
47
+	// for direct host level access on the host where the task is running.
48
+	PortConfigPublishModeHost PortConfigPublishMode = "host"
49
+)
50
+
36 51
 // PortConfigProtocol represents the protocol of a port.
37 52
 type PortConfigProtocol string
38 53
 
... ...
@@ -111,6 +111,7 @@ type TaskStatus struct {
111 111
 	Message         string          `json:",omitempty"`
112 112
 	Err             string          `json:",omitempty"`
113 113
 	ContainerStatus ContainerStatus `json:",omitempty"`
114
+	PortStatus      PortStatus      `json:",omitempty"`
114 115
 }
115 116
 
116 117
 // ContainerStatus represents the status of a container.
... ...
@@ -119,3 +120,9 @@ type ContainerStatus struct {
119 119
 	PID         int    `json:",omitempty"`
120 120
 	ExitCode    int    `json:",omitempty"`
121 121
 }
122
+
123
+// PortStatus represents the port status of a task's host ports whose
124
+// service has published host ports
125
+type PortStatus struct {
126
+	Ports []PortConfig `json:",omitempty"`
127
+}
... ...
@@ -40,12 +40,14 @@ func newCreateCommand(dockerCli *command.DockerCli) *cobra.Command {
40 40
 	flags.Var(&opts.constraints, flagConstraint, "Placement constraints")
41 41
 	flags.Var(&opts.networks, flagNetwork, "Network attachments")
42 42
 	flags.Var(&opts.secrets, flagSecret, "Specify secrets to expose to the service")
43
-	flags.VarP(&opts.endpoint.ports, flagPublish, "p", "Publish a port as a node port")
43
+	flags.VarP(&opts.endpoint.publishPorts, flagPublish, "p", "Publish a port as a node port")
44
+	flags.MarkHidden(flagPublish)
44 45
 	flags.Var(&opts.groups, flagGroup, "Set one or more supplementary user groups for the container")
45 46
 	flags.Var(&opts.dns, flagDNS, "Set custom DNS servers")
46 47
 	flags.Var(&opts.dnsOption, flagDNSOption, "Set DNS options")
47 48
 	flags.Var(&opts.dnsSearch, flagDNSSearch, "Set custom DNS search domains")
48 49
 	flags.Var(&opts.hosts, flagHost, "Set one or more custom host-to-IP mappings (host:ip)")
50
+	flags.Var(&opts.endpoint.expandedPorts, flagPort, "Publish a port")
49 51
 
50 52
 	flags.SetInterspersed(false)
51 53
 	return cmd
... ...
@@ -287,14 +287,15 @@ func convertNetworks(networks []string) []swarm.NetworkAttachmentConfig {
287 287
 }
288 288
 
289 289
 type endpointOptions struct {
290
-	mode  string
291
-	ports opts.ListOpts
290
+	mode          string
291
+	publishPorts  opts.ListOpts
292
+	expandedPorts opts.PortOpt
292 293
 }
293 294
 
294 295
 func (e *endpointOptions) ToEndpointSpec() *swarm.EndpointSpec {
295 296
 	portConfigs := []swarm.PortConfig{}
296 297
 	// We can ignore errors because the format was already validated by ValidatePort
297
-	ports, portBindings, _ := nat.ParsePortSpecs(e.ports.GetAll())
298
+	ports, portBindings, _ := nat.ParsePortSpecs(e.publishPorts.GetAll())
298 299
 
299 300
 	for port := range ports {
300 301
 		portConfigs = append(portConfigs, ConvertPortToPortConfig(port, portBindings)...)
... ...
@@ -302,7 +303,7 @@ func (e *endpointOptions) ToEndpointSpec() *swarm.EndpointSpec {
302 302
 
303 303
 	return &swarm.EndpointSpec{
304 304
 		Mode:  swarm.ResolutionMode(strings.ToLower(e.mode)),
305
-		Ports: portConfigs,
305
+		Ports: append(portConfigs, e.expandedPorts.Value()...),
306 306
 	}
307 307
 }
308 308
 
... ...
@@ -459,7 +460,7 @@ func newServiceOptions() *serviceOptions {
459 459
 		env:             opts.NewListOpts(runconfigopts.ValidateEnv),
460 460
 		envFile:         opts.NewListOpts(nil),
461 461
 		endpoint: endpointOptions{
462
-			ports: opts.NewListOpts(ValidatePort),
462
+			publishPorts: opts.NewListOpts(ValidatePort),
463 463
 		},
464 464
 		groups:    opts.NewListOpts(nil),
465 465
 		logDriver: newLogDriverOptions(),
... ...
@@ -647,6 +648,9 @@ const (
647 647
 	flagPublish               = "publish"
648 648
 	flagPublishRemove         = "publish-rm"
649 649
 	flagPublishAdd            = "publish-add"
650
+	flagPort                  = "port"
651
+	flagPortAdd               = "port-add"
652
+	flagPortRemove            = "port-rm"
650 653
 	flagReplicas              = "replicas"
651 654
 	flagReserveCPU            = "reserve-cpu"
652 655
 	flagReserveMemory         = "reserve-memory"
... ...
@@ -48,6 +48,8 @@ func newUpdateCommand(dockerCli *command.DockerCli) *cobra.Command {
48 48
 	flags.Var(newListOptsVar(), flagContainerLabelRemove, "Remove a container label by its key")
49 49
 	flags.Var(newListOptsVar(), flagMountRemove, "Remove a mount by its target path")
50 50
 	flags.Var(newListOptsVar(), flagPublishRemove, "Remove a published port by its target port")
51
+	flags.MarkHidden(flagPublishRemove)
52
+	flags.Var(newListOptsVar(), flagPortRemove, "Remove a port(target-port mandatory)")
51 53
 	flags.Var(newListOptsVar(), flagConstraintRemove, "Remove a constraint")
52 54
 	flags.Var(newListOptsVar(), flagDNSRemove, "Remove a custom DNS server")
53 55
 	flags.Var(newListOptsVar(), flagDNSOptionRemove, "Remove a DNS option")
... ...
@@ -60,7 +62,9 @@ func newUpdateCommand(dockerCli *command.DockerCli) *cobra.Command {
60 60
 	flags.Var(&opts.secrets, flagSecretAdd, "Add or update a secret on a service")
61 61
 	flags.Var(&opts.mounts, flagMountAdd, "Add or update a mount on a service")
62 62
 	flags.Var(&opts.constraints, flagConstraintAdd, "Add or update a placement constraint")
63
-	flags.Var(&opts.endpoint.ports, flagPublishAdd, "Add or update a published port")
63
+	flags.Var(&opts.endpoint.publishPorts, flagPublishAdd, "Add or update a published port")
64
+	flags.MarkHidden(flagPublishAdd)
65
+	flags.Var(&opts.endpoint.expandedPorts, flagPortAdd, "Add or update a port")
64 66
 	flags.Var(&opts.groups, flagGroupAdd, "Add an additional supplementary user group to the container")
65 67
 	flags.Var(&opts.dns, flagDNSAdd, "Add or update a custom DNS server")
66 68
 	flags.Var(&opts.dnsOption, flagDNSOptionAdd, "Add or update a DNS option")
... ...
@@ -267,7 +271,7 @@ func updateService(flags *pflag.FlagSet, spec *swarm.ServiceSpec) error {
267 267
 		}
268 268
 	}
269 269
 
270
-	if anyChanged(flags, flagPublishAdd, flagPublishRemove) {
270
+	if anyChanged(flags, flagPublishAdd, flagPublishRemove, flagPortAdd, flagPortRemove) {
271 271
 		if spec.EndpointSpec == nil {
272 272
 			spec.EndpointSpec = &swarm.EndpointSpec{}
273 273
 		}
... ...
@@ -627,7 +631,13 @@ func portConfigToString(portConfig *swarm.PortConfig) string {
627 627
 	if protocol == "" {
628 628
 		protocol = "tcp"
629 629
 	}
630
-	return fmt.Sprintf("%v/%s", portConfig.PublishedPort, protocol)
630
+
631
+	mode := portConfig.PublishMode
632
+	if mode == "" {
633
+		mode = "ingress"
634
+	}
635
+
636
+	return fmt.Sprintf("%v:%v/%s/%s", portConfig.PublishedPort, portConfig.TargetPort, protocol, mode)
631 637
 }
632 638
 
633 639
 func updatePorts(flags *pflag.FlagSet, portConfig *[]swarm.PortConfig) error {
... ...
@@ -649,6 +659,15 @@ func updatePorts(flags *pflag.FlagSet, portConfig *[]swarm.PortConfig) error {
649 649
 		}
650 650
 	}
651 651
 
652
+	if flags.Changed(flagPortAdd) {
653
+		for _, entry := range flags.Lookup(flagPortAdd).Value.(*opts.PortOpt).Value() {
654
+			if v, ok := portSet[portConfigToString(&entry)]; ok && v != entry {
655
+				return fmt.Errorf("conflicting port mapping between %v:%v/%s and %v:%v/%s", entry.PublishedPort, entry.TargetPort, entry.Protocol, v.PublishedPort, v.TargetPort, v.Protocol)
656
+			}
657
+			portSet[portConfigToString(&entry)] = entry
658
+		}
659
+	}
660
+
652 661
 	// Override previous PortConfig in service if there is any duplicate
653 662
 	for _, entry := range *portConfig {
654 663
 		if _, ok := portSet[portConfigToString(&entry)]; !ok {
... ...
@@ -657,6 +676,14 @@ func updatePorts(flags *pflag.FlagSet, portConfig *[]swarm.PortConfig) error {
657 657
 	}
658 658
 
659 659
 	toRemove := flags.Lookup(flagPublishRemove).Value.(*opts.ListOpts).GetAll()
660
+	removePortCSV := flags.Lookup(flagPortRemove).Value.(*opts.ListOpts).GetAll()
661
+	removePortOpts := &opts.PortOpt{}
662
+	for _, portCSV := range removePortCSV {
663
+		if err := removePortOpts.Set(portCSV); err != nil {
664
+			return err
665
+		}
666
+	}
667
+
660 668
 	newPorts := []swarm.PortConfig{}
661 669
 portLoop:
662 670
 	for _, port := range portSet {
... ...
@@ -666,14 +693,36 @@ portLoop:
666 666
 				continue portLoop
667 667
 			}
668 668
 		}
669
+
670
+		for _, pConfig := range removePortOpts.Value() {
671
+			if equalProtocol(port.Protocol, pConfig.Protocol) &&
672
+				port.TargetPort == pConfig.TargetPort &&
673
+				equalPublishMode(port.PublishMode, pConfig.PublishMode) {
674
+				continue portLoop
675
+			}
676
+		}
677
+
669 678
 		newPorts = append(newPorts, port)
670 679
 	}
680
+
671 681
 	// Sort the PortConfig to avoid unnecessary updates
672 682
 	sort.Sort(byPortConfig(newPorts))
673 683
 	*portConfig = newPorts
674 684
 	return nil
675 685
 }
676 686
 
687
+func equalProtocol(prot1, prot2 swarm.PortConfigProtocol) bool {
688
+	return prot1 == prot2 ||
689
+		(prot1 == swarm.PortConfigProtocol("") && prot2 == swarm.PortConfigProtocolTCP) ||
690
+		(prot2 == swarm.PortConfigProtocol("") && prot1 == swarm.PortConfigProtocolTCP)
691
+}
692
+
693
+func equalPublishMode(mode1, mode2 swarm.PortConfigPublishMode) bool {
694
+	return mode1 == mode2 ||
695
+		(mode1 == swarm.PortConfigPublishMode("") && mode2 == swarm.PortConfigPublishModeIngress) ||
696
+		(mode2 == swarm.PortConfigPublishMode("") && mode1 == swarm.PortConfigPublishModeIngress)
697
+}
698
+
677 699
 func equalPort(targetPort nat.Port, port swarm.PortConfig) bool {
678 700
 	return (string(port.Protocol) == targetPort.Proto() &&
679 701
 		port.TargetPort == uint32(targetPort.Int()))
... ...
@@ -238,7 +238,7 @@ func TestUpdatePortsDuplicateEntries(t *testing.T) {
238 238
 func TestUpdatePortsDuplicateKeys(t *testing.T) {
239 239
 	// Test case for #25375
240 240
 	flags := newUpdateCommand(nil).Flags()
241
-	flags.Set("publish-add", "80:20")
241
+	flags.Set("publish-add", "80:80")
242 242
 
243 243
 	portConfigs := []swarm.PortConfig{
244 244
 		{TargetPort: 80, PublishedPort: 80},
... ...
@@ -247,21 +247,7 @@ func TestUpdatePortsDuplicateKeys(t *testing.T) {
247 247
 	err := updatePorts(flags, &portConfigs)
248 248
 	assert.Equal(t, err, nil)
249 249
 	assert.Equal(t, len(portConfigs), 1)
250
-	assert.Equal(t, portConfigs[0].TargetPort, uint32(20))
251
-}
252
-
253
-func TestUpdatePortsConflictingFlags(t *testing.T) {
254
-	// Test case for #25375
255
-	flags := newUpdateCommand(nil).Flags()
256
-	flags.Set("publish-add", "80:80")
257
-	flags.Set("publish-add", "80:20")
258
-
259
-	portConfigs := []swarm.PortConfig{
260
-		{TargetPort: 80, PublishedPort: 80},
261
-	}
262
-
263
-	err := updatePorts(flags, &portConfigs)
264
-	assert.Error(t, err, "conflicting port mapping")
250
+	assert.Equal(t, portConfigs[0].TargetPort, uint32(80))
265 251
 }
266 252
 
267 253
 func TestUpdateHealthcheckTable(t *testing.T) {
... ...
@@ -17,10 +17,25 @@ import (
17 17
 )
18 18
 
19 19
 const (
20
-	psTaskItemFmt = "%s\t%s\t%s\t%s\t%s %s ago\t%s\n"
20
+	psTaskItemFmt = "%s\t%s\t%s\t%s\t%s %s ago\t%s\t%s\n"
21 21
 	maxErrLength  = 30
22 22
 )
23 23
 
24
+type portStatus swarm.PortStatus
25
+
26
+func (ps portStatus) String() string {
27
+	if len(ps.Ports) == 0 {
28
+		return ""
29
+	}
30
+
31
+	str := fmt.Sprintf("*:%d->%d/%s", ps.Ports[0].PublishedPort, ps.Ports[0].TargetPort, ps.Ports[0].Protocol)
32
+	for _, pConfig := range ps.Ports[1:] {
33
+		str += fmt.Sprintf(",*:%d->%d/%s", pConfig.PublishedPort, pConfig.TargetPort, pConfig.Protocol)
34
+	}
35
+
36
+	return str
37
+}
38
+
24 39
 type tasksBySlot []swarm.Task
25 40
 
26 41
 func (t tasksBySlot) Len() int {
... ...
@@ -51,7 +66,7 @@ func Print(dockerCli *command.DockerCli, ctx context.Context, tasks []swarm.Task
51 51
 
52 52
 	// Ignore flushing errors
53 53
 	defer writer.Flush()
54
-	fmt.Fprintln(writer, strings.Join([]string{"NAME", "IMAGE", "NODE", "DESIRED STATE", "CURRENT STATE", "ERROR"}, "\t"))
54
+	fmt.Fprintln(writer, strings.Join([]string{"NAME", "IMAGE", "NODE", "DESIRED STATE", "CURRENT STATE", "ERROR", "PORTS"}, "\t"))
55 55
 
56 56
 	if err := print(writer, ctx, tasks, resolver, noTrunc); err != nil {
57 57
 		return err
... ...
@@ -113,6 +128,7 @@ func print(out io.Writer, ctx context.Context, tasks []swarm.Task, resolver *idr
113 113
 			command.PrettyPrint(task.Status.State),
114 114
 			strings.ToLower(units.HumanDuration(time.Since(task.Status.Timestamp))),
115 115
 			taskErr,
116
+			portStatus(task.Status.PortStatus),
116 117
 		)
117 118
 	}
118 119
 	return nil
... ...
@@ -93,6 +93,7 @@ func endpointSpecFromGRPC(es *swarmapi.EndpointSpec) *types.EndpointSpec {
93 93
 			endpointSpec.Ports = append(endpointSpec.Ports, types.PortConfig{
94 94
 				Name:          portState.Name,
95 95
 				Protocol:      types.PortConfigProtocol(strings.ToLower(swarmapi.PortConfig_Protocol_name[int32(portState.Protocol)])),
96
+				PublishMode:   types.PortConfigPublishMode(strings.ToLower(swarmapi.PortConfig_PublishMode_name[int32(portState.PublishMode)])),
96 97
 				TargetPort:    portState.TargetPort,
97 98
 				PublishedPort: portState.PublishedPort,
98 99
 			})
... ...
@@ -112,6 +113,7 @@ func endpointFromGRPC(e *swarmapi.Endpoint) types.Endpoint {
112 112
 			endpoint.Ports = append(endpoint.Ports, types.PortConfig{
113 113
 				Name:          portState.Name,
114 114
 				Protocol:      types.PortConfigProtocol(strings.ToLower(swarmapi.PortConfig_Protocol_name[int32(portState.Protocol)])),
115
+				PublishMode:   types.PortConfigPublishMode(strings.ToLower(swarmapi.PortConfig_PublishMode_name[int32(portState.PublishMode)])),
115 116
 				TargetPort:    portState.TargetPort,
116 117
 				PublishedPort: portState.PublishedPort,
117 118
 			})
... ...
@@ -199,6 +199,7 @@ func ServiceSpecToGRPC(s types.ServiceSpec) (swarmapi.ServiceSpec, error) {
199 199
 			spec.Endpoint.Ports = append(spec.Endpoint.Ports, &swarmapi.PortConfig{
200 200
 				Name:          portConfig.Name,
201 201
 				Protocol:      swarmapi.PortConfig_Protocol(swarmapi.PortConfig_Protocol_value[strings.ToUpper(string(portConfig.Protocol))]),
202
+				PublishMode:   swarmapi.PortConfig_PublishMode(swarmapi.PortConfig_PublishMode_value[strings.ToUpper(string(portConfig.PublishMode))]),
202 203
 				TargetPort:    portConfig.TargetPort,
203 204
 				PublishedPort: portConfig.PublishedPort,
204 205
 			})
... ...
@@ -63,5 +63,19 @@ func TaskFromGRPC(t swarmapi.Task) types.Task {
63 63
 		task.NetworksAttachments = append(task.NetworksAttachments, networkAttachementFromGRPC(na))
64 64
 	}
65 65
 
66
+	if t.Status.PortStatus == nil {
67
+		return task
68
+	}
69
+
70
+	for _, p := range t.Status.PortStatus.Ports {
71
+		task.Status.PortStatus.Ports = append(task.Status.PortStatus.Ports, types.PortConfig{
72
+			Name:          p.Name,
73
+			Protocol:      types.PortConfigProtocol(strings.ToLower(swarmapi.PortConfig_Protocol_name[int32(p.Protocol)])),
74
+			PublishMode:   types.PortConfigPublishMode(strings.ToLower(swarmapi.PortConfig_PublishMode_name[int32(p.PublishMode)])),
75
+			TargetPort:    p.TargetPort,
76
+			PublishedPort: p.PublishedPort,
77
+		})
78
+	}
79
+
66 80
 	return task
67 81
 }
... ...
@@ -4,6 +4,7 @@ import (
4 4
 	"errors"
5 5
 	"fmt"
6 6
 	"net"
7
+	"strconv"
7 8
 	"strings"
8 9
 	"time"
9 10
 
... ...
@@ -17,6 +18,7 @@ import (
17 17
 	volumetypes "github.com/docker/docker/api/types/volume"
18 18
 	clustertypes "github.com/docker/docker/daemon/cluster/provider"
19 19
 	"github.com/docker/docker/reference"
20
+	"github.com/docker/go-connections/nat"
20 21
 	"github.com/docker/swarmkit/agent/exec"
21 22
 	"github.com/docker/swarmkit/api"
22 23
 	"github.com/docker/swarmkit/protobuf/ptypes"
... ...
@@ -136,17 +138,61 @@ func (c *containerConfig) image() string {
136 136
 	return reference.WithDefaultTag(ref).String()
137 137
 }
138 138
 
139
+func (c *containerConfig) portBindings() nat.PortMap {
140
+	portBindings := nat.PortMap{}
141
+	if c.task.Endpoint == nil {
142
+		return portBindings
143
+	}
144
+
145
+	for _, portConfig := range c.task.Endpoint.Ports {
146
+		if portConfig.PublishMode != api.PublishModeHost {
147
+			continue
148
+		}
149
+
150
+		port := nat.Port(fmt.Sprintf("%d/%s", portConfig.TargetPort, strings.ToLower(portConfig.Protocol.String())))
151
+		binding := []nat.PortBinding{
152
+			{},
153
+		}
154
+
155
+		if portConfig.PublishedPort != 0 {
156
+			binding[0].HostPort = strconv.Itoa(int(portConfig.PublishedPort))
157
+		}
158
+		portBindings[port] = binding
159
+	}
160
+
161
+	return portBindings
162
+}
163
+
164
+func (c *containerConfig) exposedPorts() map[nat.Port]struct{} {
165
+	exposedPorts := make(map[nat.Port]struct{})
166
+	if c.task.Endpoint == nil {
167
+		return exposedPorts
168
+	}
169
+
170
+	for _, portConfig := range c.task.Endpoint.Ports {
171
+		if portConfig.PublishMode != api.PublishModeHost {
172
+			continue
173
+		}
174
+
175
+		port := nat.Port(fmt.Sprintf("%d/%s", portConfig.TargetPort, strings.ToLower(portConfig.Protocol.String())))
176
+		exposedPorts[port] = struct{}{}
177
+	}
178
+
179
+	return exposedPorts
180
+}
181
+
139 182
 func (c *containerConfig) config() *enginecontainer.Config {
140 183
 	config := &enginecontainer.Config{
141
-		Labels:      c.labels(),
142
-		Tty:         c.spec().TTY,
143
-		User:        c.spec().User,
144
-		Env:         c.spec().Env,
145
-		Hostname:    c.spec().Hostname,
146
-		WorkingDir:  c.spec().Dir,
147
-		Image:       c.image(),
148
-		Volumes:     c.volumes(),
149
-		Healthcheck: c.healthcheck(),
184
+		Labels:       c.labels(),
185
+		Tty:          c.spec().TTY,
186
+		User:         c.spec().User,
187
+		Env:          c.spec().Env,
188
+		Hostname:     c.spec().Hostname,
189
+		WorkingDir:   c.spec().Dir,
190
+		Image:        c.image(),
191
+		Volumes:      c.volumes(),
192
+		ExposedPorts: c.exposedPorts(),
193
+		Healthcheck:  c.healthcheck(),
150 194
 	}
151 195
 
152 196
 	if len(c.spec().Command) > 0 {
... ...
@@ -333,10 +379,11 @@ func getMountMask(m *api.Mount) string {
333 333
 
334 334
 func (c *containerConfig) hostConfig() *enginecontainer.HostConfig {
335 335
 	hc := &enginecontainer.HostConfig{
336
-		Resources: c.resources(),
337
-		Binds:     c.binds(),
338
-		Tmpfs:     c.tmpfs(),
339
-		GroupAdd:  c.spec().Groups,
336
+		Resources:    c.resources(),
337
+		Binds:        c.binds(),
338
+		Tmpfs:        c.tmpfs(),
339
+		GroupAdd:     c.spec().Groups,
340
+		PortBindings: c.portBindings(),
340 341
 	}
341 342
 
342 343
 	if c.spec().DNSConfig != nil {
... ...
@@ -525,6 +572,10 @@ func (c *containerConfig) serviceConfig() *clustertypes.ServiceConfig {
525 525
 
526 526
 	if c.task.Endpoint != nil {
527 527
 		for _, ePort := range c.task.Endpoint.Ports {
528
+			if ePort.PublishMode != api.PublishModeIngress {
529
+				continue
530
+			}
531
+
528 532
 			svcCfg.ExposedPorts = append(svcCfg.ExposedPorts, &clustertypes.PortConfig{
529 533
 				Name:          ePort.Name,
530 534
 				Protocol:      int32(ePort.Protocol),
... ...
@@ -7,11 +7,14 @@ import (
7 7
 	"fmt"
8 8
 	"io"
9 9
 	"os"
10
+	"strconv"
11
+	"strings"
10 12
 	"time"
11 13
 
12 14
 	"github.com/docker/docker/api/types"
13 15
 	"github.com/docker/docker/api/types/events"
14 16
 	executorpkg "github.com/docker/docker/daemon/cluster/executor"
17
+	"github.com/docker/go-connections/nat"
15 18
 	"github.com/docker/libnetwork"
16 19
 	"github.com/docker/swarmkit/agent/exec"
17 20
 	"github.com/docker/swarmkit/api"
... ...
@@ -69,6 +72,19 @@ func (r *controller) ContainerStatus(ctx context.Context) (*api.ContainerStatus,
69 69
 	return parseContainerStatus(ctnr)
70 70
 }
71 71
 
72
+func (r *controller) PortStatus(ctx context.Context) (*api.PortStatus, error) {
73
+	ctnr, err := r.adapter.inspect(ctx)
74
+	if err != nil {
75
+		if isUnknownContainer(err) {
76
+			return nil, nil
77
+		}
78
+
79
+		return nil, err
80
+	}
81
+
82
+	return parsePortStatus(ctnr)
83
+}
84
+
72 85
 // Update tasks a recent task update and applies it to the container.
73 86
 func (r *controller) Update(ctx context.Context, t *api.Task) error {
74 87
 	// TODO(stevvooe): While assignment of tasks is idempotent, we do allow
... ...
@@ -553,6 +569,64 @@ func parseContainerStatus(ctnr types.ContainerJSON) (*api.ContainerStatus, error
553 553
 	return status, nil
554 554
 }
555 555
 
556
+func parsePortStatus(ctnr types.ContainerJSON) (*api.PortStatus, error) {
557
+	status := &api.PortStatus{}
558
+
559
+	if ctnr.NetworkSettings != nil && len(ctnr.NetworkSettings.Ports) > 0 {
560
+		exposedPorts, err := parsePortMap(ctnr.NetworkSettings.Ports)
561
+		if err != nil {
562
+			return nil, err
563
+		}
564
+		status.Ports = exposedPorts
565
+	}
566
+
567
+	return status, nil
568
+}
569
+
570
+func parsePortMap(portMap nat.PortMap) ([]*api.PortConfig, error) {
571
+	exposedPorts := make([]*api.PortConfig, 0, len(portMap))
572
+
573
+	for portProtocol, mapping := range portMap {
574
+		parts := strings.SplitN(string(portProtocol), "/", 2)
575
+		if len(parts) != 2 {
576
+			return nil, fmt.Errorf("invalid port mapping: %s", portProtocol)
577
+		}
578
+
579
+		port, err := strconv.ParseUint(parts[0], 10, 16)
580
+		if err != nil {
581
+			return nil, err
582
+		}
583
+
584
+		protocol := api.ProtocolTCP
585
+		switch strings.ToLower(parts[1]) {
586
+		case "tcp":
587
+			protocol = api.ProtocolTCP
588
+		case "udp":
589
+			protocol = api.ProtocolUDP
590
+		default:
591
+			return nil, fmt.Errorf("invalid protocol: %s", parts[1])
592
+		}
593
+
594
+		for _, binding := range mapping {
595
+			hostPort, err := strconv.ParseUint(binding.HostPort, 10, 16)
596
+			if err != nil {
597
+				return nil, err
598
+			}
599
+
600
+			// TODO(aluzzardi): We're losing the port `name` here since
601
+			// there's no way to retrieve it back from the Engine.
602
+			exposedPorts = append(exposedPorts, &api.PortConfig{
603
+				PublishMode:   api.PublishModeHost,
604
+				Protocol:      protocol,
605
+				TargetPort:    uint32(port),
606
+				PublishedPort: uint32(hostPort),
607
+			})
608
+		}
609
+	}
610
+
611
+	return exposedPorts, nil
612
+}
613
+
556 614
 type exitError struct {
557 615
 	code  int
558 616
 	cause error
... ...
@@ -32,6 +32,7 @@ func (s *DockerSwarmSuite) TestServiceUpdatePort(c *check.C) {
32 32
 			Protocol:      "tcp",
33 33
 			PublishedPort: 8082,
34 34
 			TargetPort:    8083,
35
+			PublishMode:   "ingress",
35 36
 		},
36 37
 	}
37 38
 
... ...
@@ -321,12 +321,9 @@ func (s *DockerSwarmSuite) TestSwarmPublishAdd(c *check.C) {
321 321
 	out, err = d.cmdRetryOutOfSequence("service", "update", "--publish-add", "80:80", "--publish-add", "80:20", name)
322 322
 	c.Assert(err, checker.NotNil)
323 323
 
324
-	out, err = d.cmdRetryOutOfSequence("service", "update", "--publish-add", "80:20", name)
325
-	c.Assert(err, checker.IsNil)
326
-
327 324
 	out, err = d.Cmd("service", "inspect", "--format", "{{ .Spec.EndpointSpec.Ports }}", name)
328 325
 	c.Assert(err, checker.IsNil)
329
-	c.Assert(strings.TrimSpace(out), checker.Equals, "[{ tcp 20 80}]")
326
+	c.Assert(strings.TrimSpace(out), checker.Equals, "[{ tcp 80 80 ingress}]")
330 327
 }
331 328
 
332 329
 func (s *DockerSwarmSuite) TestSwarmServiceWithGroup(c *check.C) {
333 330
new file mode 100644
... ...
@@ -0,0 +1,99 @@
0
+package opts
1
+
2
+import (
3
+	"encoding/csv"
4
+	"fmt"
5
+	"strconv"
6
+	"strings"
7
+
8
+	"github.com/docker/docker/api/types/swarm"
9
+)
10
+
11
+const (
12
+	portOptTargetPort    = "target"
13
+	portOptPublishedPort = "published"
14
+	portOptProtocol      = "protocol"
15
+	portOptMode          = "mode"
16
+)
17
+
18
+type PortOpt struct {
19
+	ports []swarm.PortConfig
20
+}
21
+
22
+// Set a new port value
23
+func (p *PortOpt) Set(value string) error {
24
+	csvReader := csv.NewReader(strings.NewReader(value))
25
+	fields, err := csvReader.Read()
26
+	if err != nil {
27
+		return err
28
+	}
29
+
30
+	pConfig := swarm.PortConfig{}
31
+	for _, field := range fields {
32
+		parts := strings.SplitN(field, "=", 2)
33
+		if len(parts) != 2 {
34
+			return fmt.Errorf("invalid field %s", field)
35
+		}
36
+
37
+		key := strings.ToLower(parts[0])
38
+		value := strings.ToLower(parts[1])
39
+
40
+		switch key {
41
+		case portOptProtocol:
42
+			if value != string(swarm.PortConfigProtocolTCP) && value != string(swarm.PortConfigProtocolUDP) {
43
+				return fmt.Errorf("invalid protocol value %s", value)
44
+			}
45
+
46
+			pConfig.Protocol = swarm.PortConfigProtocol(value)
47
+		case portOptMode:
48
+			if value != string(swarm.PortConfigPublishModeIngress) && value != string(swarm.PortConfigPublishModeHost) {
49
+				return fmt.Errorf("invalid publish mode value %s", value)
50
+			}
51
+
52
+			pConfig.PublishMode = swarm.PortConfigPublishMode(value)
53
+		case portOptTargetPort:
54
+			tPort, err := strconv.ParseUint(value, 10, 16)
55
+			if err != nil {
56
+				return err
57
+			}
58
+
59
+			pConfig.TargetPort = uint32(tPort)
60
+		case portOptPublishedPort:
61
+			pPort, err := strconv.ParseUint(value, 10, 16)
62
+			if err != nil {
63
+				return err
64
+			}
65
+
66
+			pConfig.PublishedPort = uint32(pPort)
67
+		default:
68
+			return fmt.Errorf("invalid field key %s", key)
69
+		}
70
+	}
71
+
72
+	if pConfig.TargetPort == 0 {
73
+		return fmt.Errorf("missing mandatory field %q", portOptTargetPort)
74
+	}
75
+
76
+	p.ports = append(p.ports, pConfig)
77
+	return nil
78
+}
79
+
80
+// Type returns the type of this option
81
+func (p *PortOpt) Type() string {
82
+	return "port"
83
+}
84
+
85
+// String returns a string repr of this option
86
+func (p *PortOpt) String() string {
87
+	ports := []string{}
88
+	for _, port := range p.ports {
89
+		repr := fmt.Sprintf("%v:%v/%s/%s", port.PublishedPort, port.TargetPort, port.Protocol, port.PublishMode)
90
+		ports = append(ports, repr)
91
+	}
92
+	return strings.Join(ports, ", ")
93
+}
94
+
95
+// Value returns the ports
96
+func (p *PortOpt) Value() []swarm.PortConfig {
97
+	return p.ports
98
+}