Files
virtual-kubelet/vendor/github.com/vmware/govmomi/govc/option/set.go
Loc Nguyen 513cebe7b7 VMware vSphere Integrated Containers provider (#206)
* Add Virtual Kubelet provider for VIC

Initial virtual kubelet provider for VMware VIC.  This provider currently
handles creating and starting of a pod VM via the VIC portlayer and persona
server.  Image store handling via the VIC persona server.  This provider
currently requires the feature/wolfpack branch of VIC.

* Added pod stop and delete.  Also added node capacity.

Added the ability to stop and delete pod VMs via VIC.  Also retrieve
node capacity information from the VCH.

* Cleanup and readme file

Some file clean up and added a Readme.md markdown file for the VIC
provider.

* Cleaned up errors, added function comments, moved operation code

1. Cleaned up error handling.  Set standard for creating errors.
2. Added method prototype comments for all interface functions.
3. Moved PodCreator, PodStarter, PodStopper, and PodDeleter to a new folder.

* Add mocking code and unit tests for podcache, podcreator, and podstarter

Used the unit test framework used in VIC to handle assertions in the provider's
unit test.  Mocking code generated using OSS project mockery, which is compatible
with the testify assertion framework.

* Vendored packages for the VIC provider

Requires feature/wolfpack branch of VIC and a few specific commit sha of
projects used within VIC.

* Implementation of POD Stopper and Deleter unit tests (#4)

* Updated files for initial PR
2018-06-04 15:41:32 -07:00

142 lines
2.9 KiB
Go

/*
Copyright (c) 2017 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 option
import (
"context"
"flag"
"fmt"
"strconv"
"github.com/vmware/govmomi/govc/cli"
"github.com/vmware/govmomi/govc/flags"
"github.com/vmware/govmomi/object"
"github.com/vmware/govmomi/vim25/soap"
"github.com/vmware/govmomi/vim25/types"
)
type Set struct {
*flags.ClientFlag
}
func init() {
cli.Register("option.set", &Set{})
}
func (cmd *Set) Register(ctx context.Context, f *flag.FlagSet) {
cmd.ClientFlag, ctx = flags.NewClientFlag(ctx)
cmd.ClientFlag.Register(ctx, f)
}
func (cmd *Set) Process(ctx context.Context) error {
if err := cmd.ClientFlag.Process(ctx); err != nil {
return err
}
return nil
}
func (cmd *Set) Usage() string {
return "NAME VALUE"
}
var SetDescription = `Set option NAME to VALUE.`
func (cmd *Set) Description() string {
return SetDescription + `
Examples:
govc option.set log.level info
govc option.set logger.Vsan verbose`
}
func isInvalidName(err error) bool {
if soap.IsSoapFault(err) {
soapFault := soap.ToSoapFault(err)
if _, ok := soapFault.VimFault().(types.InvalidName); ok {
return ok
}
}
return false
}
func (cmd *Set) Update(ctx context.Context, f *flag.FlagSet, m *object.OptionManager) error {
if f.NArg() != 2 {
return flag.ErrHelp
}
name := f.Arg(0)
val := f.Arg(1)
opts, err := m.Query(ctx, name)
if err != nil {
if isInvalidName(err) {
// If the option doesn't exist, creating one can only have a string Value.
// The Key prefix is limited in this case too, it seems to the config.* namespace.
return m.Update(ctx, []types.BaseOptionValue{&types.OptionValue{
Key: name,
Value: val,
}})
}
return err
}
if len(opts) != 1 {
return flag.ErrHelp
}
var set types.AnyType
switch x := opts[0].GetOptionValue().Value.(type) {
case string:
set = val
case bool:
set, err = strconv.ParseBool(val)
if err != nil {
return err
}
case int32:
s, err := strconv.ParseInt(val, 10, 32)
if err != nil {
return err
}
set = s
case int64:
set, err = strconv.ParseInt(val, 10, 64)
if err != nil {
return err
}
default:
return fmt.Errorf("type %T conversion not supported", x)
}
opts[0].GetOptionValue().Value = set
return m.Update(ctx, opts)
}
func (cmd *Set) Run(ctx context.Context, f *flag.FlagSet) error {
c, err := cmd.Client()
if err != nil {
return err
}
m := object.NewOptionManager(c, *c.ServiceContent.Setting)
return cmd.Update(ctx, f, m)
}