使用go-zero从0到1快速构建高并发的短链服务

最新推荐文章于 2024-04-29 11:39:41 发布

置顶「已注销」

最新推荐文章于 2024-04-29 11:39:41 发布

阅读量4.4k

点赞数 3

分类专栏： Golang 文章标签：数据库 golang

原文链接：https://github.com/tal-tech/go-zero/blob/master/doc/shorturl.md

版权

Golang 专栏收录该内容

12 篇文章 0 订阅

订阅专栏

go-zero

关于go-zero介绍请看上一篇博文《好未来晓黑板go-zero微服务框架: 你不需要懂微服务，懂业务就行》

使用go-zero从0到1快速构建高并发的短链服务

0. 什么是短链服务？

短链服务就是将长的URL网址，通过程序计算等方式，转换为简短的网址字符串。

写此短链服务是为了从整体上演示go-zero构建完整微服务的过程，算法和实现细节尽可能简化了，所以这不是一个高阶的短链服务。

1. 短链微服务架构图

这里把shorten和expand分开为两个微服务，并不是说一个远程调用就需要拆分为一个微服务，只是为了最简演示多个微服务而已
后面的redis和mysql也是共用的，但是在真正项目里要尽可能每个微服务使用自己的数据库，数据边界要清晰

2. 准备工作

安装etcd, mysql, redis
准备goctl工具
直接从https://github.com/tal-tech/go-zero/releases下载最新版，后续会加上自动更新
- 也可以从源码编译，在任意目录下进行，目的是为了编译goctl工具
  1. git clone https://github.com/tal-tech/go-zero
  2. 在tools/goctl目录下编译goctl工具go build goctl.go
  3. 将生成的goctl放到$PATH下，确保goctl命令可运行
创建工作目录shorturl
在shorturl目录下执行go mod init shorturl初始化go.mod

3. 编写API Gateway代码

通过goctl生成shorturl.api并编辑，为了简洁，去除了文件开头的info，代码如下：

type (
	shortenReq struct {
		url string `form:"url"`
	}

	shortenResp struct {
		shortUrl string `json:"shortUrl"`
	}
)

type (
	expandReq struct {
		key string `form:"key"`
	}

	expandResp struct {
		url string `json:"url"`
	}
)

