github.com/vmware/govmomi@v0.51.0/cli/metric/reset.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 metric
     6  
     7  import (
     8  	"context"
     9  	"flag"
    10  
    11  	"github.com/vmware/govmomi/cli"
    12  	"github.com/vmware/govmomi/vim25/methods"
    13  	"github.com/vmware/govmomi/vim25/types"
    14  )
    15  
    16  type reset struct {
    17  	*PerformanceFlag
    18  }
    19  
    20  func init() {
    21  	cli.Register("metric.reset", &reset{})
    22  }
    23  
    24  func (cmd *reset) Register(ctx context.Context, f *flag.FlagSet) {
    25  	cmd.PerformanceFlag, ctx = NewPerformanceFlag(ctx)
    26  	cmd.PerformanceFlag.Register(ctx, f)
    27  }
    28  
    29  func (cmd *reset) Usage() string {
    30  	return "NAME..."
    31  }
    32  
    33  func (cmd *reset) Description() string {
    34  	return `Reset counter NAME to the default level of data collection.
    35  
    36  Examples:
    37    govc metric.reset net.bytesRx.average net.bytesTx.average`
    38  }
    39  
    40  func (cmd *reset) Process(ctx context.Context) error {
    41  	if err := cmd.PerformanceFlag.Process(ctx); err != nil {
    42  		return err
    43  	}
    44  	return nil
    45  }
    46  
    47  func (cmd *reset) Run(ctx context.Context, f *flag.FlagSet) error {
    48  	if f.NArg() == 0 {
    49  		return flag.ErrHelp
    50  	}
    51  
    52  	m, err := cmd.Manager(ctx)
    53  	if err != nil {
    54  		return err
    55  	}
    56  
    57  	counters, err := m.CounterInfoByName(ctx)
    58  	if err != nil {
    59  		return err
    60  	}
    61  
    62  	var ids []int32
    63  
    64  	for _, name := range f.Args() {
    65  		counter, ok := counters[name]
    66  		if !ok {
    67  			return cmd.ErrNotFound(name)
    68  		}
    69  
    70  		ids = append(ids, counter.Key)
    71  	}
    72  
    73  	_, err = methods.ResetCounterLevelMapping(ctx, m.Client(), &types.ResetCounterLevelMapping{
    74  		This:     m.Reference(),
    75  		Counters: ids,
    76  	})
    77  
    78  	return err
    79  }