Golang使用cobra实现命令行程序的示例代码

package main

import (
	"fmt"
	"github.com/spf13/cobra"
	"os"
)

var rootCmd = &cobra.Command{
	Use:   "myapp",
	Short: "一个简单的 Cobra 示例应用",
	Long:  "一个演示如何使用 Cobra 构建命令行应用的示例。",
	Run: func(cmd *cobra.Command, args []string) {
		fmt.Println("Hello from myapp!")
	},
}

var greetCmd = &cobra.Command{
	Use:   "greet",
	Short: "向用户打招呼",
	Long:  "向用户打招呼，可选择性地指定姓名。",
	Run: func(cmd *cobra.Command, args []string) {
		name, _ := cmd.Flags().GetString("name")
		fmt.Printf("你好，%s!\n", name)
	},
}

func Execute() {
	rootCmd.AddCommand(greetCmd)

	if err := rootCmd.Execute(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

func init() {
	greetCmd.Flags().StringP("name", "n", "用户", "指定姓名")
}

func main() {
	Execute()
}

# 运行主命令
go run main.go

# 运行子命令 greet，指定姓名为 John
go run main.go greet --name=John

package main

import (
	"fmt"
	"github.com/spf13/cobra"
	"os"
)

var (
	rootCmd = &cobra.Command{
		Use:   "myapp",
		Short: "一个进阶 Cobra 示例应用",
		Long:  "一个演示如何使用 Cobra 进行更进阶构建的命令行应用的示例。",
		Run: func(cmd *cobra.Command, args []string) {
			fmt.Println("Hello from myapp!")
		},
	}

	greetCmd = &cobra.Command{
		Use:   "greet",
		Short: "向用户打招呼",
		Long:  "向用户打招呼，可选择性地指定姓名。",
		Run: func(cmd *cobra.Command, args []string) {
			name, _ := cmd.Flags().GetString("name")
			formal, _ := cmd.Flags().GetBool("formal")

			greeting := "你好"
			if formal {
				greeting = "您好"
			}

			fmt.Printf("%s，%s!\n", greeting, name)
		},
	}

	advancedCmd = &cobra.Command{
		Use:   "advanced",
		Short: "一个进阶子命令",
		Long:  "一个演示更进阶子命令的示例。",
		Run: func(cmd *cobra.Command, args []string) {
			fmt.Println("Hello from advanced command!")
		},
	}
)

func Execute() {
	greetCmd.Flags().StringP("name", "n", "用户", "指定姓名")
	greetCmd.Flags().BoolP("formal", "f", false, "使用正式的打招呼语")

	rootCmd.AddCommand(greetCmd, advancedCmd)

	if err := rootCmd.Execute(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

func main() {
	Execute()
}

# 运行主命令
go run main.go

# 运行子命令 greet，指定姓名为 John，使用正式的打招呼语
go run main.go greet --name=John --formal

# 运行子命令 advanced
go run main.go advanced