service shorturl-api {
	@server(
		handler: ShortenHandler
	)
	get /shorten(shortenReq) returns(shortenResp)

	@server(
		handler: ExpandHandler
	)
	get /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方法的路由、请求参数、返回参数等

使用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 OK
Content-Type: application/json
Date: Thu, 27 Aug 2020 14:31:39 GMT
Content-Length: 15

{"shortUrl":""}

可以看到我们API Gateway其实啥也没干，就返回了个空值，接下来我们会在rpc服务里实现业务逻辑

可以修改internal/svc/servicecontext.go来传递服务依赖（如果需要）
实现逻辑可以修改internal/logic下的对应文件
可以通过goctl生成各种客户端语言的api调用代码
到这里，你已经可以通过goctl生成客户端代码给客户端同学并行开发了，支持多种语言，详见文档

4. 编写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.yaml
Starting rpc server at 127.0.0.1:8080...

etc/shorten.yaml文件里可以修改侦听端口等配置

5. 编写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.yaml
Starting rpc server at 127.0.0.1:8081...

etc/expand.yaml文件里可以修改侦听端口等配置

6. 修改API Gateway代码调用shorten/expand rpc服务

修改配置文件shorter-api.yaml，增加如下内容

Shortener:
  Etcd:
    Hosts:
      - localhost:2379
    Key: shorten.rpc
Expander:
  Etcd:
    Hosts:
      - localhost:2379
    Key: expand.rpc

通过etcd自动去发现可用的shorten/expand服务

修改internal/config/config.go如下，增加shorten/expand服务依赖

type Config struct {
	rest.RestConf
	Shortener rpcx.RpcClientConf     // 手动代码
	Expander  rpcx.RpcClientConf     // 手动代码
}

修改internal/svc/servicecontext.go，如下：

type ServiceContext struct {
	Config    config.Config
	Shortener 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.Context
	logx.Logger
	expander 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.Context
	logx.Logger
	shortener 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

7. 定义数据库表结构，并生成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 cache

goctl model mysql ddl -c -src shorturl.sql -dir .

也可以用datasource命令代替ddl来指定数据库链接直接从schema生成

生成后的文件结构如下：

rpc/model
├── shorturl.sql
├── shorturlmodel.go              // CRUD+cache代码
└── vars.go                       // 定义常量和变量

8. 修改shorten/expand rpc代码调用crud+cache代码

修改rpc/expand/etc/expand.yaml，增加如下内容：
```
DataSource: root:@tcp(localhost:3306)/gozero
Table: shorturl
Cache:
  - Host: localhost:6379
```
可以使用多个redis作为cache，支持redis单点或者redis集群

修改rpc/expand/internal/config.go，如下：

type Config struct {
	rpcx.RpcServerConf
	DataSource string             // 手动代码
	Table      string             // 手动代码
	Cache      cache.CacheConf    // 手动代码
}

增加了mysql和redis cache配置

修改rpc/expand/internal/svc/servicecontext.go，如下：

type ServiceContext struct {
	c     config.Config
	Model *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.Context
	logx.Logger
	model *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)/gozero
Table: shorturl
Cache:
  - Host: localhost:6379
```
可以使用多个redis作为cache，支持redis单点或者redis集群

修改rpc/shorten/internal/config.go，如下：

type Config struct {
	rpcx.RpcServerConf
	DataSource string            // 手动代码
	Table      string            // 手动代码
	Cache      cache.CacheConf   // 手动代码
}

增加了mysql和redis cache配置

修改rpc/shorten/internal/svc/servicecontext.go，如下：

type ServiceContext struct {
	c     config.Config
	Model *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.Context
	logx.Logger
	model *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
  // 手动代码结束
}

至此代码修改完成，凡事手动修改的代码我加了标注

9. 完整调用演示

shorten api调用

curl -i "http://localhost:8888/shorten?url=http://www.xiaoheiban.cn"

返回如下：

HTTP/1.1 200 OK
Content-Type: application/json
Date: Sat, 29 Aug 2020 10:49:49 GMT
Content-Length: 21

{"shortUrl":"f35b2a"}

expand api调用

curl -i "http://localhost:8888/expand?key=f35b2a"

返回如下：

HTTP/1.1 200 OK
Content-Type: application/json
Date: Sat, 29 Aug 2020 10:51:53 GMT
Content-Length: 34

{"url":"http://www.xiaoheiban.cn"}

10. Benchmark

因为写入依赖于mysql的写入速度，就相当于压mysql了，所以压测只测试了expand接口，相当于从mysql里读取并利用缓存，shorten.lua里随机从db里获取了100个热key来生成压测请求

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-AAd7UeKi-1598715989029)(images/shorturl-benchmark.png)]

可以看出在我的MacBook Pro上能达到3万+的qps。

11. 总结

我们一直强调工具大于约定和文档。

go-zero不只是一个框架，更是一个建立在框架+工具基础上的，简化和规范了整个微服务构建的技术体系。

我们在保持简单的同时也尽可能把微服务治理的复杂度封装到了框架内部，极大的降低了开发人员的心智负担，使得业务开发得以快速推进。

通过go-zero+goctl生成的代码，包含了微服务治理的各种组件，包括：并发控制、自适应熔断、自适应降载、自动缓存控制等，可以轻松部署以承载巨大访问量。

「已注销」

关注

3
点赞
踩
15

收藏

觉得还不错? 一键收藏
2
评论
使用go-zero从0到1快速构建高并发的短链服务

go-zero关于go-zero介绍请看上一篇博文《好未来晓黑板go-zero微服务框架: 你不需要懂微服务，懂业务就行》使用go-zero从0到1快速构建高并发的短链服务0. 什么是短链服务？短链服务就是将长的URL网址，通过程序计算等方式，转换为简短的网址字符串。写此短链服务是为了从整体上演示go-zero构建完整微服务的过程，算法和实现细节尽可能简化了，所以这不是一个高阶的短链服务。1. 短链微服务架构图2. 准备工作安装etcd, mysql, redis准备goctl工具直
复制链接

扫一扫