Rackspace/Openstack - Add/enhance --*-net-* create switches.

Signed-off-by: Robert Jones <robert@justjones.org>

drivers/openstack/client.go

@@ -41,7 +41,8 @@ type Client interface {
 	GetPublicKey(keyPairName string) ([]byte, error)
 	CreateKeyPair(d *Driver, name string, publicKey string) error
 	DeleteKeyPair(d *Driver, name string) error
-	GetNetworkID(d *Driver) (string, error)
+	GetNetworkIDs(d *Driver) ([]string, error)
+	GetNetworkID(d *Driver, networkName string) (string, error)
 	GetFlavorID(d *Driver) (string, error)
 	GetImageID(d *Driver) (string, error)
 	AssignFloatingIP(d *Driver, floatingIP *FloatingIP) error
@@ -66,12 +67,12 @@ func (c *GenericClient) CreateInstance(d *Driver) (string, error) {
 		SecurityGroups:   d.SecurityGroups,
 		AvailabilityZone: d.AvailabilityZone,
 	}
-	if d.NetworkId != "" {
+	if len(d.NetworkIds) > 0 {
-		serverOpts.Networks = []servers.Network{
+		networks := make([]servers.Network, len(d.NetworkIds))
-			{
+		for i, networkId := range d.NetworkIds {
-				UUID: d.NetworkId,
+			networks[i] = servers.Network{UUID: networkId}
-			},
 		}
+		serverOpts.Networks = networks
 	}
 
 	log.Info("Creating machine...")
@@ -199,15 +200,23 @@ func (c *GenericClient) GetInstanceIPAddresses(d *Driver) ([]IPAddress, error) {
 	return addresses, nil
 }
 
-func (c *GenericClient) GetNetworkID(d *Driver) (string, error) {
+func (c *GenericClient) GetNetworkIDs(d *Driver) ([]string, error) {
-	return c.getNetworkID(d, d.NetworkName)
+	networkIDs := make([]string, len(d.NetworkNames))
+	for i, networkName := range d.NetworkNames {
+		id, err := c.GetNetworkID(d, networkName)
+		if err != nil {
+			return nil, err
+		}
+		networkIDs[i] = id
+	}
+	return networkIDs, nil
 }
 
 func (c *GenericClient) GetFloatingIPPoolID(d *Driver) (string, error) {
-	return c.getNetworkID(d, d.FloatingIpPool)
+	return c.GetNetworkID(d, d.FloatingIPPool)
 }
 
-func (c *GenericClient) getNetworkID(d *Driver, networkName string) (string, error) {
+func (c *GenericClient) GetNetworkID(d *Driver, networkName string) (string, error) {
 	opts := networks.ListOpts{Name: networkName}
 	pager := networks.List(c.Network, opts)
 	networkID := ""
@@ -345,7 +354,7 @@ func (c *GenericClient) AssignFloatingIP(d *Driver, floatingIP *FloatingIP) erro
 func (c *GenericClient) assignNovaFloatingIP(d *Driver, floatingIP *FloatingIP) error {
 	if floatingIP.Ip == "" {
 		f, err := compute_ips.Create(c.Compute, compute_ips.CreateOpts{
-			Pool: d.FloatingIpPool,
+			Pool: d.FloatingIPPool,
 		}).Extract()
 		if err != nil {
 			return err
@@ -363,7 +372,7 @@ func (c *GenericClient) assignNeutronFloatingIP(d *Driver, floatingIP *FloatingI
 	}
 	if floatingIP.Id == "" {
 		f, err := floatingips.Create(c.Network, floatingips.CreateOpts{
-			FloatingNetworkID: d.FloatingIpPoolId,
+			FloatingNetworkID: d.FloatingIPPoolId,
 			PortID:            portID,
 		}).Extract()
 		if err != nil {
@@ -400,7 +409,7 @@ func (c *GenericClient) getNovaNetworkFloatingIPs(d *Driver) ([]FloatingIP, erro
 		ipListing, err := compute_ips.ExtractFloatingIPs(page)
 
 		for _, ip := range ipListing {
-			if ip.InstanceID == "" && ip.Pool == d.FloatingIpPool {
+			if ip.InstanceID == "" && ip.Pool == d.FloatingIPPool {
 				ips = append(ips, FloatingIP{
 					Id:   ip.ID,
 					Ip:   ip.IP,
@@ -415,11 +424,11 @@ func (c *GenericClient) getNovaNetworkFloatingIPs(d *Driver) ([]FloatingIP, erro
 
 func (c *GenericClient) getNeutronNetworkFloatingIPs(d *Driver) ([]FloatingIP, error) {
 	log.Debug("Listing floating IPs", map[string]string{
-		"FloatingNetworkId": d.FloatingIpPoolId,
+		"FloatingNetworkId": d.FloatingIPPoolId,
 		"TenantID":          d.TenantId,
 	})
 	pager := floatingips.List(c.Network, floatingips.ListOpts{
-		FloatingNetworkID: d.FloatingIpPoolId,
+		FloatingNetworkID: d.FloatingIPPoolId,
 		TenantID:          d.TenantId,
 	})
 
@@ -449,7 +458,7 @@ func (c *GenericClient) getNeutronNetworkFloatingIPs(d *Driver) ([]FloatingIP, e
 func (c *GenericClient) GetInstancePortID(d *Driver) (string, error) {
 	pager := ports.List(c.Network, ports.ListOpts{
 		DeviceID:  d.MachineId,
-		NetworkID: d.NetworkId,
+		NetworkID: d.NetworkIds[0],
 	})
 
 	var portID string

drivers/openstack/openstack.go

@@ -35,13 +35,13 @@ type Driver struct {
 	ImageName        string
 	ImageId          string
 	KeyPairName      string
-	NetworkName      string
+	NetworkNames     []string
-	NetworkId        string
+	NetworkIds       []string
 	PrivateKeyFile   string
 	SecurityGroups   []string
-	FloatingIpPool   string
+	FloatingIPPool   string
 	ComputeNetwork   bool
-	FloatingIpPoolId string
+	FloatingIPPoolId string
 	IpVersion        int
 	client           Client
 }
@@ -149,22 +149,22 @@ func (d *Driver) GetCreateFlags() []mcnflag.Flag {
 			Usage:  "OpenStack keypair to use to SSH to the instance",
 			Value:  "",
 		},
-		mcnflag.StringFlag{
-			EnvVar: "OS_NETWORK_ID",
-			Name:   "openstack-net-id",
-			Usage:  "OpenStack network id the machine will be connected on",
-			Value:  "",
-		},
 		mcnflag.StringFlag{
 			EnvVar: "OS_PRIVATE_KEY_FILE",
 			Name:   "openstack-private-key-file",
 			Usage:  "Private keyfile to use for SSH (absolute path)",
 			Value:  "",
 		},
+		mcnflag.StringFlag{
+			EnvVar: "OS_NETWORK_ID",
+			Name:   "openstack-net-id",
+			Usage:  "OpenStack comma separated networks id the machine will be connected on",
+			Value:  "",
+		},
 		mcnflag.StringFlag{
 			EnvVar: "OS_NETWORK_NAME",
 			Name:   "openstack-net-name",
-			Usage:  "OpenStack network name the machine will be connected on",
+			Usage:  "OpenStack comma separated network names the machine will be connected on",
 			Value:  "",
 		},
 		mcnflag.StringFlag{
@@ -181,7 +181,7 @@ func (d *Driver) GetCreateFlags() []mcnflag.Flag {
 		mcnflag.StringFlag{
 			EnvVar: "OS_FLOATINGIP_POOL",
 			Name:   "openstack-floatingip-pool",
-			Usage:  "OpenStack floating IP pool to get an IP from to assign to the instance",
+			Usage:  "OpenStack floating IP pool to get an IP from to assign to the instance (first network only)",
 			Value:  "",
 		},
 		mcnflag.IntFlag{
@@ -258,12 +258,16 @@ func (d *Driver) SetConfigFromFlags(flags drivers.DriverOptions) error {
 	d.FlavorName = flags.String("openstack-flavor-name")
 	d.ImageId = flags.String("openstack-image-id")
 	d.ImageName = flags.String("openstack-image-name")
-	d.NetworkId = flags.String("openstack-net-id")
+	if flags.String("openstack-net-id") != "" {
-	d.NetworkName = flags.String("openstack-net-name")
+		d.NetworkIds = strings.Split(flags.String("openstack-net-id"), ",")
+	}
+	if flags.String("openstack-net-name") != "" {
+		d.NetworkNames = strings.Split(flags.String("openstack-net-name"), ",")
+	}
 	if flags.String("openstack-sec-groups") != "" {
 		d.SecurityGroups = strings.Split(flags.String("openstack-sec-groups"), ",")
 	}
-	d.FloatingIpPool = flags.String("openstack-floatingip-pool")
+	d.FloatingIPPool = flags.String("openstack-floatingip-pool")
 	d.IpVersion = flags.Int("openstack-ip-version")
 	d.ComputeNetwork = flags.Bool("openstack-nova-network")
 	d.SSHUser = flags.String("openstack-ssh-user")
@@ -298,7 +302,7 @@ func (d *Driver) GetIP() (string, error) {
 	}
 
 	addressType := Fixed
-	if d.FloatingIpPool != "" {
+	if d.FloatingIPPool != "" {
 		addressType = Floating
 	}
 
@@ -371,7 +375,7 @@ func (d *Driver) Create() error {
 	if err := d.waitForInstanceActive(); err != nil {
 		return err
 	}
-	if d.FloatingIpPool != "" {
+	if d.FloatingIPPool != "" {
 		if err := d.assignFloatingIP(); err != nil {
 			return err
 		}
@@ -468,8 +472,8 @@ func (d *Driver) checkConfig() error {
 		return fmt.Errorf(errorExclusiveOptions, "Image name", "Image id")
 	}
 
-	if d.NetworkName != "" && d.NetworkId != "" {
+	if len(d.NetworkNames) > 0 && len(d.NetworkIds) > 0 {
-		return fmt.Errorf(errorExclusiveOptions, "Network name", "Network id")
+		return fmt.Errorf(errorExclusiveOptions, "Network names", "Network ids")
 	}
 	if d.EndpointType != "" && (d.EndpointType != "publicURL" && d.EndpointType != "adminURL" && d.EndpointType != "internalURL") {
 		return fmt.Errorf(errorWrongEndpointType)
@@ -481,27 +485,27 @@ func (d *Driver) checkConfig() error {
 }
 
 func (d *Driver) resolveIds() error {
-	if d.NetworkName != "" && !d.ComputeNetwork {
+	if len(d.NetworkNames) > 0 && !d.ComputeNetwork {
 		if err := d.initNetwork(); err != nil {
 			return err
 		}
-
+		networkIds, err := d.client.GetNetworkIDs(d)
-		networkID, err := d.client.GetNetworkID(d)
-
 		if err != nil {
 			return err
 		}
 
-		if networkID == "" {
+		if len(networkIds) == 0 {
-			return fmt.Errorf(errorUnknownNetworkName, d.NetworkName)
+			return fmt.Errorf(errorUnknownNetworkName, strings.Join(d.NetworkNames, ",")) // TODO specific name
 		}
 
-		d.NetworkId = networkID
+		d.NetworkIds = networkIds
+		for i, networkName := range d.NetworkNames {
 			log.Debug("Found network id using its name", map[string]string{
-			"Name": d.NetworkName,
+				"Name": networkName,
-			"ID":   d.NetworkId,
+				"ID":   d.NetworkIds[i],
 			})
 		}
+	}
 
 	if d.FlavorName != "" {
 		if err := d.initCompute(); err != nil {
@@ -545,7 +549,7 @@ func (d *Driver) resolveIds() error {
 		})
 	}
 
-	if d.FloatingIpPool != "" && !d.ComputeNetwork {
+	if d.FloatingIPPool != "" && !d.ComputeNetwork {
 		if err := d.initNetwork(); err != nil {
 			return err
 		}
@@ -556,13 +560,13 @@ func (d *Driver) resolveIds() error {
 		}
 
 		if f == "" {
-			return fmt.Errorf(errorUnknownNetworkName, d.FloatingIpPool)
+			return fmt.Errorf(errorUnknownNetworkName, d.FloatingIPPool)
 		}
 
-		d.FloatingIpPoolId = f
+		d.FloatingIPPoolId = f
 		log.Debug("Found floating IP pool id using its name", map[string]string{
-			"Name": d.FloatingIpPool,
+			"Name": d.FloatingIPPool,
-			"ID":   d.FloatingIpPoolId,
+			"ID":   d.FloatingIPPoolId,
 		})
 	}
 
@@ -703,7 +707,7 @@ func (d *Driver) assignFloatingIP() error {
 
 	log.Debugf("Looking for an available floating IP", map[string]string{
 		"MachineId": d.MachineId,
-		"Pool":      d.FloatingIpPool,
+		"Pool":      d.FloatingIPPool,
 	})
 
 	for _, ip := range ips {

drivers/rackspace/client.go

@@ -53,6 +53,44 @@ func (c *Client) Authenticate(d *openstack.Driver) error {
 	return nil
 }
 
+func (c *Client) InitNetworkClient(d *openstack.Driver) error {
+	if c.Network != nil {
+		return nil
+	}
+
+	network, err := rackspace.NewNetworkV2(c.Provider, gophercloud.EndpointOpts{
+		Region: d.Region,
+	})
+	if err != nil {
+		return err
+	}
+	c.Network = network
+	return nil
+}
+
+func (c *Client) GetNetworkIDs(d *openstack.Driver) ([]string, error) {
+	networkIds := make([]string, len(d.NetworkNames))
+	for i, networkName := range d.NetworkNames {
+		id, err := c.GetNetworkID(d, networkName)
+		if err != nil {
+			return nil, err
+		}
+		networkIds[i] = id
+	}
+	return networkIds, nil
+}
+
+func (c *Client) GetNetworkID(d *openstack.Driver, networkName string) (string, error) {
+	switch networkName {
+	case "public", "PublicNet":
+		return "00000000-0000-0000-0000-000000000000", nil
+	case "private", "ServiceNet":
+		return "11111111-1111-1111-1111-111111111111", nil
+	default:
+		return c.GenericClient.GetNetworkID(d, networkName)
+	}
+}
+
 // StartInstance is unfortunately not supported on Rackspace at this time.
 func (c *Client) StartInstance(d *openstack.Driver) error {
 	return unsupportedOpErr("start")

drivers/rackspace/rackspace.go

@@ -2,6 +2,7 @@ package rackspace
 
 import (
 	"fmt"
+	"strings"
 
 	"github.com/docker/machine/drivers/openstack"
 	"github.com/docker/machine/libmachine/drivers"
@@ -62,6 +63,16 @@ func (d *Driver) GetCreateFlags() []mcnflag.Flag {
 			Value:  defaultFlavorID,
 			EnvVar: "OS_FLAVOR_ID",
 		},
+		mcnflag.StringFlag{
+			Name:  "rackspace-net-id",
+			Usage: "Rackspace comma separated network ids the machine will be connected on",
+			Value: "",
+		},
+		mcnflag.StringFlag{
+			Name:  "rackspace-net-name",
+			Usage: "Rackspace comma separated network names the machine will be connected on (eg, PublicNet,ServiceNet,MyNet)",
+			Value: "",
+		},
 		mcnflag.StringFlag{
 			Name:  "rackspace-ssh-user",
 			Usage: "SSH user for the newly booted machine. Set to root by default",
@@ -117,6 +128,12 @@ func (d *Driver) SetConfigFromFlags(flags drivers.DriverOptions) error {
 	d.EndpointType = flags.String("rackspace-endpoint-type")
 	d.ImageId = flags.String("rackspace-image-id")
 	d.FlavorId = flags.String("rackspace-flavor-id")
+	if flags.String("rackspace-net-id") != "" {
+		d.NetworkIds = strings.Split(flags.String("rackspace-net-id"), ",")
+	}
+	if flags.String("rackspace-net-name") != "" {
+		d.NetworkNames = strings.Split(flags.String("rackspace-net-name"), ",")
+	}
 	d.SSHUser = flags.String("rackspace-ssh-user")
 	d.SSHPort = flags.Int("rackspace-ssh-port")
 	d.SetSwarmConfigFromFlags(flags)