Golang命令行工具开发怎么优化？Golang cobra库使用指南

go install github.com/spf13/cobra@latest

cobra init my-cli
cd my-cli

cobra add command1
cobra add command2

cmd/command1.go

package cmd

import (
    "fmt"

    "github.com/spf13/cobra"
)

var command1Cmd = &cobra.Command{
    Use:   "command1",
    Short: "A brief description of command1",
    Long: `A longer description that spans multiple lines and likely contains
examples and usage of using your command. For example:

Cobra is a CLI library for Go that empowers applications.
This application is a tool to generate the needed files
to quickly create a Cobra application.`,
    Run: func(cmd *cobra.Command, args []string) {
        fmt.Println("command1 called")
    },
}

func init() {
    rootCmd.AddCommand(command1Cmd)

    // Here you will define your flags and configuration settings.

    // Cobra supports persistent flags, which, if defined here,
    // will be global for your application.

    // rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.my-cli.yaml)")

    // Cobra also supports local flags, which will only run
    // when this action is called directly.
    // command1Cmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}

init

command1Cmd.Flags().StringP("name", "n", "", "Your name")

Run

name, _ := cmd.Flags().GetString("name")
fmt.Printf("Hello, %s!\n", name)

my-cli help command1

my-cli/
├── cmd/
│   ├── module1/
│   │   ├── command1.go
│   │   └── command2.go
│   └── module2/
│       ├── command3.go
│       └── command4.go
├── internal/
│   └── ...
├── main.go
└── ...

internal

command1_test.go

package cmd

import (
    "bytes"
    "strings"
    "testing"

    "github.com/spf13/cobra"
)

func executeCommand(root *cobra.Command, args ...string) (string, error) {
    buf := new(bytes.Buffer)
    root.SetOutput(buf)
    root.SetArgs(args)

    _, err := root.ExecuteC()
    return buf.String(), err
}

func TestCommand1(t *testing.T) {
    output, err := executeCommand(rootCmd, "command1")
    if err != nil {
        t.Fatalf("Error executing command: %v", err)
    }

    if !strings.Contains(output, "command1 called") {
        t.Errorf("Expected output to contain 'command1 called', got: %s", output)
    }
}

command1

go install

go install github.com/your-username/my-cli@latest