使用Golang编写一个简单的命令行工具

 更新时间:2023年12月28日 10:39:29   作者:陪我去看海  
Cobra是一个强大的开源工具,能够帮助我们快速构建出优雅且功能丰富的命令行应用,本文将利用Cobra编写一个简单的命令行工具,感兴趣的可以了解下

前言

每周四,在群里总是会看到一些疯狂星期四的小故事,而且个个的小故事还挺有趣,但是让我这样的人肯定是编不出来了,在想刚好在学习Go,是不是可以怎么操作一下,写个命令行来获取周四文案啥的,这样就不用自己去编了。既然有了想法那就开干

准备

源码地址,结合看最好,下面代码并不完全,大佬们有好的想法,欢迎issues,pr,也欢迎star,一起整活

  • 了解使用Go写命令行的库
  • 找一个生成疯狂星期四文案的接口(后期也可以做自动生成)
  • 通过特定的命令来调用这个接口并拿到这个文案

在这里我了解到一个用Go做命令行的流行库cobra,据官方说这个被用在了很多大型项目,像Kubernetes, Hugo, GitHub CLI等等

找到的免费生成疯狂星期四文案的接口https://api.jixs.cc/api/wenan-fkxqs/index.php

实现

首先量化需求,既然已接口的形式的话,其实无非就是调用接口,拿到返回值了,只是把这个步骤放在了用户命令有对应操作之后执行,明确需求了,就可以开始写代码了

1.使用Go调用这个API,拿到返回值并返回

package cmd

import (
	"io"
	"log"
	"net/http"
)

// Fetch 发起请求
func Fetch() string {
	client := &http.Client{}
	req, _ := http.NewRequest("GET", "https://api.jixs.cc/api/wenan-fkxqs/index.php", nil)
	res, err := client.Do(req)
	if err != nil {
		log.Fatal("Http get error is ", err)
	}
	if res.StatusCode != http.StatusOK {
		log.Fatal("Http status code is ", res.StatusCode)
	}
	defer func(Body io.ReadCloser) {
		err := Body.Close()
		if err != nil {
			log.Fatal(err)
		}
	}(res.Body)
	bytes, err := io.ReadAll(res.Body)
	if err != nil {
		return ""
	}
	return string(bytes)
}

2.使用Cobra做一个特定命令行

package cmd

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

var ThursdayCmd = &cobra.Command{
	Use:   "Thursday [文案]",
	Short: "crazy Thursday",
	Long:  `crazy Thursday`,
	Run: func(cmd *cobra.Command, args []string) {
		res := Fetch()
		Print(res)
		fmt.Println(res)
	},
}

var rootCmd = &cobra.Command{
	Use: "ct",
}

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

这样当我们执行 go run main.go Thursday 后会在控制台打印出获取到的文案

但是我使用 GoLand 在这一步有点问题,就是当文案特别长的时候,他在控制台还是会强行显示一行,无法复制到全部,所以这时想起一种解决方法,那就是将文案存入一个txt文件中,这样就可以啦。

3.生成文件夹,将结果存入

package cmd

import (
	"fmt"
	"log"
	"os"
)

func Print(res string) {
	// 创建文件
	file, err := os.Create("./output/output.txt")
	if err != nil {
		fmt.Println("无法创建文件:", err)
		return
	}
	defer func(file *os.File) {
		err := file.Close()
		if err != nil {
			log.Fatal(err)
		}
	}(file)

	// 将内容写入文件
	_, err = file.WriteString(res)
	if err != nil {
		fmt.Println("无法写入文件:", err)
		return
	}

	fmt.Println("内容已成功写入文件 output.txt")
}

到此基本上就完成了,算是一个练习的小demo吧,可以拿着这个去和群友欢乐对线吧!

效果

到此这篇关于使用Golang编写一个简单的命令行工具的文章就介绍到这了,更多相关Go命令行内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

相关文章

  • Go使用Pipeline实现一个简洁而高效的数据处理流水线

    Go使用Pipeline实现一个简洁而高效的数据处理流水线

    在并发编程中,流水线Pipeline是一种常见的设计模式,它将一个复杂任务拆解为多个独立步骤,由多个协程并行处理并通过通道传递数据,Go语言天生支持这种模型,能显著提高数据处理的性能和可读性,本文将给大家介绍如何使用Go实现一个简洁而高效的数据处理流水线
    2025-08-08
  • 一文详解下划线字段在golang结构体中的应用

    一文详解下划线字段在golang结构体中的应用

    这篇文章主要为大家详细介绍了下划线字段在golang结构体中应用的相关知识,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下
    2025-08-08
  • Golang 依赖注入经典解决方案uber/fx理论解析

    Golang 依赖注入经典解决方案uber/fx理论解析

    这篇文章主要为大家介绍了Golang依赖注入经典解决方案uber/fx理论解析,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-05-05
  • 如何组织Go代码目录结构依赖注入wire使用解析

    如何组织Go代码目录结构依赖注入wire使用解析

    这篇文章主要为大家介绍了如何组织Go代码目录结构依赖注入wire使用解析,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-07-07
  • go语言Pflag Viper Cobra 核心功能使用介绍

    go语言Pflag Viper Cobra 核心功能使用介绍

    这篇文章主要为大家介绍了go语言Pflag Viper Cobra 核心功能使用介绍,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-09-09
  • golang并发安全及读写互斥锁的示例分析

    golang并发安全及读写互斥锁的示例分析

    这篇文章主要为大家介绍了golang并发安全及读写互斥锁的示例分析,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步早日升职加薪
    2022-04-04
  • 详解Go中Map类型和Slice类型的传递

    详解Go中Map类型和Slice类型的传递

    这篇文章主要为大家详细介绍了Go中Map类型和Slice类型的传递,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2017-11-11
  • Go使用协程交替打印字符

    Go使用协程交替打印字符

    这篇文章主要介绍了Go使用协程交替打印字符,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
    2021-04-04
  • 一文带你入门Go语言中定时任务库Cron的使用

    一文带你入门Go语言中定时任务库Cron的使用

    在平时的开发需求中,我们经常会有一些重复执行的操作需要触发执行,说白了就是定时任务。这篇文章主要给大家介绍一下如何在go项目中实现一个crontab功能,感兴趣的小伙伴们可以参考借鉴,希望对大家能有所帮助
    2022-08-08
  • golang如何实现mapreduce单进程版本详解

    golang如何实现mapreduce单进程版本详解

    这篇文章主要给大家介绍了关于golang如何实现mapreduce单进程版本的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧。
    2018-01-01

最新评论