github.com/vmware/govmomi@v0.51.0/cli/cluster/module/vm_add.go (about)

     1  // © Broadcom. All Rights Reserved.
     2  // The term “Broadcom” refers to Broadcom Inc. and/or its subsidiaries.
     3  // SPDX-License-Identifier: Apache-2.0
     4  
     5  package module
     6  
     7  import (
     8  	"context"
     9  	"flag"
    10  	"fmt"
    11  
    12  	"github.com/vmware/govmomi/cli"
    13  	"github.com/vmware/govmomi/cli/flags"
    14  	"github.com/vmware/govmomi/vapi/cluster"
    15  	"github.com/vmware/govmomi/vim25/mo"
    16  )
    17  
    18  type addVMs struct {
    19  	*flags.SearchFlag
    20  	moduleID string
    21  }
    22  
    23  func init() {
    24  	cli.Register("cluster.module.vm.add", &addVMs{})
    25  }
    26  
    27  func (cmd *addVMs) Register(ctx context.Context, f *flag.FlagSet) {
    28  	cmd.SearchFlag, ctx = flags.NewSearchFlag(ctx, flags.SearchVirtualMachines)
    29  	cmd.SearchFlag.Register(ctx, f)
    30  
    31  	f.StringVar(&cmd.moduleID, "id", "", "Module ID")
    32  }
    33  
    34  func (cmd *addVMs) Process(ctx context.Context) error {
    35  	return cmd.SearchFlag.Process(ctx)
    36  }
    37  
    38  func (cmd *addVMs) Usage() string {
    39  	return `VM...`
    40  }
    41  
    42  func (cmd *addVMs) Description() string {
    43  	return `Add VM(s) to a cluster module.
    44  
    45  Examples:
    46    govc cluster.module.vm.add -id module_id $vm...`
    47  }
    48  
    49  func (cmd *addVMs) Run(ctx context.Context, f *flag.FlagSet) error {
    50  	if f.NArg() < 1 {
    51  		return flag.ErrHelp
    52  	}
    53  
    54  	c, err := cmd.RestClient()
    55  	if err != nil {
    56  		return err
    57  	}
    58  
    59  	vms, err := cmd.VirtualMachines(f.Args())
    60  	if err != nil {
    61  		return err
    62  	}
    63  
    64  	refs := make([]mo.Reference, 0, len(vms))
    65  	for _, vm := range vms {
    66  		refs = append(refs, vm.Reference())
    67  	}
    68  
    69  	allAdded, err := cluster.NewManager(c).AddModuleMembers(ctx, cmd.moduleID, refs...)
    70  	if err != nil {
    71  		return err
    72  	}
    73  
    74  	if !allAdded {
    75  		return fmt.Errorf("a VM is already a member of the module or not within the module's cluster")
    76  	}
    77  
    78  	return nil
    79  }