123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153 |
- package opts
- import (
- "encoding/csv"
- "fmt"
- "regexp"
- "strconv"
- "strings"
- "github.com/docker/docker/api/types/swarm"
- "github.com/docker/go-connections/nat"
- )
- const (
- portOptTargetPort = "target"
- portOptPublishedPort = "published"
- portOptProtocol = "protocol"
- portOptMode = "mode"
- )
- // PortOpt represents a port config in swarm mode.
- type PortOpt struct {
- ports []swarm.PortConfig
- }
- // Set a new port value
- func (p *PortOpt) Set(value string) error {
- longSyntax, err := regexp.MatchString(`\w+=\w+(,\w+=\w+)*`, value)
- if err != nil {
- return err
- }
- if longSyntax {
- csvReader := csv.NewReader(strings.NewReader(value))
- fields, err := csvReader.Read()
- if err != nil {
- return err
- }
- pConfig := swarm.PortConfig{}
- for _, field := range fields {
- parts := strings.SplitN(field, "=", 2)
- if len(parts) != 2 {
- return fmt.Errorf("invalid field %s", field)
- }
- key := strings.ToLower(parts[0])
- value := strings.ToLower(parts[1])
- switch key {
- case portOptProtocol:
- if value != string(swarm.PortConfigProtocolTCP) && value != string(swarm.PortConfigProtocolUDP) {
- return fmt.Errorf("invalid protocol value %s", value)
- }
- pConfig.Protocol = swarm.PortConfigProtocol(value)
- case portOptMode:
- if value != string(swarm.PortConfigPublishModeIngress) && value != string(swarm.PortConfigPublishModeHost) {
- return fmt.Errorf("invalid publish mode value %s", value)
- }
- pConfig.PublishMode = swarm.PortConfigPublishMode(value)
- case portOptTargetPort:
- tPort, err := strconv.ParseUint(value, 10, 16)
- if err != nil {
- return err
- }
- pConfig.TargetPort = uint32(tPort)
- case portOptPublishedPort:
- pPort, err := strconv.ParseUint(value, 10, 16)
- if err != nil {
- return err
- }
- pConfig.PublishedPort = uint32(pPort)
- default:
- return fmt.Errorf("invalid field key %s", key)
- }
- }
- if pConfig.TargetPort == 0 {
- return fmt.Errorf("missing mandatory field %q", portOptTargetPort)
- }
- if pConfig.PublishMode == "" {
- pConfig.PublishMode = swarm.PortConfigPublishModeIngress
- }
- if pConfig.Protocol == "" {
- pConfig.Protocol = swarm.PortConfigProtocolTCP
- }
- p.ports = append(p.ports, pConfig)
- } else {
- // short syntax
- portConfigs := []swarm.PortConfig{}
- // We can ignore errors because the format was already validated by ValidatePort
- ports, portBindings, _ := nat.ParsePortSpecs([]string{value})
- for port := range ports {
- portConfig, err := ConvertPortToPortConfig(port, portBindings)
- if err != nil {
- return err
- }
- portConfigs = append(portConfigs, portConfig...)
- }
- p.ports = append(p.ports, portConfigs...)
- }
- return nil
- }
- // Type returns the type of this option
- func (p *PortOpt) Type() string {
- return "port"
- }
- // String returns a string repr of this option
- func (p *PortOpt) String() string {
- ports := []string{}
- for _, port := range p.ports {
- repr := fmt.Sprintf("%v:%v/%s/%s", port.PublishedPort, port.TargetPort, port.Protocol, port.PublishMode)
- ports = append(ports, repr)
- }
- return strings.Join(ports, ", ")
- }
- // Value returns the ports
- func (p *PortOpt) Value() []swarm.PortConfig {
- return p.ports
- }
- // ConvertPortToPortConfig converts ports to the swarm type
- func ConvertPortToPortConfig(
- port nat.Port,
- portBindings map[nat.Port][]nat.PortBinding,
- ) ([]swarm.PortConfig, error) {
- ports := []swarm.PortConfig{}
- for _, binding := range portBindings[port] {
- hostPort, err := strconv.ParseUint(binding.HostPort, 10, 16)
- if err != nil && binding.HostPort != "" {
- return nil, fmt.Errorf("invalid hostport binding (%s) for port (%s)", binding.HostPort, port.Port())
- }
- ports = append(ports, swarm.PortConfig{
- //TODO Name: ?
- Protocol: swarm.PortConfigProtocol(strings.ToLower(port.Proto())),
- TargetPort: uint32(port.Int()),
- PublishedPort: uint32(hostPort),
- PublishMode: swarm.PortConfigPublishModeIngress,
- })
- }
- return ports, nil
- }
|