你还在手撕微服务？快来用go-zero自动生成
kevwan@晓黑板 OSC开源社区 前天
0. 为什么说做好微服务很难？
要想做好微服务，我们需要理解和掌握的知识点非常多，从几个维度上来说：
基本功能层面
并发控制&限流，避免服务被突发流量击垮
服务注册与服务发现，确保能够动态侦测增减的节点
负载均衡，需要根据节点承受能力分发流量
超时控制，避免对已超时请求做无用功
熔断设计，快速失败，保障故障节点的恢复能力
高阶功能层面
请求认证，确保每个用户只能访问自己的数据
链路追踪，用于理解整个系统和快速定位特定请求的问题
日志，用于数据收集和问题定位
可观测性，没有度量就没有优化
对于其中每一点，我们都需要用很长的篇幅来讲述其原理和实现，那么对我们后端开发者来说，要想把这些知识点都掌握并落实到业务系统里，难度是非常大的，不过我们可以依赖已经被大流量验证过的框架体系。go-zero微服务框架就是为此而生。
另外，我们始终秉承工具大于约定和文档的理念。我们希望尽可能减少开发人员的心智负担，把精力都投入到产生业务价值的代码上，减少重复代码的编写，所以我们开发了goctl工具。
下面我通过短链微服务来演示通过go-zero快速的创建微服务的流程，走完一遍，你就会发现：原来编写微服务如此简单！
1. 什么是短链服务？
短链服务就是将长的URL网址，通过程序计算等方式，转换为简短的网址字符串。
写此短链服务是为了从整体上演示go-zero构建完整微服务的过程，算法和实现细节尽可能简化了，所以这不是一个高阶的短链服务。
2. 短链微服务架构图
这里把shorten和expand分开为两个微服务，并不是说一个远程调用就需要拆分为一个微服务，只是为了最简演示多个微服务而已
后面的redis和mysql也是共用的，但是在真正项目里要尽可能每个微服务使用自己的数据库，数据边界要清晰
3. 准备工作
安装etcd, mysql, redis
准备goctl工具
直接从
https://github.com/tal-tech/go-zero/releases下载最新版，后续会加上自动更新也可以从源码编译，在任意目录下进行，目的是为了编译goctl工具
git clone https://github.com/tal-tech/go-zero在
tools/goctl目录下编译goctl工具go build goctl.go将生成的goctl放到
$PATH下，确保goctl命令可运行创建工作目录
shorturl在
shorturl目录下执行go mod init shorturl初始化go.mod
4. 编写API Gateway代码
通过goctl生成
shorturl.api并编辑，为了简洁，去除了文件开头的info，代码如下：
type (shortenReq { url string `form:"url"`}
shortenResp {shortUrl string `json:"shortUrl"`})
type (expandReq {key string `form:"key"`}
expandResp {url string `json:"url"`})
service shorturl-api {@handler ShortenHandlerget /shorten(shortenReq) returns(shortenResp)
@handler ExpandHandlerget /expand(expandReq) returns(expandResp)}
type用法和go一致，service用来定义get/post/head/delete等api请求，解释如下：
service shorturl-api {这一行定义了service名字@server部分用来定义server端用到的属性handler定义了服务端handler名字get /shorten(shortenReq) returns(shortenResp)定义了get方法的路由、请求参数、返回参数等可以修改
internal/svc/servicecontext.go来传递服务依赖（如果需要）实现逻辑可以修改
internal/logic下的对应文件可以通过
goctl生成各种客户端语言的api调用代码到这里，你已经可以通过goctl生成客户端代码给客户端同学并行开发了，支持多种语言，详见文档
使用goctl生成API Gateway代码
goctl api go -api shorturl.api -dir api
生成的文件结构如下：
.├── api│ ├── etc│ │ └── shorturl-api.yaml // 配置文件│ ├── internal│ │ ├── config│ │ │ └── config.go // 定义配置│ │ ├── handler│ │ │ ├── expandhandler.go // 实现expandHandler│ │ │ ├── routes.go // 定义路由处理│ │ │ └── shortenhandler.go // 实现shortenHandler│ │ ├── logic│ │ │ ├── expandlogic.go // 实现ExpandLogic│ │ │ └── shortenlogic.go // 实现ShortenLogic│ │ ├── svc│ │ │ └── servicecontext.go // 定义ServiceContext│ │ └── types│ │ └── types.go // 定义请求、返回结构体│ └── shorturl.go // main入口定义├── go.mod├── go.sum└── shorturl.api启动API Gateway服务，默认侦听在8888端口
go run api/shorturl.go -f api/etc/shorturl-api.yaml
测试API Gateway服务
curl -i "http://localhost:8888/shorten?url=http://www.xiaoheiban.cn"
返回如下：
HTTP/1.1 200 OKContent-Type: application/jsonDate: Thu, 27 Aug 2020 14:31:39 GMTContent-Length: 15
{"shortUrl":""可以看到我们API Gateway其实啥也没干，就返回了个空值，接下来我们会在rpc服务里实现业务逻辑
5. 编写shorten rpc服务
在
rpc/shorten目录下编写shorten.proto文件可以通过命令生成proto文件模板
goctl rpc template -o shorten.proto
修改后文件内容如下：
syntax = "proto3"; package shorten; message shortenReq { string url = 1;} message shortenResp { string key = 1;} service shortener { rpc shorten(shortenReq) returns(shortenResp);}用
goctl生成rpc代码，在rpc/shorten目录下执行命令goctl rpc proto -src shorten.proto
文件结构如下：
rpc/shorten├── etc│ └── shorten.yaml // 配置文件├── internal│ ├── config│ │ └── config.go // 配置定义│ ├── logic│ │ └── shortenlogic.go // rpc业务逻辑在这里实现│ ├── server│ │ └── shortenerserver.go // 调用入口, 不需要修改│ └── svc│ └── servicecontext.go // 定义ServiceContext，传递依赖├── pb│ └── shorten.pb.go├── shorten.go // rpc服务main函数├── shorten.proto└── shortener ├── shortener.go // 提供了外部调用方法，无需修改 ├── shortener_mock.go // mock方法，测试用 └── types.go // request/response结构体定义直接可以运行，如下：
$ go run shorten.go -f etc/shorten.yamlStarting rpc server at 127.0.0.1:8080...etc/shorten.yaml文件里可以修改侦听端口等配置
6. 编写expand rpc服务
在
rpc/expand目录下编写expand.proto文件可以通过命令生成proto文件模板
goctl rpc template -o expand.proto
修改后文件内容如下
syntax = "proto3"; package expand; message expandReq { string key = 1;} message expandResp { string url = 1;} service expander { rpc expand(expandReq) returns(expandResp);}用
goctl生成rpc代码，在rpc/expand目录下执行命令goctl rpc proto -src expand.proto
文件结构如下：
rpc/expand├── etc│ └── expand.yaml // 配置文件├── expand.go // rpc服务main函数├── expand.proto├── expander│ ├── expander.go // 提供了外部调用方法，无需修改│ ├── expander_mock.go // mock方法，测试用│ └── types.go // request/response结构体定义├── internal│ ├── config│ │ └── config.go // 配置定义│ ├── logic│ │ └── expandlogic.go // rpc业务逻辑在这里实现│ ├── server│ │ └── expanderserver.go // 调用入口, 不需要修改│ └── svc│ └── servicecontext.go // 定义ServiceContext，传递依赖└── pb └── expand.pb.go修改
etc/expand.yaml里面的ListenOn的端口为8081，因为8080已经被shorten服务占用了修改后运行，如下：
$ go run expand.go -f etc/expand.yamlStarting rpc server at 127.0.0.1:8081...etc/expand.yaml文件里可以修改侦听端口等配置
7. 修改API Gateway代码调用shorten/expand rpc服务
修改配置文件
shorter-api.yaml，增加如下内容Shortener: Etcd: Hosts: - localhost:2379 Key: shorten.rpcExpander: Etcd: Hosts: - localhost:2379 Key: expand.rpc通过etcd自动去发现可用的shorten/expand服务
修改
internal/config/config.go如下，增加shorten/expand服务依赖
type Config struct {rest.RestConfShortener rpcx.RpcClientConf // 手动代码Expander rpcx.RpcClientConf // 手动代码}
修改
internal/svc/servicecontext.go，如下：type ServiceContext struct {Config config.ConfigShortener rpcx.Client // 手动代码Expander rpcx.Client // 手动代码} func NewServiceContext(config config.Config) *ServiceContext {return &ServiceContext{Config: config,Shortener: rpcx.MustNewClient(config.Shortener), // 手动代码Expander: rpcx.MustNewClient(config.Expander), // 手动代码}}
通过ServiceContext在不同业务逻辑之间传递依赖
修改
internal/logic/expandlogic.go，如下：type ExpandLogic struct {ctx context.Contextlogx.Loggerexpander rpcx.Client // 手动代码} func NewExpandLogic(ctx context.Context, svcCtx *svc.ServiceContext) ExpandLogic {return ExpandLogic{ctx: ctx,Logger: logx.WithContext(ctx),expander: svcCtx.Expander, // 手动代码}} func (l *ExpandLogic) Expand(req types.ExpandReq) (*types.ExpandResp, error) { // 手动代码开始resp, err := expander.NewExpander(l.expander).Expand(l.ctx, &expander.ExpandReq{Key: req.Key,})if err != nil {return nil, err} return &types.ExpandResp{Url: resp.Url,}, nil // 手动代码结束}增加了对
expander服务的依赖，并通过调用expander的Expand方法实现短链恢复到url修改
internal/logic/shortenlogic.go，如下：type ShortenLogic struct {ctx context.Contextlogx.Loggershortener rpcx.Client // 手动代码} func NewShortenLogic(ctx context.Context, svcCtx *svc.ServiceContext) ShortenLogic {return ShortenLogic{ctx: ctx,Logger: logx.WithContext(ctx),shortener: svcCtx.Shortener, // 手动代码}} func (l *ShortenLogic) Shorten(req types.ShortenReq) (*types.ShortenResp, error) { // 手动代码开始resp, err := shortener.NewShortener(l.shortener).Shorten(l.ctx, &shortener.ShortenReq{Url: req.Url,})if err != nil {return nil, err} return &types.ShortenResp{ShortUrl: resp.Key,}, nil // 手动代码结束}
增加了对shortener服务的依赖，并通过调用shortener的Shorten方法实现url到短链的变换
至此，API Gateway修改完成，虽然贴的代码多，但是期中修改的是很少的一部分，为了方便理解上下文，我贴了完整代码，接下来处理CRUD+cache
8. 定义数据库表结构，并生成CRUD+cache代码
shorturl下创建rpc/model目录：
mkdir -p rpc/model在rpc/model目录下编写创建shorturl表的sql文件
shorturl.sql，如下：CREATE TABLE `shorturl`( `shorten` varchar(255) NOT NULL COMMENT 'shorten key', `url` varchar(255) NOT NULL COMMENT 'original url', PRIMARY KEY(`shorten`)) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;创建DB和table
create database gozero;
source shorturl.sql;
在
rpc/model目录下执行如下命令生成CRUD+cache代码，-c表示使用redis cachegoctl model mysql ddl -c -src shorturl.sql -dir .
也可以用
datasource命令代替ddl来指定数据库链接直接从schema生成生成后的文件结构如下：
rpc/model├── shorturl.sql├── shorturlmodel.go // CRUD+cache代码└── vars.go // 定义常量和变量
9. 修改shorten/expand rpc代码调用crud+cache代码
修改
rpc/expand/etc/expand.yaml，增加如下内容：DataSource: root:@tcp(localhost:3306)/gozeroTable: shorturlCache: - Host: localhost:6379可以使用多个redis作为cache，支持redis单点或者redis集群
修改
rpc/expand/internal/config.go，如下：type Config struct {rpcx.RpcServerConfDataSource string // 手动代码Table string // 手动代码Cache cache.CacheConf // 手动代码}增加了mysql和redis cache配置
修改
rpc/expand/internal/svc/servicecontext.go，如下：type ServiceContext struct {c config.ConfigModel *model.ShorturlModel // 手动代码} func NewServiceContext(c config.Config) *ServiceContext {return &ServiceContext{c: c,Model: model.NewShorturlModel(sqlx.NewMysql(c.DataSource), c.Cache, c.Table), // 手动代码}}修改
rpc/expand/internal/logic/expandlogic.go，如下：type ExpandLogic struct {ctx context.Contextlogx.Loggermodel *model.ShorturlModel // 手动代码} func NewExpandLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ExpandLogic {return &ExpandLogic{ctx: ctx,Logger: logx.WithContext(ctx),model: svcCtx.Model, // 手动代码}} func (l *ExpandLogic) Expand(in *expand.ExpandReq) (*expand.ExpandResp, error) { // 手动代码开始res, err := l.model.FindOne(in.Key)if err != nil {return nil, err} return &expand.ExpandResp{Url: res.Url,}, nil // 手动代码结束}修改
rpc/shorten/etc/shorten.yaml，增加如下内容：DataSource: root:@tcp(localhost:3306)/gozeroTable: shorturlCache: - Host: localhost:6379可以使用多个redis作为cache，支持redis单点或者redis集群
修改
rpc/shorten/internal/config.go，如下：type Config struct {rpcx.RpcServerConfDataSource string // 手动代码Table string // 手动代码Cache cache.CacheConf // 手动代码}增加了mysql和redis cache配置
修改
rpc/shorten/internal/svc/servicecontext.go，如下：type ServiceContext struct {c config.ConfigModel *model.ShorturlModel // 手动代码} func NewServiceContext(c config.Config) *ServiceContext {return &ServiceContext{c: c,Model: model.NewShorturlModel(sqlx.NewMysql(c.DataSource), c.Cache, c.Table), // 手动代码}}修改
rpc/shorten/internal/logic/shortenlogic.go，如下：const keyLen = 6 type ShortenLogic struct {ctx context.Contextlogx.Loggermodel *model.ShorturlModel // 手动代码} func NewShortenLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ShortenLogic {return &ShortenLogic{ctx: ctx,Logger: logx.WithContext(ctx),model: svcCtx.Model, // 手动代码}} func (l *ShortenLogic) Shorten(in *shorten.ShortenReq) (*shorten.ShortenResp, error) { // 手动代码开始，生成短链接key := hash.Md5Hex([]byte(in.Url))[:keyLen]_, err := l.model.Insert(model.Shorturl{Shorten: key,Url: in.Url,})if err != nil {return nil, err} return &shorten.ShortenResp{Key: key,}, nil // 手动代码结束}至此代码修改完成，凡事手动修改的代码我加了标注
10. 完整调用演示
shorten api调用
curl -i "http://localhost:8888/shorten?url=http://www.xiaoheiban.cn"
返回如下：
HTTP/1.1 200 OKContent-Type: application/jsonDate: Sat, 29 Aug 2020 10:49:49 GMTContent-Length: 21 {"shortUrl":"f35b2a"}expand api调用
curl -i "http://localhost:8888/expand?key=f35b2a"返回如下：
HTTP/1.1 200 OKContent-Type: application/jsonDate: Sat, 29 Aug 2020 10:51:53 GMTContent-Length: 34 {"url":"http://www.xiaoheiban.cn"}
11. Benchmark
因为写入依赖于mysql的写入速度，就相当于压mysql了，所以压测只测试了expand接口，相当于从mysql里读取并利用缓存，shorten.lua里随机从db里获取了100个热key来生成压测请求
可以看出在我的MacBook Pro上能达到3万+的qps。
12. 总结
我们一直强调工具大于约定和文档。
go-zero不只是一个框架，更是一个建立在框架+工具基础上的，简化和规范了整个微服务构建的技术体系。
我们在保持简单的同时也尽可能把微服务治理的复杂度封装到了框架内部，极大的降低了开发人员的心智负担，使得业务开发得以快速推进。
通过go-zero+goctl生成的代码，包含了微服务治理的各种组件，包括：并发控制、自适应熔断、自适应降载、自动缓存控制等，可以轻松部署以承载巨大访问量。
13. 项目地址
https://github.com/tal-tech/go-zero