Go基础(③Cobra)

Cobra 是帮你快速开发命令行工具的框架

假设你想做一个叫 todo 的命令行工具,实现这些功能:

todo add "买牛奶" → 添加待办

todo list → 查看所有待办

todo done 1 → 标记第 1 个待办为已完成

没有 Cobra 的话,你需要自己写代码判断用户输入的是 add 还是 list,还要解析后面的参数,非常麻烦。

用 Cobra 的话,你只需要:

定义 add、list、done 这三个子命令

给每个命令写好具体执行逻辑

剩下的「解析用户输入」「判断执行哪个命令」「生成帮助信息(如 todo --help)」这些事,Cobra 全帮你做了

创建项目

bash 复制代码
mkdir cobra-demo && cd cobra-demo
go mod init cobra-demo

# 安装 Cobra 库
go get github.com/spf13/cobra@latest

# 安装 Cobra 命令行生成工具(可选,用于快速创建项目结构)
go install github.com/spf13/cobra-cli@latest

# 用 cobra-cli 初始化项目(会生成基本结构)
cobra-cli init

初始化后,项目结构如下

bash 复制代码
cobra-demo/
├── cmd/
│   └── root.go   # 根命令定义
├── go.mod
├── go.sum
└── main.go       # 程序入口

add.go代码

Go 复制代码
package cmd

import (
	"fmt"
	"strconv"

	"github.com/spf13/cobra"
)

// 定义 add 命令
var addCmd = &cobra.Command{
	Use:   "add",                               // 命令名称(使用方式:app add)
	Short: "将两个数字相加",                    // 短描述(帮助信息中显示)
	Long:  `将两个整数相加并打印结果。`,          // 长描述
	Args:  cobra.ExactArgs(2),                  // 必须传入 2 个参数
	Run: func(cmd *cobra.Command, args []string) {
		// 解析参数(字符串转整数)
		a, err := strconv.Atoi(args[0])
		if err != nil {
			fmt.Println("第一个参数必须是整数")
			return
		}

		b, err := strconv.Atoi(args[1])
		if err != nil {
			fmt.Println("第二个参数必须是整数")
			return
		}

		// 执行加法并输出结果
		fmt.Printf("%d + %d = %d\n", a, b, a+b)
	},
}

// 初始化函数:将 add 命令添加到根命令
func init() {
	rootCmd.AddCommand(addCmd)
}

然后把add.go放入cmd文件夹,然后运行

Go 复制代码
go build -o calc.exe

123

相关推荐
楼田莉子7 小时前
C++算法专题学习:栈相关的算法
开发语言·c++·算法·leetcode
晨非辰7 小时前
#C语言——刷题攻略:牛客编程入门训练(九):攻克 分支控制(三)、循环控制(一),轻松拿捏!
c语言·开发语言·经验分享·学习方法·visual studio
_oP_i7 小时前
Java 服务接口中解决跨域(CORS,Cross-Origin Resource Sharing)问题
java·开发语言
陈序猿(代码自用版)7 小时前
【考研C语言编程题】数组元素批量插入实现(含图示+三部曲拆解)
c语言·开发语言·考研
唐•苏凯7 小时前
ArcGIS Pro 遇到严重的应用程序错误而无法启动
开发语言·javascript·ecmascript
cci7 小时前
使用proxychains4反向代理解决依赖下载难题
后端
咖啡Beans8 小时前
Python工具DrissionPage推荐
后端·python
华仔啊8 小时前
工作5年没碰过分布式锁,是我太菜还是公司太稳?网友:太真实了!
java·后端
摸鱼仙人~8 小时前
深入理解 MyBatis-Plus 的 `BaseMapper`
java·开发语言·mybatis