为了过一个愉快的周四,我用Go做了一款命令行

我像麋鹿一样在林荫中走着,为着自己的香气而发狂;夜晚是五月正中的夜晚,清风是南国的清风; 我迷路了,我游荡着,我寻求那得不到的东西,我得到了我所没有寻求的东西。 ------ 泰戈尔

前言

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

准备

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

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

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

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

实现

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

  1. 使用Go调用这个API,拿到返回值并返回
Go 复制代码
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)
}
  1. 使用Cobra做一个特定命令行
Go 复制代码
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文件中,这样就可以啦。

  1. 生成文件夹,将结果存入
Go 复制代码
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吧,可以拿着这个去和群友欢乐对线吧!

效果

相关推荐
企业数字化笔记1 小时前
固定资产历史数据怎么导入系统?Excel模板、字段映射和数据校验
android·java·数据库·后端
西瓜太郎12341 小时前
request_id 如何串起请求、路由、计费和日志
后端·python
by组态1 小时前
Ricon组态系统通信配置指南
前端·后端·物联网
sbjdhjd2 小时前
ThinkPHP 5.0.10 缓存写入型 RCE 复盘:从手动搭建、换行绕过到源码单步验证 | 05
java·后端·安全·spring·网络安全·数据挖掘·php
LEE2 小时前
Agent 的控制权,为什么正在回到模型手里?
前端·后端
沃夫上校2 小时前
跨域这件事,其实浏览器在替你挡刀
前端·后端·nginx
用户298698530142 小时前
Python 将 Word 文档转换为图片的实践指南
后端·python·api
旺仔不是程序员2 小时前
致命索引失效:索引列运算与函数索引不匹配
数据库·后端·面试
老孙讲技术2 小时前
把连锁门店轮询抓拍和遮挡叫醒接进督导台-setDeviceSnapEnhanced与setMessageCallback
后端·物联网·音视频开发
EatFan2 小时前
前后端分离项目中 Token 到底应该怎么设计?Access Token + Refresh Token 实战
java·前端·后端