shorturl.md 5.1 KB

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

0. 什么是短链服务?

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

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

1. 短链微服务架构图

架构图

2. 创建工作目录并初始化go.mod

  • 创建工作目录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=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下的对应文件

4. 编写shorten/expand rpc服务(未完)

5. 定义数据库表结构

  • shorturl下创建rpc/model目录:mkdir -p rpc/model
  • 在roc/model目录下编写创建shorturl表的sql文件shorturl.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;

6. 自动生成CRUD+cache代码

  • 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

未完待续