go-zero jwt 鉴权快速实战

前面我们分享了 go-zero 的快速实战以及日志组件的剖析，本次我们来实战使用 go-zero jwt 鉴权

本次文章主要是分享关于 go-zero 中 jwt 的使用方式，会以一个 demo 的方式来进行实战，对于使用 goctl 工具以及安装细节就不在赘述，有需要的话可以查看：

• 官网

本次文章主要分为如下几个部分：

• Jwt 的简单介绍
• Go-zero 中使用 jwt 实战

Jwt 的简单介绍

关于 jwt 鉴权的细节和原理，感兴趣的朋友可以查看历史文章：JWT身份认证(附带源码讲解) | GO主题月

那么我们如何识别什么时候需要使用 jwt 呢？

• 用于授权

例如某个系统通过例如账号密码登录之后，后台会生成一个 jwt，这个用户在这个系统之后的任何操作，都会去校验这个 jwt，就不需要用户操作系统内其他模块的时候，还去进行一次登录

当然，这是需要我们做好设定，这个 jwt 针对哪一些路由可以使用，从而允许用户访问该令牌允许的路由，服务和资源

• 用于信息交换

因为 jwt 可以与各方进行安全的传输，内部使用了签名算法，公钥加密，私钥解密，而且 jwt 的数据各种中有标头，有效载荷，以及其他的签发时间，过期时间，颁发人等等，可以用来校验信息是否被篡改了

Go-zero 中使用 jwt 实战

话不多说，咱们就来开始实战吧，先来给自己提一个需求：

需求

• 同学们平时去图书馆，都是需要登录自己的账号，才可以进入系统查询书籍余量的
• 那么我们就来实现一下，用账号密码登录图书系统，并生成一个 jwt，后续该用户进行书籍查询的时候，就可以直接使用 jwt 来进行鉴权

分析

那么，根据上述需求，显然，咱们会使用到数据库，本次这里咱们仍然使用 mysql 进行演示，并且会涉及到用户表和图书表

我们可以现在创建一下这两张表

• user table

user.sql

