92 lines
2.0 KiB
Go
92 lines
2.0 KiB
Go
package driver
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
|
|
"github.com/vmware/govmomi/find"
|
|
"github.com/vmware/govmomi/object"
|
|
"github.com/vmware/govmomi/vim25/mo"
|
|
"github.com/vmware/govmomi/vim25/types"
|
|
)
|
|
|
|
type ResourcePool struct {
|
|
pool *object.ResourcePool
|
|
driver *VCenterDriver
|
|
}
|
|
|
|
func (d *VCenterDriver) NewResourcePool(ref *types.ManagedObjectReference) *ResourcePool {
|
|
return &ResourcePool{
|
|
pool: object.NewResourcePool(d.client.Client, *ref),
|
|
driver: d,
|
|
}
|
|
}
|
|
|
|
func (d *VCenterDriver) FindResourcePool(cluster string, host string, name string) (*ResourcePool, error) {
|
|
var res string
|
|
if cluster != "" {
|
|
res = cluster
|
|
} else {
|
|
res = host
|
|
}
|
|
|
|
resourcePath := fmt.Sprintf("%v/Resources/%v", res, name)
|
|
p, err := d.finder.ResourcePool(d.ctx, resourcePath)
|
|
if err != nil {
|
|
log.Printf("[WARN] %s not found. Looking for default resource pool.", resourcePath)
|
|
dp, dperr := d.finder.DefaultResourcePool(d.ctx)
|
|
if _, ok := dperr.(*find.NotFoundError); ok {
|
|
// VirtualApp extends ResourcePool, so it should support VirtualApp types.
|
|
vapp, verr := d.finder.VirtualApp(d.ctx, name)
|
|
if verr != nil {
|
|
return nil, err
|
|
}
|
|
dp = vapp.ResourcePool
|
|
} else if dperr != nil {
|
|
return nil, err
|
|
}
|
|
p = dp
|
|
}
|
|
|
|
return &ResourcePool{
|
|
pool: p,
|
|
driver: d,
|
|
}, nil
|
|
}
|
|
|
|
func (p *ResourcePool) Info(params ...string) (*mo.ResourcePool, error) {
|
|
var params2 []string
|
|
if len(params) == 0 {
|
|
params2 = []string{"*"}
|
|
} else {
|
|
params2 = params
|
|
}
|
|
var info mo.ResourcePool
|
|
err := p.pool.Properties(p.driver.ctx, p.pool.Reference(), params2, &info)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &info, nil
|
|
}
|
|
|
|
func (p *ResourcePool) Path() (string, error) {
|
|
poolInfo, err := p.Info("name", "parent")
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if poolInfo.Parent.Type == "ComputeResource" {
|
|
return "", nil
|
|
} else {
|
|
parent := p.driver.NewResourcePool(poolInfo.Parent)
|
|
parentPath, err := parent.Path()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if parentPath == "" {
|
|
return poolInfo.Name, nil
|
|
} else {
|
|
return fmt.Sprintf("%v/%v", parentPath, poolInfo.Name), nil
|
|
}
|
|
}
|
|
}
|