25e454d18f
* Update vendor for azure vent support * Add support for Azure custom vnets. Use pointers intead of values. This allows the client to pass back returned data from Azure.
75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package azure
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net"
|
|
"strings"
|
|
|
|
"github.com/BurntSushi/toml"
|
|
"github.com/virtual-kubelet/virtual-kubelet/providers"
|
|
)
|
|
|
|
type providerConfig struct {
|
|
ResourceGroup string
|
|
Region string
|
|
OperatingSystem string
|
|
CPU string
|
|
Memory string
|
|
Pods string
|
|
SubnetName string
|
|
SubnetCIDR string
|
|
}
|
|
|
|
func (p *ACIProvider) loadConfig(r io.Reader) error {
|
|
var config providerConfig
|
|
if _, err := toml.DecodeReader(r, &config); err != nil {
|
|
return err
|
|
}
|
|
p.region = config.Region
|
|
p.resourceGroup = config.ResourceGroup
|
|
|
|
// Default to 20 mcpu
|
|
p.cpu = "20"
|
|
if config.CPU != "" {
|
|
p.cpu = config.CPU
|
|
}
|
|
// Default to 100Gi
|
|
p.memory = "100Gi"
|
|
if config.Memory != "" {
|
|
p.memory = config.Memory
|
|
}
|
|
// Default to 20 pods
|
|
p.pods = "20"
|
|
if config.Pods != "" {
|
|
p.pods = config.Pods
|
|
}
|
|
|
|
// Default to Linux if the operating system was not defined in the config.
|
|
if config.OperatingSystem == "" {
|
|
config.OperatingSystem = providers.OperatingSystemLinux
|
|
} else {
|
|
// Validate operating system from config.
|
|
ok, _ := providers.ValidOperatingSystems[config.OperatingSystem]
|
|
if !ok {
|
|
return fmt.Errorf("%q is not a valid operating system, try one of the following instead: %s", config.OperatingSystem, strings.Join(providers.ValidOperatingSystems.Names(), " | "))
|
|
}
|
|
}
|
|
|
|
// default subnet name
|
|
if config.SubnetName != "" {
|
|
p.subnetName = config.SubnetName
|
|
}
|
|
if config.SubnetCIDR != "" {
|
|
if config.SubnetName == "" {
|
|
return fmt.Errorf("subnet CIDR is set but no subnet name provided, must provide a subnet name in order to set a subnet CIDR")
|
|
}
|
|
if _, _, err := net.ParseCIDR(config.SubnetCIDR); err != nil {
|
|
return fmt.Errorf("error parsing provided subnet CIDR: %v", err)
|
|
}
|
|
}
|
|
|
|
p.operatingSystem = config.OperatingSystem
|
|
return nil
|
|
}
|