2019-09-27 21:51:53 +00:00
// +build !providerless
/ *
Copyright 2016 The Kubernetes Authors .
Licensed under the Apache License , Version 2.0 ( the "License" ) ;
you may not use this file except in compliance with the License .
You may obtain a copy of the License at
http : //www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing , software
distributed under the License is distributed on an "AS IS" BASIS ,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND , either express or implied .
See the License for the specific language governing permissions and
limitations under the License .
* /
package vsphere
import (
"context"
"errors"
"fmt"
"io"
"net"
"net/url"
"os"
"path"
"path/filepath"
2020-08-10 17:43:49 +00:00
"reflect"
2019-09-27 21:51:53 +00:00
"runtime"
"strings"
"sync"
"time"
"gopkg.in/gcfg.v1"
"github.com/vmware/govmomi/object"
"github.com/vmware/govmomi/vapi/rest"
"github.com/vmware/govmomi/vapi/tags"
"github.com/vmware/govmomi/vim25/mo"
vmwaretypes "github.com/vmware/govmomi/vim25/types"
v1 "k8s.io/api/core/v1"
k8stypes "k8s.io/apimachinery/pkg/types"
"k8s.io/apimachinery/pkg/util/sets"
"k8s.io/client-go/informers"
"k8s.io/client-go/tools/cache"
cloudprovider "k8s.io/cloud-provider"
nodehelpers "k8s.io/cloud-provider/node/helpers"
volumehelpers "k8s.io/cloud-provider/volume/helpers"
2020-08-10 17:43:49 +00:00
"k8s.io/klog/v2"
2019-09-27 21:51:53 +00:00
"k8s.io/legacy-cloud-providers/vsphere/vclib"
"k8s.io/legacy-cloud-providers/vsphere/vclib/diskmanagers"
)
// VSphere Cloud Provider constants
const (
ProviderName = "vsphere"
VolDir = "kubevols"
RoundTripperDefaultCount = 3
DummyVMPrefixName = "vsphere-k8s"
CleanUpDummyVMRoutineInterval = 5
)
var cleanUpRoutineInitialized = false
var datastoreFolderIDMap = make ( map [ string ] map [ string ] string )
var cleanUpRoutineInitLock sync . Mutex
var cleanUpDummyVMLock sync . RWMutex
// Error Messages
const (
MissingUsernameErrMsg = "Username is missing"
MissingPasswordErrMsg = "Password is missing"
NoZoneTagInVCErrMsg = "No zone tags found in vCenter"
)
// Error constants
var (
ErrUsernameMissing = errors . New ( MissingUsernameErrMsg )
ErrPasswordMissing = errors . New ( MissingPasswordErrMsg )
ErrNoZoneTagInVC = errors . New ( NoZoneTagInVCErrMsg )
)
var _ cloudprovider . Interface = ( * VSphere ) ( nil )
var _ cloudprovider . Instances = ( * VSphere ) ( nil )
var _ cloudprovider . Zones = ( * VSphere ) ( nil )
var _ cloudprovider . PVLabeler = ( * VSphere ) ( nil )
// VSphere is an implementation of cloud provider Interface for VSphere.
type VSphere struct {
cfg * VSphereConfig
hostName string
// Maps the VSphere IP address to VSphereInstance
vsphereInstanceMap map [ string ] * VSphereInstance
// Responsible for managing discovery of k8s node, their location etc.
nodeManager * NodeManager
vmUUID string
isSecretInfoProvided bool
2020-08-10 17:43:49 +00:00
isSecretManaged bool
2019-09-27 21:51:53 +00:00
}
// Represents a vSphere instance where one or more kubernetes nodes are running.
type VSphereInstance struct {
conn * vclib . VSphereConnection
cfg * VirtualCenterConfig
}
// Structure that represents Virtual Center configuration
type VirtualCenterConfig struct {
// vCenter username.
User string ` gcfg:"user" `
// vCenter password in clear text.
Password string ` gcfg:"password" `
// vCenter port.
VCenterPort string ` gcfg:"port" `
// Datacenter in which VMs are located.
Datacenters string ` gcfg:"datacenters" `
// Soap round tripper count (retries = RoundTripper - 1)
RoundTripperCount uint ` gcfg:"soap-roundtrip-count" `
// Thumbprint of the VCenter's certificate thumbprint
Thumbprint string ` gcfg:"thumbprint" `
}
// Structure that represents the content of vsphere.conf file.
// Users specify the configuration of one or more Virtual Centers in vsphere.conf where
// the Kubernetes master and worker nodes are running.
// NOTE: Cloud config files should follow the same Kubernetes deprecation policy as
// flags or CLIs. Config fields should not change behavior in incompatible ways and
// should be deprecated for at least 2 release prior to removing.
// See https://kubernetes.io/docs/reference/using-api/deprecation-policy/#deprecating-a-flag-or-cli
// for more details.
type VSphereConfig struct {
Global struct {
// vCenter username.
User string ` gcfg:"user" `
// vCenter password in clear text.
Password string ` gcfg:"password" `
// Deprecated. Use VirtualCenter to specify multiple vCenter Servers.
// vCenter IP.
VCenterIP string ` gcfg:"server" `
// vCenter port.
VCenterPort string ` gcfg:"port" `
// True if vCenter uses self-signed cert.
InsecureFlag bool ` gcfg:"insecure-flag" `
// Specifies the path to a CA certificate in PEM format. Optional; if not
// configured, the system's CA certificates will be used.
CAFile string ` gcfg:"ca-file" `
// Thumbprint of the VCenter's certificate thumbprint
Thumbprint string ` gcfg:"thumbprint" `
// Datacenter in which VMs are located.
// Deprecated. Use "datacenters" instead.
Datacenter string ` gcfg:"datacenter" `
// Datacenter in which VMs are located.
Datacenters string ` gcfg:"datacenters" `
// Datastore in which vmdks are stored.
// Deprecated. See Workspace.DefaultDatastore
DefaultDatastore string ` gcfg:"datastore" `
// WorkingDir is path where VMs can be found. Also used to create dummy VMs.
// Deprecated.
WorkingDir string ` gcfg:"working-dir" `
// Soap round tripper count (retries = RoundTripper - 1)
RoundTripperCount uint ` gcfg:"soap-roundtrip-count" `
// Is required on the controller-manager if it does not run on a VMware machine
// VMUUID is the VM Instance UUID of virtual machine which can be retrieved from instanceUuid
// property in VmConfigInfo, or also set as vc.uuid in VMX file.
// If not set, will be fetched from the machine via sysfs (requires root)
VMUUID string ` gcfg:"vm-uuid" `
// Deprecated as virtual machine will be automatically discovered.
// VMName is the VM name of virtual machine
// Combining the WorkingDir and VMName can form a unique InstanceID.
// When vm-name is set, no username/password is required on worker nodes.
VMName string ` gcfg:"vm-name" `
// Name of the secret were vCenter credentials are present.
SecretName string ` gcfg:"secret-name" `
// Secret Namespace where secret will be present that has vCenter credentials.
SecretNamespace string ` gcfg:"secret-namespace" `
2020-08-10 17:43:49 +00:00
// Secret changes being ingnored for cloud resources
SecretNotManaged bool ` gcfg:"secret-not-managed" `
2019-09-27 21:51:53 +00:00
}
VirtualCenter map [ string ] * VirtualCenterConfig
Network struct {
// PublicNetwork is name of the network the VMs are joined to.
PublicNetwork string ` gcfg:"public-network" `
}
Disk struct {
// SCSIControllerType defines SCSI controller to be used.
SCSIControllerType string ` dcfg:"scsicontrollertype" `
}
// Endpoint used to create volumes
Workspace struct {
VCenterIP string ` gcfg:"server" `
Datacenter string ` gcfg:"datacenter" `
Folder string ` gcfg:"folder" `
DefaultDatastore string ` gcfg:"default-datastore" `
ResourcePoolPath string ` gcfg:"resourcepool-path" `
}
// Tag categories and tags which correspond to "built-in node labels: zones and region"
Labels struct {
Zone string ` gcfg:"zone" `
Region string ` gcfg:"region" `
}
}
type Volumes interface {
// AttachDisk attaches given disk to given node. Current node
// is used when nodeName is empty string.
AttachDisk ( vmDiskPath string , storagePolicyName string , nodeName k8stypes . NodeName ) ( diskUUID string , err error )
// DetachDisk detaches given disk to given node. Current node
// is used when nodeName is empty string.
// Assumption: If node doesn't exist, disk is already detached from node.
DetachDisk ( volPath string , nodeName k8stypes . NodeName ) error
// DiskIsAttached checks if a disk is attached to the given node.
// Assumption: If node doesn't exist, disk is not attached to the node.
DiskIsAttached ( volPath string , nodeName k8stypes . NodeName ) ( bool , error )
// DisksAreAttached checks if a list disks are attached to the given node.
// Assumption: If node doesn't exist, disks are not attached to the node.
DisksAreAttached ( nodeVolumes map [ k8stypes . NodeName ] [ ] string ) ( map [ k8stypes . NodeName ] map [ string ] bool , error )
// CreateVolume creates a new vmdk with specified parameters.
CreateVolume ( volumeOptions * vclib . VolumeOptions ) ( volumePath string , err error )
// DeleteVolume deletes vmdk.
DeleteVolume ( vmDiskPath string ) error
}
// Parses vSphere cloud config file and stores it into VSphereConfig.
func readConfig ( config io . Reader ) ( VSphereConfig , error ) {
if config == nil {
err := fmt . Errorf ( "no vSphere cloud provider config file given" )
return VSphereConfig { } , err
}
var cfg VSphereConfig
err := gcfg . ReadInto ( & cfg , config )
return cfg , err
}
func init ( ) {
vclib . RegisterMetrics ( )
cloudprovider . RegisterCloudProvider ( ProviderName , func ( config io . Reader ) ( cloudprovider . Interface , error ) {
// If vSphere.conf file is not present then it is worker node.
if config == nil {
return newWorkerNode ( )
}
cfg , err := readConfig ( config )
if err != nil {
return nil , err
}
return newControllerNode ( cfg )
} )
}
// Initialize passes a Kubernetes clientBuilder interface to the cloud provider
func ( vs * VSphere ) Initialize ( clientBuilder cloudprovider . ControllerClientBuilder , stop <- chan struct { } ) {
}
// Initialize Node Informers
func ( vs * VSphere ) SetInformers ( informerFactory informers . SharedInformerFactory ) {
if vs . cfg == nil {
return
}
if vs . isSecretInfoProvided {
secretCredentialManager := & SecretCredentialManager {
SecretName : vs . cfg . Global . SecretName ,
SecretNamespace : vs . cfg . Global . SecretNamespace ,
SecretLister : informerFactory . Core ( ) . V1 ( ) . Secrets ( ) . Lister ( ) ,
Cache : & SecretCache {
VirtualCenter : make ( map [ string ] * Credential ) ,
} ,
}
2020-08-10 17:43:49 +00:00
if vs . isSecretManaged {
klog . V ( 4 ) . Infof ( "Setting up secret informers for vSphere Cloud Provider" )
secretInformer := informerFactory . Core ( ) . V1 ( ) . Secrets ( ) . Informer ( )
secretInformer . AddEventHandler ( cache . ResourceEventHandlerFuncs {
AddFunc : vs . SecretAdded ,
UpdateFunc : vs . SecretUpdated ,
} )
klog . V ( 4 ) . Infof ( "Secret informers in vSphere cloud provider initialized" )
}
2019-09-27 21:51:53 +00:00
vs . nodeManager . UpdateCredentialManager ( secretCredentialManager )
}
// Only on controller node it is required to register listeners.
// Register callbacks for node updates
klog . V ( 4 ) . Infof ( "Setting up node informers for vSphere Cloud Provider" )
nodeInformer := informerFactory . Core ( ) . V1 ( ) . Nodes ( ) . Informer ( )
nodeInformer . AddEventHandler ( cache . ResourceEventHandlerFuncs {
AddFunc : vs . NodeAdded ,
DeleteFunc : vs . NodeDeleted ,
} )
klog . V ( 4 ) . Infof ( "Node informers in vSphere cloud provider initialized" )
}
// Creates new worker node interface and returns
func newWorkerNode ( ) ( * VSphere , error ) {
var err error
vs := VSphere { }
vs . hostName , err = os . Hostname ( )
if err != nil {
klog . Errorf ( "Failed to get hostname. err: %+v" , err )
return nil , err
}
vs . vmUUID , err = GetVMUUID ( )
if err != nil {
klog . Errorf ( "Failed to get uuid. err: %+v" , err )
return nil , err
}
return & vs , nil
}
func populateVsphereInstanceMap ( cfg * VSphereConfig ) ( map [ string ] * VSphereInstance , error ) {
vsphereInstanceMap := make ( map [ string ] * VSphereInstance )
isSecretInfoProvided := true
if cfg . Global . SecretName == "" || cfg . Global . SecretNamespace == "" {
klog . Warningf ( "SecretName and/or SecretNamespace is not provided. " +
"VCP will use username and password from config file" )
isSecretInfoProvided = false
}
if isSecretInfoProvided {
if cfg . Global . User != "" {
klog . Warning ( "Global.User and Secret info provided. VCP will use secret to get credentials" )
cfg . Global . User = ""
}
if cfg . Global . Password != "" {
klog . Warning ( "Global.Password and Secret info provided. VCP will use secret to get credentials" )
cfg . Global . Password = ""
}
}
// Check if the vsphere.conf is in old format. In this
// format the cfg.VirtualCenter will be nil or empty.
if cfg . VirtualCenter == nil || len ( cfg . VirtualCenter ) == 0 {
klog . V ( 4 ) . Infof ( "Config is not per virtual center and is in old format." )
if ! isSecretInfoProvided {
if cfg . Global . User == "" {
klog . Error ( "Global.User is empty!" )
return nil , ErrUsernameMissing
}
if cfg . Global . Password == "" {
klog . Error ( "Global.Password is empty!" )
return nil , ErrPasswordMissing
}
}
if cfg . Global . WorkingDir == "" {
klog . Error ( "Global.WorkingDir is empty!" )
return nil , errors . New ( "Global.WorkingDir is empty!" )
}
if cfg . Global . VCenterIP == "" {
klog . Error ( "Global.VCenterIP is empty!" )
return nil , errors . New ( "Global.VCenterIP is empty!" )
}
if cfg . Global . Datacenter == "" {
klog . Error ( "Global.Datacenter is empty!" )
return nil , errors . New ( "Global.Datacenter is empty!" )
}
cfg . Workspace . VCenterIP = cfg . Global . VCenterIP
cfg . Workspace . Datacenter = cfg . Global . Datacenter
cfg . Workspace . Folder = cfg . Global . WorkingDir
cfg . Workspace . DefaultDatastore = cfg . Global . DefaultDatastore
vcConfig := VirtualCenterConfig {
User : cfg . Global . User ,
Password : cfg . Global . Password ,
VCenterPort : cfg . Global . VCenterPort ,
Datacenters : cfg . Global . Datacenter ,
RoundTripperCount : cfg . Global . RoundTripperCount ,
Thumbprint : cfg . Global . Thumbprint ,
}
// Note: If secrets info is provided username and password will be populated
// once secret is created.
vSphereConn := vclib . VSphereConnection {
Username : vcConfig . User ,
Password : vcConfig . Password ,
Hostname : cfg . Global . VCenterIP ,
Insecure : cfg . Global . InsecureFlag ,
RoundTripperCount : vcConfig . RoundTripperCount ,
Port : vcConfig . VCenterPort ,
CACert : cfg . Global . CAFile ,
Thumbprint : cfg . Global . Thumbprint ,
}
vsphereIns := VSphereInstance {
conn : & vSphereConn ,
cfg : & vcConfig ,
}
vsphereInstanceMap [ cfg . Global . VCenterIP ] = & vsphereIns
} else {
if cfg . Workspace . VCenterIP == "" || cfg . Workspace . Folder == "" || cfg . Workspace . Datacenter == "" {
msg := fmt . Sprintf ( "All fields in workspace are mandatory." +
" vsphere.conf does not have the workspace specified correctly. cfg.Workspace: %+v" , cfg . Workspace )
klog . Error ( msg )
return nil , errors . New ( msg )
}
for vcServer , vcConfig := range cfg . VirtualCenter {
klog . V ( 4 ) . Infof ( "Initializing vc server %s" , vcServer )
if vcServer == "" {
klog . Error ( "vsphere.conf does not have the VirtualCenter IP address specified" )
return nil , errors . New ( "vsphere.conf does not have the VirtualCenter IP address specified" )
}
if ! isSecretInfoProvided {
if vcConfig . User == "" {
vcConfig . User = cfg . Global . User
if vcConfig . User == "" {
klog . Errorf ( "vcConfig.User is empty for vc %s!" , vcServer )
return nil , ErrUsernameMissing
}
}
if vcConfig . Password == "" {
vcConfig . Password = cfg . Global . Password
if vcConfig . Password == "" {
klog . Errorf ( "vcConfig.Password is empty for vc %s!" , vcServer )
return nil , ErrPasswordMissing
}
}
} else {
if vcConfig . User != "" {
klog . Warningf ( "vcConfig.User for server %s and Secret info provided. VCP will use secret to get credentials" , vcServer )
vcConfig . User = ""
}
if vcConfig . Password != "" {
klog . Warningf ( "vcConfig.Password for server %s and Secret info provided. VCP will use secret to get credentials" , vcServer )
vcConfig . Password = ""
}
}
if vcConfig . VCenterPort == "" {
vcConfig . VCenterPort = cfg . Global . VCenterPort
}
if vcConfig . Datacenters == "" {
if cfg . Global . Datacenters != "" {
vcConfig . Datacenters = cfg . Global . Datacenters
} else {
// cfg.Global.Datacenter is deprecated, so giving it the last preference.
vcConfig . Datacenters = cfg . Global . Datacenter
}
}
if vcConfig . RoundTripperCount == 0 {
vcConfig . RoundTripperCount = cfg . Global . RoundTripperCount
}
// Note: If secrets info is provided username and password will be populated
// once secret is created.
vSphereConn := vclib . VSphereConnection {
Username : vcConfig . User ,
Password : vcConfig . Password ,
Hostname : vcServer ,
Insecure : cfg . Global . InsecureFlag ,
RoundTripperCount : vcConfig . RoundTripperCount ,
Port : vcConfig . VCenterPort ,
CACert : cfg . Global . CAFile ,
Thumbprint : vcConfig . Thumbprint ,
}
vsphereIns := VSphereInstance {
conn : & vSphereConn ,
cfg : vcConfig ,
}
vsphereInstanceMap [ vcServer ] = & vsphereIns
}
}
return vsphereInstanceMap , nil
}
// getVMUUID allows tests to override GetVMUUID
var getVMUUID = GetVMUUID
// Creates new Controller node interface and returns
func newControllerNode ( cfg VSphereConfig ) ( * VSphere , error ) {
vs , err := buildVSphereFromConfig ( cfg )
if err != nil {
return nil , err
}
vs . hostName , err = os . Hostname ( )
if err != nil {
klog . Errorf ( "Failed to get hostname. err: %+v" , err )
return nil , err
}
if cfg . Global . VMUUID != "" {
vs . vmUUID = cfg . Global . VMUUID
} else {
vs . vmUUID , err = getVMUUID ( )
if err != nil {
klog . Errorf ( "Failed to get uuid. err: %+v" , err )
return nil , err
}
}
runtime . SetFinalizer ( vs , logout )
return vs , nil
}
// Initializes vSphere from vSphere CloudProvider Configuration
func buildVSphereFromConfig ( cfg VSphereConfig ) ( * VSphere , error ) {
isSecretInfoProvided := false
if cfg . Global . SecretName != "" && cfg . Global . SecretNamespace != "" {
isSecretInfoProvided = true
}
if cfg . Disk . SCSIControllerType == "" {
cfg . Disk . SCSIControllerType = vclib . PVSCSIControllerType
} else if ! vclib . CheckControllerSupported ( cfg . Disk . SCSIControllerType ) {
klog . Errorf ( "%v is not a supported SCSI Controller type. Please configure 'lsilogic-sas' OR 'pvscsi'" , cfg . Disk . SCSIControllerType )
return nil , errors . New ( "Controller type not supported. Please configure 'lsilogic-sas' OR 'pvscsi'" )
}
if cfg . Global . WorkingDir != "" {
cfg . Global . WorkingDir = path . Clean ( cfg . Global . WorkingDir )
}
if cfg . Global . RoundTripperCount == 0 {
cfg . Global . RoundTripperCount = RoundTripperDefaultCount
}
if cfg . Global . VCenterPort == "" {
cfg . Global . VCenterPort = "443"
}
vsphereInstanceMap , err := populateVsphereInstanceMap ( & cfg )
if err != nil {
return nil , err
}
vs := VSphere {
vsphereInstanceMap : vsphereInstanceMap ,
nodeManager : & NodeManager {
vsphereInstanceMap : vsphereInstanceMap ,
nodeInfoMap : make ( map [ string ] * NodeInfo ) ,
registeredNodes : make ( map [ string ] * v1 . Node ) ,
} ,
isSecretInfoProvided : isSecretInfoProvided ,
2020-08-10 17:43:49 +00:00
isSecretManaged : ! cfg . Global . SecretNotManaged ,
2019-09-27 21:51:53 +00:00
cfg : & cfg ,
}
return & vs , nil
}
func logout ( vs * VSphere ) {
for _ , vsphereIns := range vs . vsphereInstanceMap {
vsphereIns . conn . Logout ( context . TODO ( ) )
}
}
// Instances returns an implementation of Instances for vSphere.
func ( vs * VSphere ) Instances ( ) ( cloudprovider . Instances , bool ) {
return vs , true
}
2020-08-10 17:43:49 +00:00
// InstancesV2 returns an implementation of InstancesV2 for vSphere.
// TODO: implement ONLY for external cloud provider
func ( vs * VSphere ) InstancesV2 ( ) ( cloudprovider . InstancesV2 , bool ) {
return nil , false
}
2019-09-27 21:51:53 +00:00
func getLocalIP ( ) ( [ ] v1 . NodeAddress , error ) {
// hashtable with VMware-allocated OUIs for MAC filtering
// List of official OUIs: http://standards-oui.ieee.org/oui.txt
vmwareOUI := map [ string ] bool {
"00:05:69" : true ,
"00:0c:29" : true ,
"00:1c:14" : true ,
"00:50:56" : true ,
}
addrs := [ ] v1 . NodeAddress { }
ifaces , err := net . Interfaces ( )
if err != nil {
klog . Errorf ( "net.Interfaces() failed for NodeAddresses - %v" , err )
return nil , err
}
for _ , i := range ifaces {
if i . Flags & net . FlagLoopback != 0 {
continue
}
localAddrs , err := i . Addrs ( )
if err != nil {
klog . Warningf ( "Failed to extract addresses for NodeAddresses - %v" , err )
} else {
for _ , addr := range localAddrs {
if ipnet , ok := addr . ( * net . IPNet ) ; ok {
2020-01-15 01:52:20 +00:00
if ! ipnet . IP . IsLinkLocalUnicast ( ) {
2019-09-27 21:51:53 +00:00
// Filter external IP by MAC address OUIs from vCenter and from ESX
vmMACAddr := strings . ToLower ( i . HardwareAddr . String ( ) )
// Making sure that the MAC address is long enough
if len ( vmMACAddr ) < 17 {
klog . V ( 4 ) . Infof ( "Skipping invalid MAC address: %q" , vmMACAddr )
continue
}
if vmwareOUI [ vmMACAddr [ : 8 ] ] {
nodehelpers . AddToNodeAddresses ( & addrs ,
v1 . NodeAddress {
Type : v1 . NodeExternalIP ,
Address : ipnet . IP . String ( ) ,
} ,
v1 . NodeAddress {
Type : v1 . NodeInternalIP ,
Address : ipnet . IP . String ( ) ,
} ,
)
klog . V ( 4 ) . Infof ( "Detected local IP address as %q" , ipnet . IP . String ( ) )
} else {
klog . V ( 4 ) . Infof ( "Failed to patch IP for interface %q as MAC address %q does not belong to a VMware platform" , i . Name , vmMACAddr )
}
}
}
}
}
}
return addrs , nil
}
func ( vs * VSphere ) getVSphereInstance ( nodeName k8stypes . NodeName ) ( * VSphereInstance , error ) {
vsphereIns , err := vs . nodeManager . GetVSphereInstance ( nodeName )
if err != nil {
klog . Errorf ( "Cannot find node %q in cache. Node not found!!!" , nodeName )
return nil , err
}
return & vsphereIns , nil
}
func ( vs * VSphere ) getVSphereInstanceForServer ( vcServer string , ctx context . Context ) ( * VSphereInstance , error ) {
vsphereIns , ok := vs . vsphereInstanceMap [ vcServer ]
if ! ok {
klog . Errorf ( "cannot find vcServer %q in cache. VC not found!!!" , vcServer )
return nil , fmt . Errorf ( "cannot find node %q in vsphere configuration map" , vcServer )
}
// Ensure client is logged in and session is valid
err := vs . nodeManager . vcConnect ( ctx , vsphereIns )
if err != nil {
klog . Errorf ( "failed connecting to vcServer %q with error %+v" , vcServer , err )
return nil , err
}
return vsphereIns , nil
}
// Get the VM Managed Object instance by from the node
func ( vs * VSphere ) getVMFromNodeName ( ctx context . Context , nodeName k8stypes . NodeName ) ( * vclib . VirtualMachine , error ) {
nodeInfo , err := vs . nodeManager . GetNodeInfo ( nodeName )
if err != nil {
return nil , err
}
return nodeInfo . vm , nil
}
// NodeAddresses is an implementation of Instances.NodeAddresses.
func ( vs * VSphere ) NodeAddresses ( ctx context . Context , nodeName k8stypes . NodeName ) ( [ ] v1 . NodeAddress , error ) {
if vs . hostName == convertToString ( nodeName ) {
2020-08-10 17:43:49 +00:00
return vs . getNodeAddressesFromLocalIP ( )
2019-09-27 21:51:53 +00:00
}
2020-08-10 17:43:49 +00:00
return vs . getNodeAddressesFromVM ( ctx , nodeName )
}
2019-09-27 21:51:53 +00:00
2020-08-10 17:43:49 +00:00
// getNodeAddressesFromLocalIP get local IP addresses if node is local node.
func ( vs * VSphere ) getNodeAddressesFromLocalIP ( ) ( [ ] v1 . NodeAddress , error ) {
addrs , err := getLocalIP ( )
if err != nil {
return nil , err
}
// add the hostname address
nodehelpers . AddToNodeAddresses ( & addrs , v1 . NodeAddress { Type : v1 . NodeHostName , Address : vs . hostName } )
return addrs , nil
}
// getNodeAddressesFromVM get vm IP addresses if node is vm.
func ( vs * VSphere ) getNodeAddressesFromVM ( ctx context . Context , nodeName k8stypes . NodeName ) ( [ ] v1 . NodeAddress , error ) {
2019-09-27 21:51:53 +00:00
if vs . cfg == nil {
return nil , cloudprovider . InstanceNotFound
}
// Below logic can be executed only on master as VC details are present.
addrs := [ ] v1 . NodeAddress { }
// Create context
ctx , cancel := context . WithCancel ( ctx )
defer cancel ( )
vsi , err := vs . getVSphereInstance ( nodeName )
if err != nil {
return nil , err
}
// Ensure client is logged in and session is valid
err = vs . nodeManager . vcConnect ( ctx , vsi )
if err != nil {
return nil , err
}
vm , err := vs . getVMFromNodeName ( ctx , nodeName )
if err != nil {
klog . Errorf ( "Failed to get VM object for node: %q. err: +%v" , convertToString ( nodeName ) , err )
return nil , err
}
vmMoList , err := vm . Datacenter . GetVMMoList ( ctx , [ ] * vclib . VirtualMachine { vm } , [ ] string { "guest.net" } )
if err != nil {
klog . Errorf ( "Failed to get VM Managed object with property guest.net for node: %q. err: +%v" , convertToString ( nodeName ) , err )
return nil , err
}
// retrieve VM's ip(s)
for _ , v := range vmMoList [ 0 ] . Guest . Net {
if vs . cfg . Network . PublicNetwork == v . Network {
for _ , ip := range v . IpAddress {
2020-01-15 01:52:20 +00:00
if ! net . ParseIP ( ip ) . IsLinkLocalUnicast ( ) {
2019-09-27 21:51:53 +00:00
nodehelpers . AddToNodeAddresses ( & addrs ,
v1 . NodeAddress {
Type : v1 . NodeExternalIP ,
Address : ip ,
} , v1 . NodeAddress {
Type : v1 . NodeInternalIP ,
Address : ip ,
} ,
)
}
}
}
}
return addrs , nil
}
// NodeAddressesByProviderID returns the node addresses of an instances with the specified unique providerID
// This method will not be called from the node that is requesting this ID. i.e. metadata service
// and other local methods cannot be used here
func ( vs * VSphere ) NodeAddressesByProviderID ( ctx context . Context , providerID string ) ( [ ] v1 . NodeAddress , error ) {
return vs . NodeAddresses ( ctx , convertToK8sType ( providerID ) )
}
// AddSSHKeyToAllInstances add SSH key to all instances
func ( vs * VSphere ) AddSSHKeyToAllInstances ( ctx context . Context , user string , keyData [ ] byte ) error {
return cloudprovider . NotImplemented
}
// CurrentNodeName gives the current node name
func ( vs * VSphere ) CurrentNodeName ( ctx context . Context , hostname string ) ( k8stypes . NodeName , error ) {
return convertToK8sType ( vs . hostName ) , nil
}
func convertToString ( nodeName k8stypes . NodeName ) string {
return string ( nodeName )
}
func convertToK8sType ( vmName string ) k8stypes . NodeName {
return k8stypes . NodeName ( vmName )
}
// InstanceExistsByProviderID returns true if the instance with the given provider id still exists and is running.
// If false is returned with no error, the instance will be immediately deleted by the cloud controller manager.
func ( vs * VSphere ) InstanceExistsByProviderID ( ctx context . Context , providerID string ) ( bool , error ) {
nodeName , err := vs . GetNodeNameFromProviderID ( providerID )
if err != nil {
klog . Errorf ( "Error while getting nodename for providerID %s" , providerID )
return false , err
}
_ , err = vs . InstanceID ( ctx , convertToK8sType ( nodeName ) )
if err == nil {
return true , nil
}
return false , err
}
// InstanceShutdownByProviderID returns true if the instance is in safe state to detach volumes
func ( vs * VSphere ) InstanceShutdownByProviderID ( ctx context . Context , providerID string ) ( bool , error ) {
nodeName , err := vs . GetNodeNameFromProviderID ( providerID )
if err != nil {
klog . Errorf ( "Error while getting nodename for providerID %s" , providerID )
return false , err
}
vsi , err := vs . getVSphereInstance ( convertToK8sType ( nodeName ) )
if err != nil {
return false , err
}
// Ensure client is logged in and session is valid
if err := vs . nodeManager . vcConnect ( ctx , vsi ) ; err != nil {
return false , err
}
vm , err := vs . getVMFromNodeName ( ctx , convertToK8sType ( nodeName ) )
if err != nil {
klog . Errorf ( "Failed to get VM object for node: %q. err: +%v" , nodeName , err )
return false , err
}
isActive , err := vm . IsActive ( ctx )
if err != nil {
klog . Errorf ( "Failed to check whether node %q is active. err: %+v." , nodeName , err )
return false , err
}
return ! isActive , nil
}
// InstanceID returns the cloud provider ID of the node with the specified Name.
func ( vs * VSphere ) InstanceID ( ctx context . Context , nodeName k8stypes . NodeName ) ( string , error ) {
instanceIDInternal := func ( ) ( string , error ) {
if vs . hostName == convertToString ( nodeName ) {
return vs . vmUUID , nil
}
// Below logic can be performed only on master node where VC details are preset.
if vs . cfg == nil {
return "" , fmt . Errorf ( "The current node can't determine InstanceID for %q" , convertToString ( nodeName ) )
}
// Create context
ctx , cancel := context . WithCancel ( context . Background ( ) )
defer cancel ( )
vsi , err := vs . getVSphereInstance ( nodeName )
if err != nil {
return "" , err
}
// Ensure client is logged in and session is valid
err = vs . nodeManager . vcConnect ( ctx , vsi )
if err != nil {
return "" , err
}
vm , err := vs . getVMFromNodeName ( ctx , nodeName )
if err != nil {
klog . Errorf ( "Failed to get VM object for node: %q. err: +%v" , convertToString ( nodeName ) , err )
return "" , err
}
isActive , err := vm . IsActive ( ctx )
if err != nil {
klog . Errorf ( "Failed to check whether node %q is active. err: %+v." , convertToString ( nodeName ) , err )
return "" , err
}
if isActive {
return vs . vmUUID , nil
}
klog . Warningf ( "The VM: %s is not in %s state" , convertToString ( nodeName ) , vclib . ActivePowerState )
return "" , cloudprovider . InstanceNotFound
}
instanceID , err := instanceIDInternal ( )
if err != nil {
if vclib . IsManagedObjectNotFoundError ( err ) {
err = vs . nodeManager . RediscoverNode ( nodeName )
if err == nil {
klog . V ( 4 ) . Infof ( "InstanceID: Found node %q" , convertToString ( nodeName ) )
instanceID , err = instanceIDInternal ( )
} else if err == vclib . ErrNoVMFound {
return "" , cloudprovider . InstanceNotFound
}
}
}
return instanceID , err
}
// InstanceTypeByProviderID returns the cloudprovider instance type of the node with the specified unique providerID
// This method will not be called from the node that is requesting this ID. i.e. metadata service
// and other local methods cannot be used here
func ( vs * VSphere ) InstanceTypeByProviderID ( ctx context . Context , providerID string ) ( string , error ) {
return "" , nil
}
func ( vs * VSphere ) InstanceType ( ctx context . Context , name k8stypes . NodeName ) ( string , error ) {
return "" , nil
}
func ( vs * VSphere ) Clusters ( ) ( cloudprovider . Clusters , bool ) {
return nil , true
}
// ProviderName returns the cloud provider ID.
func ( vs * VSphere ) ProviderName ( ) string {
return ProviderName
}
// LoadBalancer returns an implementation of LoadBalancer for vSphere.
func ( vs * VSphere ) LoadBalancer ( ) ( cloudprovider . LoadBalancer , bool ) {
return nil , false
}
func ( vs * VSphere ) isZoneEnabled ( ) bool {
return vs . cfg != nil && vs . cfg . Labels . Zone != "" && vs . cfg . Labels . Region != ""
}
// Zones returns an implementation of Zones for vSphere.
func ( vs * VSphere ) Zones ( ) ( cloudprovider . Zones , bool ) {
if vs . isZoneEnabled ( ) {
return vs , true
}
klog . V ( 1 ) . Info ( "The vSphere cloud provider does not support zones" )
return nil , false
}
// Routes returns a false since the interface is not supported for vSphere.
func ( vs * VSphere ) Routes ( ) ( cloudprovider . Routes , bool ) {
return nil , false
}
// AttachDisk attaches given virtual disk volume to the compute running kubelet.
func ( vs * VSphere ) AttachDisk ( vmDiskPath string , storagePolicyName string , nodeName k8stypes . NodeName ) ( diskUUID string , err error ) {
attachDiskInternal := func ( vmDiskPath string , storagePolicyName string , nodeName k8stypes . NodeName ) ( diskUUID string , err error ) {
if nodeName == "" {
nodeName = convertToK8sType ( vs . hostName )
}
// Create context
ctx , cancel := context . WithCancel ( context . Background ( ) )
defer cancel ( )
vsi , err := vs . getVSphereInstance ( nodeName )
if err != nil {
return "" , err
}
// Ensure client is logged in and session is valid
err = vs . nodeManager . vcConnect ( ctx , vsi )
if err != nil {
return "" , err
}
vm , err := vs . getVMFromNodeName ( ctx , nodeName )
if err != nil {
klog . Errorf ( "Failed to get VM object for node: %q. err: +%v" , convertToString ( nodeName ) , err )
return "" , err
}
diskUUID , err = vm . AttachDisk ( ctx , vmDiskPath , & vclib . VolumeOptions { SCSIControllerType : vclib . PVSCSIControllerType , StoragePolicyName : storagePolicyName } )
if err != nil {
klog . Errorf ( "Failed to attach disk: %s for node: %s. err: +%v" , vmDiskPath , convertToString ( nodeName ) , err )
return "" , err
}
return diskUUID , nil
}
requestTime := time . Now ( )
diskUUID , err = attachDiskInternal ( vmDiskPath , storagePolicyName , nodeName )
if err != nil {
if vclib . IsManagedObjectNotFoundError ( err ) {
err = vs . nodeManager . RediscoverNode ( nodeName )
if err == nil {
klog . V ( 4 ) . Infof ( "AttachDisk: Found node %q" , convertToString ( nodeName ) )
diskUUID , err = attachDiskInternal ( vmDiskPath , storagePolicyName , nodeName )
klog . V ( 4 ) . Infof ( "AttachDisk: Retry: diskUUID %s, err +%v" , diskUUID , err )
}
}
}
klog . V ( 4 ) . Infof ( "AttachDisk executed for node %s and volume %s with diskUUID %s. Err: %s" , convertToString ( nodeName ) , vmDiskPath , diskUUID , err )
vclib . RecordvSphereMetric ( vclib . OperationAttachVolume , requestTime , err )
return diskUUID , err
}
// DetachDisk detaches given virtual disk volume from the compute running kubelet.
func ( vs * VSphere ) DetachDisk ( volPath string , nodeName k8stypes . NodeName ) error {
detachDiskInternal := func ( volPath string , nodeName k8stypes . NodeName ) error {
if nodeName == "" {
nodeName = convertToK8sType ( vs . hostName )
}
// Create context
ctx , cancel := context . WithCancel ( context . Background ( ) )
defer cancel ( )
vsi , err := vs . getVSphereInstance ( nodeName )
if err != nil {
// If node doesn't exist, disk is already detached from node.
if err == vclib . ErrNoVMFound {
klog . Infof ( "Node %q does not exist, disk %s is already detached from node." , convertToString ( nodeName ) , volPath )
return nil
}
return err
}
// Ensure client is logged in and session is valid
err = vs . nodeManager . vcConnect ( ctx , vsi )
if err != nil {
return err
}
vm , err := vs . getVMFromNodeName ( ctx , nodeName )
if err != nil {
// If node doesn't exist, disk is already detached from node.
if err == vclib . ErrNoVMFound {
klog . Infof ( "Node %q does not exist, disk %s is already detached from node." , convertToString ( nodeName ) , volPath )
return nil
}
klog . Errorf ( "Failed to get VM object for node: %q. err: +%v" , convertToString ( nodeName ) , err )
return err
}
err = vm . DetachDisk ( ctx , volPath )
if err != nil {
klog . Errorf ( "Failed to detach disk: %s for node: %s. err: +%v" , volPath , convertToString ( nodeName ) , err )
return err
}
return nil
}
requestTime := time . Now ( )
err := detachDiskInternal ( volPath , nodeName )
if err != nil {
if vclib . IsManagedObjectNotFoundError ( err ) {
err = vs . nodeManager . RediscoverNode ( nodeName )
if err == nil {
err = detachDiskInternal ( volPath , nodeName )
}
}
}
vclib . RecordvSphereMetric ( vclib . OperationDetachVolume , requestTime , err )
return err
}
// DiskIsAttached returns if disk is attached to the VM using controllers supported by the plugin.
func ( vs * VSphere ) DiskIsAttached ( volPath string , nodeName k8stypes . NodeName ) ( bool , error ) {
diskIsAttachedInternal := func ( volPath string , nodeName k8stypes . NodeName ) ( bool , error ) {
var vSphereInstance string
if nodeName == "" {
vSphereInstance = vs . hostName
nodeName = convertToK8sType ( vSphereInstance )
} else {
vSphereInstance = convertToString ( nodeName )
}
// Create context
ctx , cancel := context . WithCancel ( context . Background ( ) )
defer cancel ( )
vsi , err := vs . getVSphereInstance ( nodeName )
if err != nil {
return false , err
}
// Ensure client is logged in and session is valid
err = vs . nodeManager . vcConnect ( ctx , vsi )
if err != nil {
return false , err
}
vm , err := vs . getVMFromNodeName ( ctx , nodeName )
if err != nil {
if err == vclib . ErrNoVMFound {
klog . Warningf ( "Node %q does not exist, vsphere CP will assume disk %v is not attached to it." , nodeName , volPath )
// make the disk as detached and return false without error.
return false , nil
}
klog . Errorf ( "Failed to get VM object for node: %q. err: +%v" , vSphereInstance , err )
return false , err
}
volPath = vclib . RemoveStorageClusterORFolderNameFromVDiskPath ( volPath )
attached , err := vm . IsDiskAttached ( ctx , volPath )
if err != nil {
klog . Errorf ( "DiskIsAttached failed to determine whether disk %q is still attached on node %q" ,
volPath ,
vSphereInstance )
}
klog . V ( 4 ) . Infof ( "DiskIsAttached result: %v and error: %q, for volume: %q" , attached , err , volPath )
return attached , err
}
requestTime := time . Now ( )
isAttached , err := diskIsAttachedInternal ( volPath , nodeName )
if err != nil {
if vclib . IsManagedObjectNotFoundError ( err ) {
err = vs . nodeManager . RediscoverNode ( nodeName )
if err == vclib . ErrNoVMFound {
isAttached , err = false , nil
} else if err == nil {
isAttached , err = diskIsAttachedInternal ( volPath , nodeName )
}
}
}
vclib . RecordvSphereMetric ( vclib . OperationDiskIsAttached , requestTime , err )
return isAttached , err
}
// DisksAreAttached returns if disks are attached to the VM using controllers supported by the plugin.
// 1. Converts volPaths into canonical form so that it can be compared with the VM device path.
// 2. Segregates nodes by vCenter and Datacenter they are present in. This reduces calls to VC.
// 3. Creates go routines per VC-DC to find whether disks are attached to the nodes.
// 4. If the some of the VMs are not found or migrated then they are added to a list.
// 5. After successful execution of goroutines,
// 5a. If there are any VMs which needs to be retried, they are rediscovered and the whole operation is initiated again for only rediscovered VMs.
// 5b. If VMs are removed from vSphere inventory they are ignored.
func ( vs * VSphere ) DisksAreAttached ( nodeVolumes map [ k8stypes . NodeName ] [ ] string ) ( map [ k8stypes . NodeName ] map [ string ] bool , error ) {
disksAreAttachedInternal := func ( nodeVolumes map [ k8stypes . NodeName ] [ ] string ) ( map [ k8stypes . NodeName ] map [ string ] bool , error ) {
// disksAreAttach checks whether disks are attached to the nodes.
// Returns nodes that need to be retried if retry is true
// Segregates nodes per VC and DC
// Creates go routines per VC-DC to find whether disks are attached to the nodes.
disksAreAttach := func ( ctx context . Context , nodeVolumes map [ k8stypes . NodeName ] [ ] string , attached map [ string ] map [ string ] bool , retry bool ) ( [ ] k8stypes . NodeName , error ) {
var wg sync . WaitGroup
var localAttachedMaps [ ] map [ string ] map [ string ] bool
var nodesToRetry [ ] k8stypes . NodeName
var globalErr error
globalErr = nil
globalErrMutex := & sync . Mutex { }
nodesToRetryMutex := & sync . Mutex { }
// Segregate nodes according to VC-DC
dcNodes := make ( map [ string ] [ ] k8stypes . NodeName )
for nodeName := range nodeVolumes {
nodeInfo , err := vs . nodeManager . GetNodeInfo ( nodeName )
if err != nil {
klog . Errorf ( "Failed to get node info: %+v. err: %+v" , nodeInfo . vm , err )
return nodesToRetry , err
}
VC_DC := nodeInfo . vcServer + nodeInfo . dataCenter . String ( )
dcNodes [ VC_DC ] = append ( dcNodes [ VC_DC ] , nodeName )
}
2020-08-10 17:43:49 +00:00
for _ , nodeNames := range dcNodes {
2019-09-27 21:51:53 +00:00
localAttachedMap := make ( map [ string ] map [ string ] bool )
localAttachedMaps = append ( localAttachedMaps , localAttachedMap )
// Start go routines per VC-DC to check disks are attached
2020-08-10 17:43:49 +00:00
go func ( nodes [ ] k8stypes . NodeName ) {
2019-09-27 21:51:53 +00:00
nodesToRetryLocal , err := vs . checkDiskAttached ( ctx , nodes , nodeVolumes , localAttachedMap , retry )
if err != nil {
if ! vclib . IsManagedObjectNotFoundError ( err ) {
globalErrMutex . Lock ( )
globalErr = err
globalErrMutex . Unlock ( )
klog . Errorf ( "Failed to check disk attached for nodes: %+v. err: %+v" , nodes , err )
}
}
nodesToRetryMutex . Lock ( )
nodesToRetry = append ( nodesToRetry , nodesToRetryLocal ... )
nodesToRetryMutex . Unlock ( )
wg . Done ( )
2020-08-10 17:43:49 +00:00
} ( nodeNames )
2019-09-27 21:51:53 +00:00
wg . Add ( 1 )
}
wg . Wait ( )
if globalErr != nil {
return nodesToRetry , globalErr
}
for _ , localAttachedMap := range localAttachedMaps {
for key , value := range localAttachedMap {
attached [ key ] = value
}
}
return nodesToRetry , nil
}
klog . V ( 4 ) . Infof ( "Starting DisksAreAttached API for vSphere with nodeVolumes: %+v" , nodeVolumes )
// Create context
ctx , cancel := context . WithCancel ( context . Background ( ) )
defer cancel ( )
disksAttached := make ( map [ k8stypes . NodeName ] map [ string ] bool )
if len ( nodeVolumes ) == 0 {
return disksAttached , nil
}
// Convert VolPaths into canonical form so that it can be compared with the VM device path.
vmVolumes , err := vs . convertVolPathsToDevicePaths ( ctx , nodeVolumes )
if err != nil {
klog . Errorf ( "Failed to convert volPaths to devicePaths: %+v. err: %+v" , nodeVolumes , err )
return nil , err
}
attached := make ( map [ string ] map [ string ] bool )
nodesToRetry , err := disksAreAttach ( ctx , vmVolumes , attached , false )
if err != nil {
return nil , err
}
if len ( nodesToRetry ) != 0 {
// Rediscover nodes which are need to be retried
remainingNodesVolumes := make ( map [ k8stypes . NodeName ] [ ] string )
for _ , nodeName := range nodesToRetry {
err = vs . nodeManager . RediscoverNode ( nodeName )
if err != nil {
if err == vclib . ErrNoVMFound {
klog . V ( 4 ) . Infof ( "node %s not found. err: %+v" , nodeName , err )
continue
}
klog . Errorf ( "Failed to rediscover node %s. err: %+v" , nodeName , err )
return nil , err
}
remainingNodesVolumes [ nodeName ] = nodeVolumes [ nodeName ]
}
// If some remaining nodes are still registered
if len ( remainingNodesVolumes ) != 0 {
nodesToRetry , err = disksAreAttach ( ctx , remainingNodesVolumes , attached , true )
if err != nil || len ( nodesToRetry ) != 0 {
klog . Errorf ( "Failed to retry disksAreAttach for nodes %+v. err: %+v" , remainingNodesVolumes , err )
return nil , err
}
}
for nodeName , volPaths := range attached {
disksAttached [ convertToK8sType ( nodeName ) ] = volPaths
}
}
klog . V ( 4 ) . Infof ( "DisksAreAttach successfully executed. result: %+v" , attached )
return disksAttached , nil
}
requestTime := time . Now ( )
attached , err := disksAreAttachedInternal ( nodeVolumes )
vclib . RecordvSphereMetric ( vclib . OperationDisksAreAttached , requestTime , err )
return attached , err
}
// CreateVolume creates a volume of given size (in KiB) and return the volume path.
// If the volumeOptions.Datastore is part of datastore cluster for example - [DatastoreCluster/sharedVmfs-0] then
// return value will be [DatastoreCluster/sharedVmfs-0] kubevols/<volume-name>.vmdk
// else return value will be [sharedVmfs-0] kubevols/<volume-name>.vmdk
func ( vs * VSphere ) CreateVolume ( volumeOptions * vclib . VolumeOptions ) ( canonicalVolumePath string , err error ) {
klog . V ( 1 ) . Infof ( "Starting to create a vSphere volume with volumeOptions: %+v" , volumeOptions )
createVolumeInternal := func ( volumeOptions * vclib . VolumeOptions ) ( canonicalVolumePath string , err error ) {
var datastoreInfo * vclib . DatastoreInfo
var dsList [ ] * vclib . DatastoreInfo
// Create context
ctx , cancel := context . WithCancel ( context . Background ( ) )
defer cancel ( )
vsi , err := vs . getVSphereInstanceForServer ( vs . cfg . Workspace . VCenterIP , ctx )
if err != nil {
return "" , err
}
// If datastore not specified, then use default datastore
datastoreName := strings . TrimSpace ( volumeOptions . Datastore )
if datastoreName == "" {
datastoreName = strings . TrimSpace ( vs . cfg . Workspace . DefaultDatastore )
}
// The given datastoreName may be present in more than one datacenter
candidateDatastoreInfos , err := vs . FindDatastoreByName ( ctx , datastoreName )
if err != nil {
return "" , err
}
// Each of the datastores found is a candidate for Volume creation.
// One of these will be selected based on given policy and/or zone.
candidateDatastores := make ( map [ string ] * vclib . DatastoreInfo )
for _ , dsInfo := range candidateDatastoreInfos {
candidateDatastores [ dsInfo . Info . Url ] = dsInfo
}
var vmOptions * vclib . VMOptions
var zonesToSearch [ ] string
if volumeOptions . SelectedNode != nil {
if len ( volumeOptions . Zone ) > 1 {
// In waitForFirstConsumer mode, if more than one allowedTopologies is specified, the volume should satisfy all these.
zonesToSearch = volumeOptions . Zone
} else {
// Pick the selectedNode's zone, if available.
nodeInfo , err := vs . nodeManager . GetNodeInfoWithNodeObject ( volumeOptions . SelectedNode )
if err != nil {
klog . Errorf ( "Unable to get node information for %s. err: %+v" , volumeOptions . SelectedNode . Name , err )
return "" , err
}
klog . V ( 4 ) . Infof ( "selectedNode info : %s" , nodeInfo )
if nodeInfo . zone != nil && nodeInfo . zone . FailureDomain != "" {
zonesToSearch = append ( zonesToSearch , nodeInfo . zone . FailureDomain )
}
}
} else {
// If no selectedNode, pick allowedTopologies, if provided.
zonesToSearch = volumeOptions . Zone
}
klog . V ( 1 ) . Infof ( "Volume topology : %s" , zonesToSearch )
if volumeOptions . VSANStorageProfileData != "" || volumeOptions . StoragePolicyName != "" {
// If datastore and zone are specified, first validate if the datastore is in the provided zone.
if len ( zonesToSearch ) != 0 && volumeOptions . Datastore != "" {
klog . V ( 4 ) . Infof ( "Specified zone : %s, datastore : %s" , zonesToSearch , volumeOptions . Datastore )
dsList , err = getDatastoresForZone ( ctx , vs . nodeManager , zonesToSearch )
if err != nil {
klog . Errorf ( "Failed to find a shared datastore matching zone %s. err: %+v" , zonesToSearch , err )
return "" , err
}
// Validate if the datastore provided belongs to the zone. If not, fail the operation.
found := false
for _ , ds := range dsList {
if datastoreInfo , found = candidateDatastores [ ds . Info . Url ] ; found {
break
}
}
if ! found {
err := fmt . Errorf ( "The specified datastore %s does not match the provided zones : %s" , volumeOptions . Datastore , zonesToSearch )
klog . Error ( err )
return "" , err
}
}
// Acquire a read lock to ensure multiple PVC requests can be processed simultaneously.
cleanUpDummyVMLock . RLock ( )
defer cleanUpDummyVMLock . RUnlock ( )
// Create a new background routine that will delete any dummy VM's that are left stale.
// This routine will get executed for every 5 minutes and gets initiated only once in its entire lifetime.
cleanUpRoutineInitLock . Lock ( )
if ! cleanUpRoutineInitialized {
klog . V ( 1 ) . Infof ( "Starting a clean up routine to remove stale dummy VM's" )
go vs . cleanUpDummyVMs ( DummyVMPrefixName )
cleanUpRoutineInitialized = true
}
cleanUpRoutineInitLock . Unlock ( )
}
if volumeOptions . StoragePolicyName != "" && volumeOptions . Datastore == "" {
if len ( zonesToSearch ) == 0 {
klog . V ( 4 ) . Infof ( "Selecting a shared datastore as per the storage policy %s" , volumeOptions . StoragePolicyName )
datastoreInfo , err = getPbmCompatibleDatastore ( ctx , vsi . conn . Client , volumeOptions . StoragePolicyName , vs . nodeManager )
} else {
// If zone is specified, first get the datastores in the zone.
dsList , err = getDatastoresForZone ( ctx , vs . nodeManager , zonesToSearch )
if err != nil {
klog . Errorf ( "Failed to find a shared datastore matching zone %s. err: %+v" , zonesToSearch , err )
return "" , err
}
klog . V ( 4 ) . Infof ( "Specified zone : %s. Picking a datastore as per the storage policy %s among the zoned datastores : %s" , zonesToSearch ,
volumeOptions . StoragePolicyName , dsList )
// Among the compatible datastores, select the one based on the maximum free space.
datastoreInfo , err = getPbmCompatibleZonedDatastore ( ctx , vsi . conn . Client , volumeOptions . StoragePolicyName , dsList )
}
if err != nil {
klog . Errorf ( "Failed to get pbm compatible datastore with storagePolicy: %s. err: %+v" , volumeOptions . StoragePolicyName , err )
return "" , err
}
klog . V ( 1 ) . Infof ( "Datastore selected as per policy : %s" , datastoreInfo . Info . Name )
} else {
// If zone is specified, pick the datastore in the zone with maximum free space within the zone.
if volumeOptions . Datastore == "" && len ( zonesToSearch ) != 0 {
klog . V ( 4 ) . Infof ( "Specified zone : %s" , zonesToSearch )
dsList , err = getDatastoresForZone ( ctx , vs . nodeManager , zonesToSearch )
if err != nil {
klog . Errorf ( "Failed to find a shared datastore matching zone %s. err: %+v" , zonesToSearch , err )
return "" , err
}
// If unable to get any datastore, fail the operation
if len ( dsList ) == 0 {
err := fmt . Errorf ( "Failed to find a shared datastore matching zone %s" , zonesToSearch )
klog . Error ( err )
return "" , err
}
datastoreInfo , err = getMostFreeDatastore ( ctx , nil , dsList )
if err != nil {
klog . Errorf ( "Failed to get shared datastore: %+v" , err )
return "" , err
}
klog . V ( 1 ) . Infof ( "Specified zone : %s. Selected datastore : %s" , zonesToSearch , datastoreInfo . Info . Name )
} else {
var sharedDsList [ ] * vclib . DatastoreInfo
var err error
if len ( zonesToSearch ) == 0 {
// If zone is not provided, get the shared datastore across all node VMs.
klog . V ( 4 ) . Infof ( "Validating if datastore %s is shared across all node VMs" , datastoreName )
sharedDsList , err = getSharedDatastoresInK8SCluster ( ctx , vs . nodeManager )
if err != nil {
klog . Errorf ( "Failed to get shared datastore: %+v" , err )
return "" , err
}
// Prepare error msg to be used later, if required.
err = fmt . Errorf ( "The specified datastore %s is not a shared datastore across node VMs" , datastoreName )
} else {
// If zone is provided, get the shared datastores in that zone.
klog . V ( 4 ) . Infof ( "Validating if datastore %s is in zone %s " , datastoreName , zonesToSearch )
sharedDsList , err = getDatastoresForZone ( ctx , vs . nodeManager , zonesToSearch )
if err != nil {
klog . Errorf ( "Failed to find a shared datastore matching zone %s. err: %+v" , zonesToSearch , err )
return "" , err
}
// Prepare error msg to be used later, if required.
err = fmt . Errorf ( "The specified datastore %s does not match the provided zones : %s" , datastoreName , zonesToSearch )
}
found := false
// Check if the selected datastore belongs to the list of shared datastores computed.
for _ , sharedDs := range sharedDsList {
if datastoreInfo , found = candidateDatastores [ sharedDs . Info . Url ] ; found {
klog . V ( 4 ) . Infof ( "Datastore validation succeeded" )
found = true
break
}
}
if ! found {
klog . Error ( err )
return "" , err
}
}
}
// if datastoreInfo is still not determined, it is an error condition
if datastoreInfo == nil {
klog . Errorf ( "ambiguous datastore name %s, cannot be found among: %v" , datastoreName , candidateDatastoreInfos )
return "" , fmt . Errorf ( "ambiguous datastore name %s" , datastoreName )
}
ds := datastoreInfo . Datastore
volumeOptions . Datastore = datastoreInfo . Info . Name
vmOptions , err = vs . setVMOptions ( ctx , vsi . conn , ds )
if err != nil {
klog . Errorf ( "failed to set VM options required to create a vsphere volume. err: %+v" , err )
return "" , err
}
kubeVolsPath := filepath . Clean ( ds . Path ( VolDir ) ) + "/"
err = ds . CreateDirectory ( ctx , kubeVolsPath , false )
if err != nil && err != vclib . ErrFileAlreadyExist {
klog . Errorf ( "Cannot create dir %#v. err %s" , kubeVolsPath , err )
return "" , err
}
volumePath := kubeVolsPath + volumeOptions . Name + ".vmdk"
disk := diskmanagers . VirtualDisk {
DiskPath : volumePath ,
VolumeOptions : volumeOptions ,
VMOptions : vmOptions ,
}
volumePath , err = disk . Create ( ctx , ds )
if err != nil {
klog . Errorf ( "Failed to create a vsphere volume with volumeOptions: %+v on datastore: %s. err: %+v" , volumeOptions , ds , err )
return "" , err
}
// Get the canonical path for the volume path.
canonicalVolumePath , err = getcanonicalVolumePath ( ctx , datastoreInfo . Datacenter , volumePath )
if err != nil {
klog . Errorf ( "Failed to get canonical vsphere volume path for volume: %s with volumeOptions: %+v on datastore: %s. err: %+v" , volumePath , volumeOptions , ds , err )
return "" , err
}
if filepath . Base ( datastoreName ) != datastoreName {
// If datastore is within cluster, add cluster path to the volumePath
canonicalVolumePath = strings . Replace ( canonicalVolumePath , filepath . Base ( datastoreName ) , datastoreName , 1 )
}
return canonicalVolumePath , nil
}
requestTime := time . Now ( )
canonicalVolumePath , err = createVolumeInternal ( volumeOptions )
vclib . RecordCreateVolumeMetric ( volumeOptions , requestTime , err )
klog . V ( 4 ) . Infof ( "The canonical volume path for the newly created vSphere volume is %q" , canonicalVolumePath )
return canonicalVolumePath , err
}
// DeleteVolume deletes a volume given volume name.
func ( vs * VSphere ) DeleteVolume ( vmDiskPath string ) error {
klog . V ( 1 ) . Infof ( "Starting to delete vSphere volume with vmDiskPath: %s" , vmDiskPath )
deleteVolumeInternal := func ( vmDiskPath string ) error {
// Create context
ctx , cancel := context . WithCancel ( context . Background ( ) )
defer cancel ( )
vsi , err := vs . getVSphereInstanceForServer ( vs . cfg . Workspace . VCenterIP , ctx )
if err != nil {
return err
}
dc , err := vclib . GetDatacenter ( ctx , vsi . conn , vs . cfg . Workspace . Datacenter )
if err != nil {
return err
}
disk := diskmanagers . VirtualDisk {
DiskPath : vmDiskPath ,
VolumeOptions : & vclib . VolumeOptions { } ,
VMOptions : & vclib . VMOptions { } ,
}
err = disk . Delete ( ctx , dc )
if err != nil {
klog . Errorf ( "Failed to delete vsphere volume with vmDiskPath: %s. err: %+v" , vmDiskPath , err )
}
return err
}
requestTime := time . Now ( )
err := deleteVolumeInternal ( vmDiskPath )
vclib . RecordvSphereMetric ( vclib . OperationDeleteVolume , requestTime , err )
return err
}
// HasClusterID returns true if the cluster has a clusterID
func ( vs * VSphere ) HasClusterID ( ) bool {
return true
}
// Notification handler when node is added into k8s cluster.
func ( vs * VSphere ) NodeAdded ( obj interface { } ) {
node , ok := obj . ( * v1 . Node )
if node == nil || ! ok {
klog . Warningf ( "NodeAdded: unrecognized object %+v" , obj )
return
}
klog . V ( 4 ) . Infof ( "Node added: %+v" , node )
if err := vs . nodeManager . RegisterNode ( node ) ; err != nil {
klog . Errorf ( "failed to add node %+v: %v" , node , err )
}
}
// Notification handler when node is removed from k8s cluster.
func ( vs * VSphere ) NodeDeleted ( obj interface { } ) {
node , ok := obj . ( * v1 . Node )
if node == nil || ! ok {
klog . Warningf ( "NodeDeleted: unrecognized object %+v" , obj )
return
}
klog . V ( 4 ) . Infof ( "Node deleted: %+v" , node )
if err := vs . nodeManager . UnRegisterNode ( node ) ; err != nil {
klog . Errorf ( "failed to delete node %s: %v" , node . Name , err )
}
}
2020-08-10 17:43:49 +00:00
// Notification handler when credentials secret is added.
func ( vs * VSphere ) SecretAdded ( obj interface { } ) {
secret , ok := obj . ( * v1 . Secret )
if secret == nil || ! ok {
klog . Warningf ( "Unrecognized secret object %T" , obj )
return
}
if secret . Name != vs . cfg . Global . SecretName ||
secret . Namespace != vs . cfg . Global . SecretNamespace {
return
}
2020-10-14 19:03:41 +00:00
klog . V ( 4 ) . Infof ( "refreshing node cache for secret: %s/%s" , secret . Namespace , secret . Name )
2020-08-10 17:43:49 +00:00
vs . refreshNodesForSecretChange ( )
}
// Notification handler when credentials secret is updated.
func ( vs * VSphere ) SecretUpdated ( obj interface { } , newObj interface { } ) {
oldSecret , ok := obj . ( * v1 . Secret )
if oldSecret == nil || ! ok {
klog . Warningf ( "Unrecognized secret object %T" , obj )
return
}
secret , ok := newObj . ( * v1 . Secret )
if secret == nil || ! ok {
klog . Warningf ( "Unrecognized secret object %T" , newObj )
return
}
if secret . Name != vs . cfg . Global . SecretName ||
secret . Namespace != vs . cfg . Global . SecretNamespace ||
reflect . DeepEqual ( secret . Data , oldSecret . Data ) {
return
}
2020-10-14 19:03:41 +00:00
klog . V ( 4 ) . Infof ( "refreshing node cache for secret: %s/%s" , secret . Namespace , secret . Name )
2020-08-10 17:43:49 +00:00
vs . refreshNodesForSecretChange ( )
}
func ( vs * VSphere ) refreshNodesForSecretChange ( ) {
err := vs . nodeManager . refreshNodes ( )
if err != nil {
klog . Errorf ( "failed to rediscover nodes: %v" , err )
}
}
2019-09-27 21:51:53 +00:00
func ( vs * VSphere ) NodeManager ( ) ( nodeManager * NodeManager ) {
if vs == nil {
return nil
}
return vs . nodeManager
}
func withTagsClient ( ctx context . Context , connection * vclib . VSphereConnection , f func ( c * rest . Client ) error ) error {
c := rest . NewClient ( connection . Client )
signer , err := connection . Signer ( ctx , connection . Client )
if err != nil {
return err
}
if signer == nil {
user := url . UserPassword ( connection . Username , connection . Password )
err = c . Login ( ctx , user )
} else {
err = c . LoginByToken ( c . WithSigner ( ctx , signer ) )
}
if err != nil {
return err
}
defer func ( ) {
if err := c . Logout ( ctx ) ; err != nil {
klog . Errorf ( "failed to logout: %v" , err )
}
} ( )
return f ( c )
}
// GetZone implements Zones.GetZone
func ( vs * VSphere ) GetZone ( ctx context . Context ) ( cloudprovider . Zone , error ) {
nodeName , err := vs . CurrentNodeName ( ctx , vs . hostName )
if err != nil {
klog . Errorf ( "Cannot get node name." )
return cloudprovider . Zone { } , err
}
zone := cloudprovider . Zone { }
vsi , err := vs . getVSphereInstanceForServer ( vs . cfg . Workspace . VCenterIP , ctx )
if err != nil {
klog . Errorf ( "Cannot connect to vsphere. Get zone for node %s error" , nodeName )
return cloudprovider . Zone { } , err
}
dc , err := vclib . GetDatacenter ( ctx , vsi . conn , vs . cfg . Workspace . Datacenter )
if err != nil {
klog . Errorf ( "Cannot connect to datacenter. Get zone for node %s error" , nodeName )
return cloudprovider . Zone { } , err
}
vmHost , err := dc . GetHostByVMUUID ( ctx , vs . vmUUID )
if err != nil {
klog . Errorf ( "Cannot find VM runtime host. Get zone for node %s error" , nodeName )
return cloudprovider . Zone { } , err
}
pc := vsi . conn . Client . ServiceContent . PropertyCollector
err = withTagsClient ( ctx , vsi . conn , func ( c * rest . Client ) error {
client := tags . NewManager ( c )
// example result: ["Folder", "Datacenter", "Cluster", "Host"]
objects , err := mo . Ancestors ( ctx , vsi . conn . Client , pc , * vmHost )
if err != nil {
return err
}
// search the hierarchy, example order: ["Host", "Cluster", "Datacenter", "Folder"]
for i := range objects {
obj := objects [ len ( objects ) - 1 - i ]
tags , err := client . ListAttachedTags ( ctx , obj )
if err != nil {
klog . Errorf ( "Cannot list attached tags. Get zone for node %s: %s" , nodeName , err )
return err
}
for _ , value := range tags {
tag , err := client . GetTag ( ctx , value )
if err != nil {
klog . Errorf ( "Get tag %s: %s" , value , err )
return err
}
category , err := client . GetCategory ( ctx , tag . CategoryID )
if err != nil {
klog . Errorf ( "Get category %s error" , value )
return err
}
found := func ( ) {
klog . Errorf ( "Found %q tag (%s) for %s attached to %s" , category . Name , tag . Name , vs . vmUUID , obj . Reference ( ) )
}
switch {
case category . Name == vs . cfg . Labels . Zone :
zone . FailureDomain = tag . Name
found ( )
case category . Name == vs . cfg . Labels . Region :
zone . Region = tag . Name
found ( )
}
if zone . FailureDomain != "" && zone . Region != "" {
return nil
}
}
}
if zone . Region == "" {
return fmt . Errorf ( "vSphere region category %q does not match any tags for node %s [%s]" , vs . cfg . Labels . Region , nodeName , vs . vmUUID )
}
if zone . FailureDomain == "" {
return fmt . Errorf ( "vSphere zone category %q does not match any tags for node %s [%s]" , vs . cfg . Labels . Zone , nodeName , vs . vmUUID )
}
return nil
} )
if err != nil {
klog . Errorf ( "Get zone for node %s: %s" , nodeName , err )
return cloudprovider . Zone { } , err
}
return zone , nil
}
func ( vs * VSphere ) GetZoneByNodeName ( ctx context . Context , nodeName k8stypes . NodeName ) ( cloudprovider . Zone , error ) {
return cloudprovider . Zone { } , cloudprovider . NotImplemented
}
func ( vs * VSphere ) GetZoneByProviderID ( ctx context . Context , providerID string ) ( cloudprovider . Zone , error ) {
return cloudprovider . Zone { } , cloudprovider . NotImplemented
}
// GetLabelsForVolume implements the PVLabeler interface for VSphere
// since this interface is used by the PV label admission controller.
func ( vs * VSphere ) GetLabelsForVolume ( ctx context . Context , pv * v1 . PersistentVolume ) ( map [ string ] string , error ) {
// ignore if zones not enabled
if ! vs . isZoneEnabled ( ) {
klog . V ( 4 ) . Infof ( "Zone labels for volume is not enabled in vsphere.conf" )
return nil , nil
}
// ignore if not vSphere volume
if pv . Spec . VsphereVolume == nil {
return nil , nil
}
return vs . GetVolumeLabels ( pv . Spec . VsphereVolume . VolumePath )
}
// GetVolumeLabels returns the well known zone and region labels for given volume
func ( vs * VSphere ) GetVolumeLabels ( volumePath string ) ( map [ string ] string , error ) {
// Create context
ctx , cancel := context . WithCancel ( context . Background ( ) )
defer cancel ( )
// if zones is not enabled return no labels
if ! vs . isZoneEnabled ( ) {
klog . V ( 4 ) . Infof ( "Volume zone labels is not enabled in vsphere.conf" )
return nil , nil
}
// Find the datastore on which this volume resides
datastorePathObj , err := vclib . GetDatastorePathObjFromVMDiskPath ( volumePath )
if err != nil {
klog . Errorf ( "Failed to get datastore for volume: %v: %+v" , volumePath , err )
return nil , err
}
dsInfos , err := vs . FindDatastoreByName ( ctx , datastorePathObj . Datastore )
if err != nil {
klog . Errorf ( "Failed to get datastore by name: %v: %+v" , datastorePathObj . Datastore , err )
return nil , err
}
var datastore * vclib . Datastore
for _ , dsInfo := range dsInfos {
if dsInfo . Datastore . Exists ( ctx , datastorePathObj . Path ) {
datastore = dsInfo . Datastore
}
}
if datastore == nil {
klog . Errorf ( "Could not find %s among %v" , volumePath , dsInfos )
return nil , fmt . Errorf ( "could not find the datastore for volume: %s" , volumePath )
}
dsZones , err := vs . GetZonesForDatastore ( ctx , datastore )
if err != nil {
klog . Errorf ( "Failed to get zones for datastore %v: %+v" , datastorePathObj . Datastore , err )
return nil , err
}
dsZones , err = vs . collapseZonesInRegion ( ctx , dsZones )
if err != nil {
klog . Errorf ( "Failed to collapse zones. %v" , err )
return nil , err
}
// FIXME: For now, pick the first zone of datastore as the zone of volume
labels := make ( map [ string ] string )
if len ( dsZones ) > 0 {
labels [ v1 . LabelZoneRegion ] = dsZones [ 0 ] . Region
labels [ v1 . LabelZoneFailureDomain ] = dsZones [ 0 ] . FailureDomain
}
return labels , nil
}
// collapse all zones in same region. Join FailureDomain with well known separator
func ( vs * VSphere ) collapseZonesInRegion ( ctx context . Context , zones [ ] cloudprovider . Zone ) ( [ ] cloudprovider . Zone , error ) {
// first create a map of region -> list of zones in that region
regionToZones := make ( map [ string ] [ ] string )
for _ , zone := range zones {
fds , exists := regionToZones [ zone . Region ]
if ! exists {
fds = make ( [ ] string , 0 )
}
regionToZones [ zone . Region ] = append ( fds , zone . FailureDomain )
}
// Join all fds in same region and return Zone instances
collapsedZones := make ( [ ] cloudprovider . Zone , 0 )
for region , fds := range regionToZones {
fdSet := sets . NewString ( fds ... )
appendedZone := volumehelpers . ZonesSetToLabelValue ( fdSet )
collapsedZones = append ( collapsedZones , cloudprovider . Zone { FailureDomain : appendedZone , Region : region } )
}
return collapsedZones , nil
}
// GetZonesForDatastore returns all the zones from which this datastore is visible
func ( vs * VSphere ) GetZonesForDatastore ( ctx context . Context , datastore * vclib . Datastore ) ( [ ] cloudprovider . Zone , error ) {
vsi , err := vs . getVSphereInstanceForServer ( vs . cfg . Workspace . VCenterIP , ctx )
if err != nil {
klog . Errorf ( "Failed to get vSphere instance: %+v" , err )
return nil , err
}
// get the hosts mounted on this datastore
// datastore -> ["host-1", "host-2", "host-3", ...]
dsHosts , err := datastore . GetDatastoreHostMounts ( ctx )
if err != nil {
klog . Errorf ( "Failed to get datastore host mounts for %v: %+v" , datastore , err )
return nil , err
}
klog . V ( 4 ) . Infof ( "Got host mounts for datastore: %v: %v" , datastore , dsHosts )
// compute map of zone to list of hosts in that zone across all hosts in vsphere
// zone -> ["host-i", "host-j", "host-k", ...]
zoneToHosts , err := vs . GetZoneToHosts ( ctx , vsi )
if err != nil {
klog . Errorf ( "Failed to get zones for hosts: %+v" , err )
return nil , err
}
klog . V ( 4 ) . Infof ( "Got zone to hosts: %v" , zoneToHosts )
// datastore belongs to a zone if all hosts in that zone mount that datastore
dsZones := make ( [ ] cloudprovider . Zone , 0 )
for zone , zoneHosts := range zoneToHosts {
// if zone is valid and zoneHosts is a subset of dsHosts, then add zone
if zone . Region != "" && containsAll ( dsHosts , zoneHosts ) {
dsZones = append ( dsZones , zone )
}
}
klog . V ( 4 ) . Infof ( "Datastore %s belongs to zones: %v" , datastore , dsZones )
return dsZones , nil
}
// GetZoneToHosts returns a map of 'zone' -> 'list of hosts in that zone' in given VC
func ( vs * VSphere ) GetZoneToHosts ( ctx context . Context , vsi * VSphereInstance ) ( map [ cloudprovider . Zone ] [ ] vmwaretypes . ManagedObjectReference , error ) {
// Approach is to find tags with the category of 'vs.cfg.Labels.Zone'
zoneToHosts := make ( map [ cloudprovider . Zone ] [ ] vmwaretypes . ManagedObjectReference )
getHostsInTagCategory := func ( ctx context . Context , tagCategoryName string ) ( map [ vmwaretypes . ManagedObjectReference ] string , error ) {
hostToTag := make ( map [ vmwaretypes . ManagedObjectReference ] string )
err := withTagsClient ( ctx , vsi . conn , func ( c * rest . Client ) error {
// Look whether the zone/region tag is defined in VC
tagManager := tags . NewManager ( c )
tagsForCat , err := tagManager . GetTagsForCategory ( ctx , tagCategoryName )
if err != nil {
klog . V ( 4 ) . Infof ( "No tags with category %s exists in VC. So ignoring." , tagCategoryName )
// return a special error so that tag unavailability can be ignored
return ErrNoZoneTagInVC
}
klog . V ( 4 ) . Infof ( "List of tags under category %s: %v" , tagCategoryName , tagsForCat )
// Each such tag is a different 'zone' marked in vCenter.
// Query for objects associated with each tag. Consider Host, Cluster and Datacenter kind of objects.
tagToObjects := make ( map [ string ] [ ] mo . Reference )
for _ , tag := range tagsForCat {
klog . V ( 4 ) . Infof ( "Getting objects associated with tag %s" , tag . Name )
objects , err := tagManager . ListAttachedObjects ( ctx , tag . Name )
if err != nil {
klog . Errorf ( "Error fetching objects associated with zone tag %s: %+v" , tag . Name , err )
return err
}
tagToObjects [ tag . Name ] = objects
}
klog . V ( 4 ) . Infof ( "Map of tag to objects: %v" , tagToObjects )
// Infer zone for hosts within Datacenter, hosts within clusters and hosts - in this order of increasing priority
// The below nested for-loops goes over all the objects in tagToObjects three times over.
for _ , moType := range [ ] string { vclib . DatacenterType , vclib . ClusterComputeResourceType , vclib . HostSystemType } {
for tagName , objects := range tagToObjects {
for _ , obj := range objects {
if obj . Reference ( ) . Type == moType {
klog . V ( 4 ) . Infof ( "Found zone tag %s associated with %s of type %T: %s" , tagName , obj , obj , obj . Reference ( ) . Value )
switch moType {
case "Datacenter" :
// mark that all hosts in this datacenter has tag applied
dcObjRef := object . NewReference ( vsi . conn . Client , obj . Reference ( ) )
klog . V ( 4 ) . Infof ( "Converted mo obj %v to govmomi object ref %v" , obj , dcObjRef )
dcObj , ok := dcObjRef . ( * object . Datacenter )
if ! ok {
errMsg := fmt . Sprintf ( "Not able to convert object to Datacenter %v" , obj )
klog . Errorf ( errMsg )
return errors . New ( errMsg )
}
klog . V ( 4 ) . Infof ( "Converted to object Datacenter %v" , dcObj )
dc := vclib . Datacenter { Datacenter : dcObj }
hosts , err := dc . GetAllHosts ( ctx )
if err != nil {
klog . Errorf ( "Could not get hosts from datacenter %v: %+v" , dc , err )
return err
}
for _ , host := range hosts {
hostToTag [ host ] = tagName
}
case "ClusterComputeResource" :
// mark that all hosts in this cluster has tag applied
clusterObjRef := object . NewReference ( vsi . conn . Client , obj . Reference ( ) )
clusterObj , ok := clusterObjRef . ( * object . ClusterComputeResource )
if ! ok {
errMsg := fmt . Sprintf ( "Not able to convert object ClusterComputeResource %v" , obj )
klog . Errorf ( errMsg )
return errors . New ( errMsg )
}
hostSystemList , err := clusterObj . Hosts ( ctx )
if err != nil {
klog . Errorf ( "Not able to get hosts in cluster %v: %+v" , clusterObj , err )
return err
}
for _ , host := range hostSystemList {
hostToTag [ host . Reference ( ) ] = tagName
}
case "HostSystem" :
// mark that this host has tag applied
hostToTag [ obj . Reference ( ) ] = tagName
}
}
}
}
}
return nil // no error
} )
if err != nil {
klog . Errorf ( "Error processing tag category %s: %+v" , tagCategoryName , err )
return nil , err
}
klog . V ( 6 ) . Infof ( "Computed hostToTag: %v" , hostToTag )
return hostToTag , nil
}
hostToZone , err := getHostsInTagCategory ( ctx , vs . cfg . Labels . Zone )
if err != nil {
if err == ErrNoZoneTagInVC {
return zoneToHosts , nil
}
klog . Errorf ( "Get hosts in tag category %s failed: %+v" , vs . cfg . Labels . Zone , err )
return nil , err
}
hostToRegion , err := getHostsInTagCategory ( ctx , vs . cfg . Labels . Region )
if err != nil {
if err == ErrNoZoneTagInVC {
return zoneToHosts , nil
}
klog . Errorf ( "Get hosts in tag category %s failed: %+v" , vs . cfg . Labels . Region , err )
return nil , err
}
// populate zoneToHosts based on hostToZone and hostToRegion
klog . V ( 6 ) . Infof ( "hostToZone: %v" , hostToZone )
klog . V ( 6 ) . Infof ( "hostToRegion: %v" , hostToRegion )
for host , zone := range hostToZone {
region , regionExists := hostToRegion [ host ]
if ! regionExists {
klog . Errorf ( "Host %s has a zone, but no region. So ignoring." , host )
continue
}
cpZone := cloudprovider . Zone { FailureDomain : zone , Region : region }
hosts , exists := zoneToHosts [ cpZone ]
if ! exists {
hosts = make ( [ ] vmwaretypes . ManagedObjectReference , 0 )
}
zoneToHosts [ cpZone ] = append ( hosts , host )
}
klog . V ( 4 ) . Infof ( "Final zoneToHosts: %v" , zoneToHosts )
return zoneToHosts , nil
}
// returns true if s1 contains all elements from s2; false otherwise
func containsAll ( s1 [ ] vmwaretypes . ManagedObjectReference , s2 [ ] vmwaretypes . ManagedObjectReference ) bool {
// put all elements of s1 into a map
s1Map := make ( map [ vmwaretypes . ManagedObjectReference ] bool )
for _ , mor := range s1 {
s1Map [ mor ] = true
}
// verify if all elements of s2 are present in s1Map
for _ , mor := range s2 {
if _ , found := s1Map [ mor ] ; ! found {
return false
}
}
return true
}