Go 语言中的命令行参数操作详解

// main.go
package main
import (
	"fmt"
	"os"
	"strconv"
)
func main() {
	// os.Args 至少包含程序名，所以长度至少为 1
	// 我们需要两个参数，所以长度至少为 3
	if len(os.Args) < 3 {
		fmt.Println("用法: adder <数字1> <数字2>")
		os.Exit(1) // 非零退出码表示错误
	}
	// 获取参数
	arg1 := os.Args[1]
	arg2 := os.Args[2]
	// 将字符串转换为整数
	num1, err := strconv.Atoi(arg1)
	if err != nil {
		fmt.Printf("错误: '%s' 不是一个有效的整数\n", arg1)
		os.Exit(1)
	}
	num2, err := strconv.Atoi(arg2)
	if err != nil {
		fmt.Printf("错误: '%s' 不是一个有效的整数\n", arg2)
		os.Exit(1)
	}
	// 计算并打印结果
	sum := num1 + num2
	fmt.Printf("结果: %d + %d = %d\n", num1, num2, sum)
}

# 编译
go build -o adder main.go
# 运行
./adder 10 25
# 输出: 结果: 10 + 25 = 35
# 错误情况
./adder 10 abc
# 输出: 错误: 'abc' 不是一个有效的整数

// main.go
package main
import (
	"flag"
	"fmt"
)
func main() {
	// 定义标志
	// flag.String(标志名, 默认值, 帮助信息)
	// 返回一个字符串指针
	port := flag.Int("port", 8080, "Web 服务器监听的端口号")
	host := flag.String("host", "localhost", "Web 服务器绑定的主机名")
	verbose := flag.Bool("v", false, "启用详细输出模式")
	// 定义一个简短形式的标志，例如 -h 可以是 --host 的简写
	// flag 包本身不支持直接别名，但可以通过定义两个变量指向同一个值来实现
	// 这里我们只演示标准用法
	// 解析命令行参数
	// 必须在所有 flag 定义之后，在访问它们的值之前调用
	flag.Parse()
	// 使用标志的值（通过解引用指针 *port）
	fmt.Println("--- 配置信息 ---")
	fmt.Printf("主机: %s\n", *host)
	fmt.Printf("端口: %d\n", *port)
	fmt.Printf("详细模式: %t\n", *verbose)
	// 获取非标志参数
	// 例如: ./webserver --port=9999 file1.txt file2.txt
	// flag.Args() 将返回 [file1.txt file2.txt]
	args := flag.Args()
	if len(args) > 0 {
		fmt.Println("\n--- 其他文件参数 ---")
		for i, arg := range args {
			fmt.Printf("参数 %d: %s\n", i+1, arg)
		}
	}
	fmt.Println("\n服务器模拟启动...")
}

# 编译
go build -o webserver main.go
# 使用默认值运行
./webserver
# 输出:
# --- 配置信息 ---
# 主机: localhost
# 端口: 8080
# 详细模式: false
# 服务器模拟启动...
# 使用自定义标志运行
./webserver --port=9999 --host=0.0.0.0 -v
# 输出:
# --- 配置信息 ---
# 主机: 0.0.0.0
# 端口: 9999
# 详细模式: true
# ...
# 混合使用标志和位置参数
./webserver -v config.yaml data.json
# 输出:
# --- 配置信息 ---
# 主机: localhost
# 端口: 8080
# 详细模式: true
#
# --- 其他文件参数 ---
# 参数 1: config.yaml
# 参数 2: data.json
# ...
# 自动生成帮助信息
./webserver -h
# 或
./webserver --help
# 输出:
# Usage of ./webserver:
#   -host string
#         Web 服务器绑定的主机名 (default "localhost")
#   -port int
#         Web 服务器监听的端口号 (default 8080)
#   -v    启用详细输出模式

# 安装 Cobra CLI 工具
go install github.com/spf13/cobra-cli@latest
# 在你的项目中添加 Cobra 依赖
go get -u github.com/spf13/cobra@latest

mkdir mygit && cd mygit
go mod init mygit
# 使用 cobra-cli 初始化
cobra-cli init

# 添加 commit 子命令
cobra-cli add commit
# 添加 clone 子命令
cobra-cli add clone

