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

相关推荐
Humbunklung15 小时前
VC++ 使用OpenSSL创建RSA密钥PEM文件
开发语言·c++·openssl
Humbunklung15 小时前
填坑:VC++ 采用OpenSSL 3.0接口方式生成RSA密钥
开发语言·c++·rsa·openssl 3.0
zl218786544817 小时前
Playwright同步、异步、并行、串行执行效率比较
开发语言·python·测试工具
Tony Bai17 小时前
【Go开发者的数据库设计之道】05 落地篇:Go 语言四种数据访问方案深度对比
开发语言·数据库·后端·golang
gopyer17 小时前
180课时吃透Go语言游戏后端开发3:Go语言中其他常用的数据类型
开发语言·游戏·golang·游戏后端开发
come1123417 小时前
Go vs. PHP:核心优势劣势对比
开发语言·golang·php
eqwaak018 小时前
Flask实战指南:从基础到高阶的完整开发流程
开发语言·后端·python·学习·flask
笨蛋不要掉眼泪18 小时前
SpringBoot项目Excel成绩录入功能详解:从文件上传到数据入库的全流程解析
java·vue.js·spring boot·后端·spring·excel
默|笙18 小时前
【c++】红黑树的部分实现
开发语言·c++
轩情吖19 小时前
Qt常用控件之QSpinBox
开发语言·c++·qt·控件·桌面级开发·qspinbox·微调框