cmd

Documentation

Overview

cmd package contains all the command line flag and initialization logic for commands

Index

• Variables
• func Execute(args []string, overrides map[string]string) (string, error)
• func RunTtopCollect(c *conf.CollectConf, hook shutdown.CancelHook) error

Variables

var LocalCollectCmd = &cobra.Command{
	Use:   "local-collect",
	Short: "retrieves all the dremio logs and diagnostics for the local node and saves the results in a compatible format for Dremio support",
	Long:  `Retrieves all the dremio logs and diagnostics for the local node and saves the results in a compatible format for Dremio support. This subcommand needs to be run with enough permissions to read the /proc filesystem, the dremio logs and configuration files`,
	Run: func(cobraCmd *cobra.Command, args []string) {
		overrides := make(map[string]string)

		cobraCmd.Flags().Visit(func(flag *pflag.Flag) {
			if flag.Name == conf.KeyDremioPatToken {
				if flag.Value.String() == "" {
					pat, err := masking.PromptForPAT()
					if err != nil {
						fmt.Printf("unable to get PAT due to: %v\n", err)
						os.Exit(1)
					}
					overrides[flag.Name] = pat
				} else {
					overrides[flag.Name] = flag.Value.String()
				}

				simplelog.Debugf("overriding yaml with cli flag %v and value 'REDACTED'", flag.Name)
			} else {
				simplelog.Debugf("overriding yaml with cli flag %v and value %q", flag.Name, flag.Value.String())
				overrides[flag.Name] = flag.Value.String()
			}
		})
		if patStdIn {
			var inputReader io.Reader = cobraCmd.InOrStdin()
			b, err := io.ReadAll(inputReader)
			if err != nil {
				fmt.Printf("\nCRITICAL ERROR: %v\n", err)
				os.Exit(1)
			}
			pat := strings.TrimSpace(string(b[:]))
			if pat != "" {
				overrides[conf.KeyDremioPatToken] = pat
			}
		}
		msg, err := Execute(args, overrides)
		if err != nil {
			fmt.Printf("\nCRITICAL ERROR: %v\n", err)
			os.Exit(1)
		}
		fmt.Println(msg)
	},
}

Functions

func Execute

func Execute(args []string, overrides map[string]string) (string, error)

func RunTtopCollect

func RunTtopCollect(c *conf.CollectConf, hook shutdown.CancelHook) error