（服务计算）CLI 命令行实用程序开发实战 - Agenda

本文链接： https://blog.csdn.net/sodifferent/article/details/102672543

goonline分享地址：http://139.9.57.167:20080/share/bmn4she76kvs669u22jg?secret=false

任务描述

使用cobra实现简单的agenda，由于是经过简化的项目，所以只实现了两个功能，一个用户的注册，一个用户的登录。

任务要求

使用json储存用户信息
将最近使用的用户使用curUser.txt文件中
使用log记录命令执行情况

实现过程

首先是安装cobra
在$GOPATH/src/golang.org/x目录下执行git https://github.com/golang/text与git https://github.com/golang/sys，然后再执行go get -v github.com/spf13/cobra/cobra获取包，最后执行go install github.com/spf13/cobra/cobra进行安装。
创建cobra项目
执行cobra init --pkg-name demo，其中demo为项目名称创建项目，创建成功后会在src/目录下生成对应的demo文件，在此文件夹下会有cmd文件夹与main.go。
添加命令
添加register命令，执行cobra add register，即可在cmd文件夹下看到register.go文件，然后对此文件进行编写。在init函数中可以获取用户的命令行输入，具体方法是调用registerCmd.Flags()来获取，后面加.StringP表示获取一个string变量。

然后在registercmd=&cobra.Command中对输入的参数进行识别操作。首先调用函数获取输入的参数：

然后对输入的用户名进行查找，在data.json中寻找这个用户是否被注册（data.json文件用来储存注册的用户信息）。

其中readinfo函数是用来获取整个json文件中的所有信息：

如果在用户信息中有这个用户，就返回错误信息，如果没有，就注册成功，并保存在用户信息里，然后将命令信息记录在log文件中。

在这里写入的时候由于是对json文件的写入，所以需要将其转换为相应的格式：
```
us := readinfo()
us.Id=append(us.Id,str)
data, _:= json.Marshal(us)
saveinfo(data)
```
添加命令login：cobra add login，然后打开cmd目录下的login.go进行编写。
同样首先在init函数中识别命令行的输入的参数，然后在Run函数中对其进行处理。对登录的用户的识别通过输入的用户名和密码来进行匹配，首先获取整个文件里的用户信息，然后对其进行遍历，寻找是否有匹配的。

判断完毕后同样返回给用户登录情况，然后将相关信息写入log文件中。

这里登录成功之后还需要将此用户信息写入curUser.txt中储存起来，表示目前正在操作的用户。