// cmd/root.go
package cmd
import (
	"fmt"
	"os"
	"github.com/spf13/cobra"
)
var cfgFile string // 用于存储 --config 标志的值
// rootCmd represents the base command when called without any subcommands
var rootCmd = &cobra.Command{
	Use:   "mygit",
	Short: "一个模拟 Git 的版本控制工具",
	Long:  `MyGit 是一个用 Go 和 Cobra 库编写的命令行应用程序，用于演示如何构建复杂的 CLI 工具。它模拟了 Git 的一些基本功能。`,
}
// Execute adds all child commands to the root command and sets flags appropriately.
// This is called by main.main(). It only needs to happen once to the rootCmd.
func Execute() {
	err := rootCmd.Execute()
	if err != nil {
		os.Exit(1)
	}
}
func init() {
	// 在这里定义全局标志和配置。
	// Cobra 支持持久性标志，该标志对此命令及其每个子命令都可用。
	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "配置文件 (默认为 $HOME/.mygit.yaml)")
	// Cobra 也支持本地标志，该标志仅对直接调用此命令时可用。
	// rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
}

// cmd/commit.go
package cmd
import (
	"fmt"
	"github.com/spf13/cobra"
)
var commitMessage string
// commitCmd represents the commit command
var commitCmd = &cobra.Command{
	Use:   "commit",
	Short: "记录对仓库的更改",
	Long:  `Commit 命令用于将暂存区的更改保存到本地仓库的历史记录中。它需要一个提交信息来描述这次更改的内容。`,
	Run: func(cmd *cobra.Command, args []string) {
		fmt.Println("执行 commit 命令...")
		if commitMessage == "" {
			fmt.Println("错误: 必须使用 -m 标志提供提交信息。")
			return
		}
		fmt.Printf("提交信息: \"%s\"\n", commitMessage)
		fmt.Println("成功提交更改！")
	},
}
func init() {
	// 将 commit 命令添加到根命令
	rootCmd.AddCommand(commitCmd)
	// 在这里定义 commit 命令的标志和配置。
	// StringVarP 是 StringVar 的增强版，支持简写形式，例如 -m
	// commitCmd.Flags().StringVarP(&commitMessage, "message", "m", "", "提交信息")
	// 更标准的写法是直接使用 "m" 作为简写
	commitCmd.Flags().StringVarP(&commitMessage, "message", "m", "", "提交信息 (必需)")
	
	// 可以将标志标记为必需
	commitCmd.MarkFlagRequired("message")
}

// cmd/clone.go
package cmd
import (
	"fmt"
	"github.com/spf13/cobra"
)
// cloneCmd represents the clone command
var cloneCmd = &cobra.Command{
	Use:   "clone [url]",
	Short: "将一个仓库克隆到新目录中",
	Long:  `Clone 命令用于从远程 URL 下载一个仓库，并创建一个本地副本。`,
	Args: cobra.ExactArgs(1), // 验证必须有且仅有 1 个参数
	Run: func(cmd *cobra.Command, args []string) {
		fmt.Println("执行 clone 命令...")
		repoURL := args[0]
		fmt.Printf("正在从 '%s' 克隆仓库...\n", repoURL)
		fmt.Println("克隆完成！")
	},
}
func init() {
	rootCmd.AddCommand(cloneCmd)
}

// main.go
package main
import "mygit/cmd"
func main() {
	cmd.Execute()
}

# 编译整个项目
go build -o mygit .
# 查看根命令帮助
./mygit -h
# 输出非常详细和美观的帮助信息
# 查看子命令帮助
./mygit commit -h
./mygit clone -h
# 运行 commit 命令
./mygit commit -m "Initial commit"
# 输出:
# 执行 commit 命令...
# 提交信息: "Initial commit"
# 成功提交更改！
# 测试必需标志
./mygit commit
# 输出:
# Error: required flag(s) "message" not set
# Usage:...
# (会显示帮助信息)
# 运行 clone 命令
./mygit clone https://github.com/user/repo.git
# 输出:
# 执行 clone 命令...
# 正在从 'https://github.com/user/repo.git' 克隆仓库...
# 克隆完成！
# 测试参数验证
./mygit clone
# 输出:
# Error: accepts 1 arg(s), received 0
# Usage:...