CREATE TABLE `user`
(
  `stu_id` varchar(255) NOT NULL COMMENT 'stu_id',
  `name` varchar(255) NOT NULL COMMENT 'name',
  `password` varchar(255) NOT NULL COMMENT 'password',
  `gender` varchar(255) NOT NULL COMMENT 'gender',
  PRIMARY KEY(`stu_id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

• book table

book.sql

CREATE TABLE `book`
(
  `book_id` varchar(255) NOT NULL COMMENT 'stu_id',
  `name` varchar(255) NOT NULL COMMENT 'name',
  `count` INTEGER (255) NOT NULL COMMENT 'password',
  PRIMARY KEY(`book_id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

咱们可以向表中插入一些 demo 数据，便于后续使用

insert into book(book_id,name,count)values("000001","kubernetes in action",99);
insert into book(book_id,name,count)values("000002","effective go",88);
insert into book(book_id,name,count)values("000003","穷爸爸富爸爸",21);

insert into user(stu_id,name,password,gender)values("13141549", "小胖", "Aa123123", "男");
insert into user(stu_id,name,password,gender)values("15161766", "阿强", "6r*,oo", "男");
insert into user(stu_id,name,password,gender)values("1325590", "marry", "123456", "女");

开始实战 go-zero 的 jwt

先来看看基本的代码目录

1. 新建代码目录，并编写 book 和 user 的 api

新建目录

mkdir my_test_demo/my_book_sys/book/api -p
mkdir my_test_demo/my_book_sys/book/model -p
mkdir my_test_demo/my_book_sys/user/api -p
mkdir my_test_demo/my_book_sys/user/rpc -p
mkdir my_test_demo/my_book_sys/user/model -p
cd my_test_demo
go mod init my_test_demo
go mod tidy

编写 api 文件 并生成 go 代码

book.api

定义 book api

• GET /search/do 查询书籍接口，使用 jwt: Auth 进行标识
  • 通过请求书名，鉴权完毕之后，响应响应书名对应的剩余数量

user.api

定义 user api

• 定义 POST /user/login 接口
  • 通过账号密码请求接口，内部校验完毕之后，返回 jwt token

\

使用工具将上述 api 生成 go 对应的代码

cd my_test_demo/my_book_sys/book/api
vim book.api
goctl api go -api book.api  -dir .

cd my_test_demo/my_book_sys/user/api
vim user.api
goctl api go -api user.api  -dir .

1. 编写数据库的 sql 语句，使用 goctl 生成 go 代码

• 直接将上述的 ****book.sql 拷贝到 my_test_demo/my_book_sys/book/model 下
• 将 user.sql 拷贝到 ****my_test_demo/my_book_sys/user/model 下

生成数据库相关的 go 代码文件

cd my_test_demo/my_book_sys/book/model
goctl model mysql ddl -src book.sql -dir .

cd my_test_demo/my_book_sys/user/model
goctl model mysql ddl -src user.sql -dir .

这个时候，咱们就已经完成了大部分的准备工作，接下来我们来查看一下代码目录

添加数据库配置和 jwt 相关配置

User 部分

1. 修改 my_test_demo/my_book_sys/user/api/etc/user-api.yaml ，加上数据库配置和 Auth 配置

• DataSource 数据库配置
• Auth jwt 配置
  • AccessSecret jwt 需要的密钥
  • AccessExpire 过期时间，单位 秒

  

1. 修改 my_test_demo/my_book_sys/user/api/internal/config/config.go 增加配置对应的数据结构

1. 补充 svc 层的数据结构，修改 my_test_demo/my_book_sys/user/api/internal/svc/servicecontext.go ，以及补充 NewServiceContext 的实现

1. 补充咱们的核心逻辑层的代码，my_test_demo/my_book_sys/user/api/internal/logic/loginlogic.go，大体逻辑如下

• 校验入参，去掉输入的用户名和密码的前后空格
• 查询数据库进行数据校验，此处我们手动在 my_test_demo/my_book_sys/user/model/usermodel_gen.go 文件中添加了数据库操作的 FindOneByName 方法
• 校验通过之后生成 jwt token 进行响应

此处的 model 方法可以加在 my_test_demo/my_book_sys/user/model/usermodel_gen.go

Book 部分

1. 修改 my_test_demo/my_book_sys/book/api/etc/book-api.yaml ，加上数据库配置和 Auth 配置

其中 AccessSecret 和 AccessExpire 的字段名和值保持和上述 user 路径下的内容一致

Name: search-api
Host: 0.0.0.0
Port: 9001
DataSource: root:123456@tcp(localhost:3306)/test_demo
Auth:
  AccessSecret: secretoooppppoooo
  AccessExpire: 3600

1. 同理，添加配置对应的数据结构，修改 my_test_demo/my_book_sys/book/api/internal/config/config.go

type Config struct {
   rest.RestConf
   DataSource string
   Auth struct {
      AccessSecret string
      AccessExpire int64
   }
}

1. 补充 svc 层的数据结构，修改 my_test_demo/my_book_sys/book/api/internal/svc/servicecontext.go ，以及补充 NewServiceContext 的实现

1. 补充咱们的核心逻辑层的代码，my_test_demo/my_book_sys/book/api/internal/logic/searchlogic.go，大体逻辑如下

• 通过书名查询数据库
• 返回具体书名的剩余数量

上述代码中 l.svcCtx.BookHttpModel.FindOneByName 是我们自定义修改了 model 下的文件内容，新增 FindOneByName 方法 ，修改 my_test_demo/my_book_sys/book/model/bookmodel_gen.go

验收成果

咱们分别进入到项目对应的 api 路径下，启动服务

终端1：

cd my_test_demo/my_book_sys/book/api
go run search.go

终端2：

cd my_test_demo/my_book_sys/user/api
go run user.go

终端3：

1. 咱们先来请求 book 的 api，在没有登录系统的情况下来查询书籍，看看是什么样的效果

curl -i -X GET \
  'http://localhost:9001/search/do?name=effective%20go'

1. 很明显，服务给我们返回了一个未授权，说明咱们的鉴权机制此处是有效果的，那么接下来登录一个用户

// post 用户登录
  curl -i -X POST   http://localhost:9002/user/login   -H 'Content-Type: application/json'   -d '{
    "username":"小胖",
    "password":"Aa123123"
}'

可以看到是登录成功了，且服务端给我们返回了对应的 jwt token，接下来咱们执行第一步，请求查询一下数据的数量

    // get 书籍
curl -i -X GET \
  'http://localhost:9001/search/do?name=effective%20go'  -H 'Authorization: eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE2NjA5NzMwNDEsImlhdCI6MTY2MDk2OTQ0MSwic3R1SWQiOiIxMzE0MTU0OSJ9.Jae_5GPr-xuB2mfqospfisV93ReUnGTZJ87WsIQ-bhE'

很明显，此处的鉴权机制 ok，鉴权完毕之后，正确查询数据库，查询到我们期望的书籍剩余数量

兄弟们，动起手来吧，先来应用 go-zero 的 jwt ，再去看 go-zero 的实际源码细节，很快就能明白

具体的源码地址可以查看：https://github.com/qingconglaixueit/my_test_Demo

至此，本篇内容结束