短链服务就是将长的URL网址,通过程序计算等方式,转换为简短的网址字符串。
写此短链服务是为了从整体上演示go-zero构建完整微服务的过程,算法和实现细节尽可能简化了,所以这不是一个高阶的短链服务。

shorturlshorturl目录下执行go mod init shorturl初始化go.modshorturl.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
go run api/shorturl.go -f api/etc/shorturl-api.yaml
curl -i "http://localhost:8888/shorten?url=a"
返回如下:
HTTP/1.1 200 OK
Content-Type: application/json
Date: Thu, 27 Aug 2020 14:31:39 GMT
Content-Length: 15
{"shortUrl":""}
internal/svc/servicecontext.go来传递服务依赖(如果需要)internal/logic下的对应文件mkdir -p rpc/modelshorturl.sql,如下: CREATE TABLE `shorturl`
(
`id` bigint(10) NOT NULL AUTO_INCREMENT,
`key` varchar(255) NOT NULL DEFAULT '' COMMENT 'shorten key',
`url` varchar(255) DEFAULT '' COMMENT 'original url',
PRIMARY KEY(`id`),
UNIQUE KEY `key_index`(`key`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;
rpc/model目录下执行如下命令生成CRUD+cache代码,-c表示使用redis cache goctl model mysql ddl -c -src shorturl.sql -dir .
生成后的文件结构如下:
.
├── api
│ ├── etc
│ │ └── shorturl-api.yaml
│ ├── internal
│ │ ├── config
│ │ │ └── config.go
│ │ ├── handler
│ │ │ ├── expandhandler.go
│ │ │ ├── routes.go
│ │ │ └── shortenhandler.go
│ │ ├── logic
│ │ │ ├── expandlogic.go
│ │ │ └── shortenlogic.go
│ │ ├── svc
│ │ │ └── servicecontext.go
│ │ └── types
│ │ └── types.go
│ └── shorturl.go
├── go.mod
├── go.sum
├── rpc
│ └── model
│ ├── shorturl.sql
│ ├── shorturlmodel.go // CRUD+cache代码
│ └── vars.go // 定义常量和变量
├── shorturl.api
└── shorturl.sql