refactor(esxi): fetchVms and fetchHardwareInfo

1. fetchHardwareInfo 只有一种error,原因是moVM的某些字段为nil,这种情况下,完全可以打印日志直接返回。
2. fetchVms 现在只返回[]*SVirtualMachine, 进一步的过滤(是不是template)交给调用者。
This commit is contained in:
rainzm
2020-09-09 21:58:16 +08:00
parent c81563b03d
commit 40b3e8e382
4 changed files with 35 additions and 29 deletions
+9 -18
View File
@@ -237,32 +237,23 @@ func (dc *SDatacenter) getDcObj() *object.Datacenter {
return object.NewDatacenter(dc.manager.client.Client, dc.object.Reference())
}
// fetchVms will identify if VM is a template and return two different arrays; the latter contains all template vms.
func (dc *SDatacenter) fetchVms(vmRefs []types.ManagedObjectReference, all bool) ([]cloudprovider.ICloudVM, []*SVirtualMachine, error) {
var vms []mo.VirtualMachine
func (dc *SDatacenter) fetchVms(vmRefs []types.ManagedObjectReference, all bool) ([]*SVirtualMachine, error) {
var movms []mo.VirtualMachine
if vmRefs != nil {
err := dc.manager.references2Objects(vmRefs, VIRTUAL_MACHINE_PROPS, &vms)
err := dc.manager.references2Objects(vmRefs, VIRTUAL_MACHINE_PROPS, &movms)
if err != nil {
return nil, nil, errors.Wrap(err, "dc.manager.references2Objects")
return nil, errors.Wrap(err, "dc.manager.references2Objects")
}
}
// avoid applying new memory and copying
retVms := make([]cloudprovider.ICloudVM, 0, len(vms)/2)
templateVMs := make([]*SVirtualMachine, 0, 2)
for i := 0; i < len(vms); i += 1 {
if all || !strings.HasPrefix(vms[i].Entity().Name, api.ESXI_IMAGE_CACHE_TMP_PREFIX) {
vmObj := NewVirtualMachine(dc.manager, &vms[i], dc)
if vms[i].Config != nil && vms[i].Config.Template {
templateVMs = append(templateVMs, vmObj)
continue
}
if vmObj != nil {
retVms = append(retVms, vmObj)
}
vms := make([]*SVirtualMachine, 0, len(movms))
for i := range movms {
if all || !strings.HasPrefix(movms[i].Entity().Name, api.ESXI_IMAGE_CACHE_TMP_PREFIX) {
vms = append(vms, NewVirtualMachine(dc.manager, &movms[i], dc))
}
}
return retVms, templateVMs, nil
return vms, nil
}
func (dc *SDatacenter) fetchDatastores(datastoreRefs []types.ManagedObjectReference) ([]cloudprovider.ICloudStorage, error) {
+9 -3
View File
@@ -179,6 +179,7 @@ func (self *SHost) fetchVMs(all bool) error {
}
MAX_TRIES := 3
var vms []*SVirtualMachine
for tried := 0; tried < MAX_TRIES; tried += 1 {
hostVms := self.getHostSystem().Vm
if len(hostVms) == 0 {
@@ -186,15 +187,20 @@ func (self *SHost) fetchVMs(all bool) error {
return nil
}
vms, templatevms, err := dc.fetchVms(hostVms, all)
vms, err = dc.fetchVms(hostVms, all)
if err != nil {
log.Errorf("dc.fetchVms fail %s", err)
time.Sleep(time.Second)
self.Refresh()
continue
}
self.vms = vms
self.tempalteVMs = templatevms
}
for _, vm := range vms {
if vm.IsTemplate() {
self.tempalteVMs = append(self.tempalteVMs, vm)
} else {
self.vms = append(self.vms, vm)
}
}
return nil
}
+8 -1
View File
@@ -259,7 +259,14 @@ func (self *SDatastore) getVMs() ([]cloudprovider.ICloudVM, error) {
if len(vms) == 0 {
return nil, nil
}
ret, _, err := dc.fetchVms(vms, false)
svms, err := dc.fetchVms(vms, false)
if err != nil {
return nil, err
}
ret := make([]cloudprovider.ICloudVM, len(svms))
for i := range svms {
ret[i] = svms[i]
}
return ret, err
}
+9 -7
View File
@@ -72,10 +72,7 @@ func (d byDiskType) Less(i, j int) bool {
func NewVirtualMachine(manager *SESXiClient, vm *mo.VirtualMachine, dc *SDatacenter) *SVirtualMachine {
svm := &SVirtualMachine{SManagedObject: newManagedObject(manager, vm, dc)}
err := svm.fetchHardwareInfo()
if err != nil {
return nil
}
svm.fetchHardwareInfo()
return svm
}
@@ -708,7 +705,7 @@ func (self *SVirtualMachine) UpdateUserData(userData string) error {
return nil
}
func (self *SVirtualMachine) fetchHardwareInfo() error {
func (self *SVirtualMachine) fetchHardwareInfo() {
self.vnics = make([]SVirtualNIC, 0)
self.vdisks = make([]SVirtualDisk, 0)
self.cdroms = make([]SVirtualCdrom, 0)
@@ -722,7 +719,8 @@ func (self *SVirtualMachine) fetchHardwareInfo() error {
}
if moVM == nil || moVM.Config == nil || moVM.Config.Hardware.Device == nil {
return errors.Error("invalid vm config")
log.Errorf("invalid vm config, moVM: %v", moVM)
return
}
for i := 0; i < len(moVM.Config.Hardware.Device); i += 1 {
@@ -750,7 +748,6 @@ func (self *SVirtualMachine) fetchHardwareInfo() error {
sort.Slice(self.vdisks, func(i, j int) bool {
return self.vdisks[i].GetIndex() < self.vdisks[j].GetIndex()
})
return nil
}
func (self *SVirtualMachine) getVdev(key int32) SVirtualDevice {
@@ -1218,3 +1215,8 @@ func (self *SVirtualMachine) FindMinDiffKey(limit int32) int32 {
}
return limit
}
func (self *SVirtualMachine) IsTemplate() bool {
movm := self.getVirtualMachine()
return movm.Config != nil && movm.Config.Template
}