mirror of
https://github.com/nikdoof/vsphere-influxdb-go.git
synced 2025-12-20 14:09:21 +00:00
add vendoring with go dep
This commit is contained in:
177
vendor/github.com/vmware/govmomi/govc/host/autostart/autostart.go
generated
vendored
Normal file
177
vendor/github.com/vmware/govmomi/govc/host/autostart/autostart.go
generated
vendored
Normal file
@@ -0,0 +1,177 @@
|
||||
/*
|
||||
Copyright (c) 2015 VMware, Inc. All Rights Reserved.
|
||||
|
||||
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 autostart
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"flag"
|
||||
|
||||
"github.com/vmware/govmomi/govc/flags"
|
||||
"github.com/vmware/govmomi/object"
|
||||
"github.com/vmware/govmomi/vim25/methods"
|
||||
"github.com/vmware/govmomi/vim25/mo"
|
||||
"github.com/vmware/govmomi/vim25/types"
|
||||
)
|
||||
|
||||
type AutostartFlag struct {
|
||||
*flags.ClientFlag
|
||||
*flags.DatacenterFlag
|
||||
*flags.HostSystemFlag
|
||||
}
|
||||
|
||||
func newAutostartFlag(ctx context.Context) (*AutostartFlag, context.Context) {
|
||||
f := &AutostartFlag{}
|
||||
f.ClientFlag, ctx = flags.NewClientFlag(ctx)
|
||||
f.DatacenterFlag, ctx = flags.NewDatacenterFlag(ctx)
|
||||
f.HostSystemFlag, ctx = flags.NewHostSystemFlag(ctx)
|
||||
return f, ctx
|
||||
}
|
||||
|
||||
func (f *AutostartFlag) Register(ctx context.Context, fs *flag.FlagSet) {
|
||||
f.ClientFlag.Register(ctx, fs)
|
||||
f.DatacenterFlag.Register(ctx, fs)
|
||||
f.HostSystemFlag.Register(ctx, fs)
|
||||
}
|
||||
|
||||
func (f *AutostartFlag) Process(ctx context.Context) error {
|
||||
if err := f.ClientFlag.Process(ctx); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := f.DatacenterFlag.Process(ctx); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := f.HostSystemFlag.Process(ctx); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// VirtualMachines returns list of virtual machine objects based on the
|
||||
// arguments specified on the command line. This helper is defined in
|
||||
// flags.SearchFlag as well, but that pulls in other virtual machine flags that
|
||||
// are not relevant here.
|
||||
func (f *AutostartFlag) VirtualMachines(args []string) ([]*object.VirtualMachine, error) {
|
||||
ctx := context.TODO()
|
||||
if len(args) == 0 {
|
||||
return nil, errors.New("no argument")
|
||||
}
|
||||
|
||||
finder, err := f.Finder()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var out []*object.VirtualMachine
|
||||
for _, arg := range args {
|
||||
vms, err := finder.VirtualMachineList(ctx, arg)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
out = append(out, vms...)
|
||||
}
|
||||
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (f *AutostartFlag) HostAutoStartManager() (*mo.HostAutoStartManager, error) {
|
||||
ctx := context.TODO()
|
||||
h, err := f.HostSystem()
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var mhs mo.HostSystem
|
||||
err = h.Properties(ctx, h.Reference(), []string{"configManager.autoStartManager"}, &mhs)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var mhas mo.HostAutoStartManager
|
||||
err = h.Properties(ctx, *mhs.ConfigManager.AutoStartManager, nil, &mhas)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &mhas, nil
|
||||
}
|
||||
|
||||
func (f *AutostartFlag) ReconfigureDefaults(template types.AutoStartDefaults) error {
|
||||
ctx := context.TODO()
|
||||
c, err := f.Client()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
mhas, err := f.HostAutoStartManager()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
req := types.ReconfigureAutostart{
|
||||
This: mhas.Reference(),
|
||||
Spec: types.HostAutoStartManagerConfig{
|
||||
Defaults: &template,
|
||||
},
|
||||
}
|
||||
|
||||
_, err = methods.ReconfigureAutostart(ctx, c, &req)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (f *AutostartFlag) ReconfigureVMs(args []string, template types.AutoStartPowerInfo) error {
|
||||
ctx := context.TODO()
|
||||
c, err := f.Client()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
mhas, err := f.HostAutoStartManager()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
req := types.ReconfigureAutostart{
|
||||
This: mhas.Reference(),
|
||||
Spec: types.HostAutoStartManagerConfig{
|
||||
PowerInfo: make([]types.AutoStartPowerInfo, 0),
|
||||
},
|
||||
}
|
||||
|
||||
vms, err := f.VirtualMachines(args)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for _, vm := range vms {
|
||||
pi := template
|
||||
pi.Key = vm.Reference()
|
||||
req.Spec.PowerInfo = append(req.Spec.PowerInfo, pi)
|
||||
}
|
||||
|
||||
_, err = methods.ReconfigureAutostart(ctx, c, &req)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
Reference in New Issue
Block a user