关于go-zero单体服务使用泛型简化注册Handler路由的问题
作者：修车课代表 发布时间：2023-08-30 20:19:08
标签：go,zero,单体服务,泛型
一、Golang环境安装及配置Go Module
https://go-zero.dev/cn/docs/prepare/golang-install
mac OS安装Go#
下载并安装Go for Mac
验证安装结果
$ go version
go version go1.15.1 darwin/amd64
linux 安装Go#
下载Go for Linux
解压压缩包至/usr/local
$ tar -C /usr/local -xzf go1.15.8.linux-amd64.tar.gz
添加/usr/local/go/bin到环境变量
$ $HOME/.profile
$ export PATH=$PATH:/usr/local/go/bin
$ source $HOME/.profile
验证安装结果
$ go version
go version go1.15.1 linux/amd64
Windows安装Go#
下载并安装Go for Windows验证安装结果
$ go version
go version go1.15.1 windows/amd64
MODULE配置
Go Module是Golang管理依赖性的方式，像Java中的Maven，Android中的Gradle类似。
查看GO111MODULE开启情况
$ go env GO111MODULE
on
开启GO111MODULE，如果已开启（即执行go env GO111MODULE结果为on）请跳过。
$ go env -w GO111MODULE="on"
设置GOPROXY
$ go env -w GOPROXY=https://goproxy.cn
设置GOMODCACHE
查看GOMODCACHE
$ go env GOMODCACHE
如果目录不为空或者/dev/null，请跳过。
go env -w GOMODCACHE=$GOPATH/pkg/mod
二、Goctl 安装
Goctl在go-zero项目开发着有着很大的作用，其可以有效的帮助开发者大大提高开发效率，减少代码的出错率，缩短业务开发的工作量，更多的Goctl的介绍请阅读Goctl介绍
安装(mac&linux)
### Go 1.15 及之前版本
GO111MODULE=on GOPROXY=https://goproxy.cn/,direct go get -u github.com/zeromicro/go-zero/tools/goctl@latest
### Go 1.16 及以后版本
GOPROXY=https://goproxy.cn/,direct go install github.com/zeromicro/go-zero/tools/goctl@latest
安装(windows)
go install github.com/zeromicro/go-zero/tools/goctl@latest
环境变量检测(mac&linux)
go get 下载编译后的二进制文件位于 \$GOPATH/bin 目录下，要确保 $GOPATH/bin已经添加到环境变量。
sudo vim /etc/paths //添加环境变量
在最后一行添加如下内容 //$GOPATH 为你本机上的文件地址
$GOPATH/bin
安装结果验证
$ goctl -v
goctl version 1.1.4 darwin/amd64
二、初始化go-zero
快速生成 api 服务
goctl api new greet
cd greet
go mod init
go mod tidy
go run greet.go -f etc/greet-api.yaml
默认侦听在 8888 端口
侦听端口可以在greet-api.yaml配置文件里修改，此时，可以通过 curl 请求，或者直接在浏览器中打开http://localhost:8888/from/you
$ curl -i http://localhost:8888/from/you
HTTP/1.1 200 OK
Content-Type: application/json; charset=utf-8
Traceparent: 00-45fa9e7a7c505bad3a53a024e425ace9-eb5787234cf3e308-00
Date: Thu, 22 Oct 2020 14:03:18 GMT
Content-Length: 14
null
greet服务的目录结构
$ tree greet
greet
├── etc
│ └── greet-api.yaml
├── greet.api
├── greet.go
└── internal
├── config
│ └── config.go
├── handler
│ ├── greethandler.go
│ └── routes.go
├── logic
│ └── greetlogic.go
├── svc
│ └── servicecontext.go
└── types
└── types.go
三、查看注册Handler路由流程greet.go
var configFile = flag.String("f", "etc/greet-api.yaml", "the config file")
func main() {
flag.Parse()
var c config.Config
conf.MustLoad(*configFile, &c)
server := rest.MustNewServer(c.RestConf)
defer server.Stop()
//上面的都是加载配置什么的
ctx := svc.NewServiceContext(c)
handler.RegisterHandlers(server, ctx) //此方法是注册路由和路由映射Handler，重点在这里
fmt.Printf("Starting server at ％s:％d...\n", c.Host, c.Port)
server.Start()
}
RegisterHandlers在internal\handler\routes.go中
func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
server.AddRoutes( //往rest.Server中添加路由
[]rest.Route{ //路由数组
{
Method: http.MethodGet,
Path: "/from/:name", //路由
Handler: GreetHandler(serverCtx),//当前路由的处理Handler
},
},
)
}
GreetHandler在internal\handler\greethandler.go中
func GreetHandler(ctx *svc.ServiceContext) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
var req types.Request
if err := httpx.Parse(r, &req); err != nil { //请求的错误判断，这个可以不用管
httpx.Error(w, err)
return
}
l := logic.NewGreetLogic(r.Context(), ctx) //GreetHandler处理函数将请求转发到了GreetLogic中，调用NewGreetLogic进行结构体的初始化
resp, err := l.Greet(req) //然后调用Greet来进行处理请求，所以我们在GreetLogic.Greet方法中可以看到一句话// todo: add your logic here and delete this line
if err != nil {
httpx.Error(w, err)
} else {
httpx.OkJson(w, resp)
}
}
}
四、对注册Handler路由进行简化
项目文件的增加
在路由注册时，我们如果服务越加越多，那么相对应的func xxxxHandler(ctx *svc.ServiceContext) http.HandlerFunc就要进行多次的添加，并且这个方法体内部1到5行是属于额外的重复添加
例如：我们添加一个customlogic.go
按照命名的正确和规范性，需要在internal\logic目录下添加customlogic.go文件，然后在internal\handler目录下添加customhandler.go文件，并且两个文件都添加相对应的结构体和函数等，最后在routes.go中再添加一次
{
Method: http.MethodGet,
Path: "/custom/:name",
Handler: CustomHandler(serverCtx),
},
此时，我们的文件结构应该是这样
greet
├── etc
│ └── greet-api.yaml
├── greet.api
├── greet.go
└── internal
├── config
│ └── config.go
├── handler
│ ├── greethandler.go
│ ├── customhandler.go
│ ├── ...
│ └── routes.go
├── logic
│ ├── greetlogic.go
│ ├── ...
│ └── customlogic.go
├── svc
│ └── servicecontext.go
└── types
└── types.go
当单体应用达到一定的数量级，handler和logic文件夹下将会同步增加很多的文件
引入泛型概念
自Go1.18开始，go开始使用泛型，泛型的广泛定义 ：是一种把明确类型的工作推迟到创建对象或者调用方法的时候才去明确的特殊的类型。 也就是说在泛型使用过程中，操作的数据类型被指定为一个参数，而这种参数类型可以用在 类、方法和接口 中，分别被称为 泛型类 、 泛型方法 、 泛型接口 。
我们可以利用泛型，让在添加路由时就要固定死的Handler: GreetHandler(serverCtx)推迟到后面，去根据实际的Logic结构体去判断需要真正执行的logic.NewGreetLogic(r.Context(), ctx)初始化结构体和l.Greet(req)逻辑处理方法
如何去做在internal\logic下添加一个baselogic.go文件,参考Go泛型实战 | 如何在结构体中使用泛型
package logic
import (
"greet/internal/svc"
"greet/internal/types"
"net/http"
)
type BaseLogic interface {
any
Handler(req types.Request, w http.ResponseWriter, r *http.Request, svcCtx *svc.ServiceContext) //每一个结构体中必须要继承一下Handler方法，例如customlogic.go和greetlogic.go中的Handler方法
}
type logic[T BaseLogic] struct {
data T
}
func New[T BaseLogic]() logic[T] {
c := logic[T]{}
var ins T
c.data = ins
return c
}
func (a *logic[T]) LogicHandler(req types.Request, w http.ResponseWriter, r *http.Request, svcCtx *svc.ServiceContext) { //作为一个中转处理方法，最终执行结构体的Handler
a.data.Handler(req, w, r, svcCtx)
}
将greethandler.go文件修改成basehandler.go,注释掉之前的GreetHandler方法
package handler
import (
"net/http"
"greet/internal/logic"
"greet/internal/svc"
"greet/internal/types"
"github.com/zeromicro/go-zero/rest/httpx"
)
// func GreetHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
// return BaseHandlerFunc(svcCtx)
// // return func(w http.ResponseWriter, r *http.Request) {
// // var req types.Request
// // if err := httpx.Parse(r, &req); err != nil {
// // httpx.Error(w, err)
// // return
// // }
// // l := logic.NewGreetLogic(r.Context(), svcCtx)
// // resp, err := l.Greet(&req)
// // if err != nil {
// // httpx.Error(w, err)
// // } else {
// // httpx.OkJson(w, resp)
// // }
// // }
// }
func BaseHandlerFunc[T logic.BaseLogic](svcCtx *svc.ServiceContext, t T) http.HandlerFunc {
return func(w http.ResponseWriter, r *http.Request) {
var req types.Request
if err := httpx.Parse(r, &req); err != nil {
httpx.Error(w, err)
return
}
//通过泛型动态调用不同结构体的Handler方法
cc := logic.New[T]()
cc.LogicHandler(req, w, r, svcCtx)
}
}
在internal\logic\greetlogic.go中增加一个Handler方法
package logic
import (
"context"
"net/http"
"greet/internal/svc"
"greet/internal/types"
"github.com/zeromicro/go-zero/core/logx"
"github.com/zeromicro/go-zero/rest/httpx"
)
type GreetLogic struct {
logx.Logger
ctx context.Context
svcCtx *svc.ServiceContext
}
func NewGreetLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GreetLogic {
return &GreetLogic{
Logger: logx.WithContext(ctx),
ctx: ctx,
svcCtx: svcCtx,
}
}
func (a GreetLogic) Handler(req types.Request, w http.ResponseWriter, r *http.Request, svcCtx *svc.ServiceContext) { //新增方法
l := NewGreetLogic(r.Context(), svcCtx)
resp, err := l.Greet(&req)
if err != nil {
httpx.Error(w, err)
} else {
httpx.OkJson(w, resp)
}
}
func (l *GreetLogic) Greet(req *types.Request) (resp *types.Response, err error) {
// todo: add your logic here and delete this line
response := new(types.Response)
if (*req).Name == "me" {
response.Message = "greetLogic: listen to me, thank you."
} else {
response.Message = "greetLogic: listen to you, thank me."
}
return response, nil
}
然后修改internal\handler\routes.go下面的server.AddRoutes部分
func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
server.AddRoutes( //往rest.Server中添加路由
[]rest.Route{ //路由数组
{
Method: http.MethodGet,
Path: "/from/:name", //路由
Handler: BaseHandlerFunc(serverCtx,logic.GreetLogic{}),
},
},
)
}
现在就大功告成了，我们启动一下
go run greet.go -f etc/greet-api.yaml
然后在浏览器中请求一下http://localhost:8888/from/you
验证一下新增api路由在internal\logic下新增一个customlogic.go文件
package logic
import (
"context"
"net/http"
"greet/internal/svc"
"greet/internal/types"
"github.com/zeromicro/go-zero/core/logx"
"github.com/zeromicro/go-zero/rest/httpx"
)
type CustomLogic struct {
logx.Logger
ctx context.Context
svcCtx *svc.ServiceContext
}
func NewCustomLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CustomLogic {
return &CustomLogic{
Logger: logx.WithContext(ctx),
ctx: ctx,
svcCtx: svcCtx,
}
}
func (a CustomLogic) Handler(req types.Request, w http.ResponseWriter, r *http.Request, svcCtx *svc.ServiceContext) {
l := NewCustomLogic(r.Context(), svcCtx)
resp, err := l.Custom(&req)
if err != nil {
httpx.Error(w, err)
} else {
httpx.OkJson(w, resp)
}
}
func (l *CustomLogic) Custom(req *types.Request) (resp *types.Response, err error) { //response.Message稍微修改了一下，便于区分
// todo: add your logic here and delete this line
response := new(types.Response)
if (*req).Name == "me" {
response.Message = "customLogic: listen to me, thank you."
} else {
response.Message = "customLogic: listen to you, thank me."
}
return response, nil
}
然后修改internal\handler\routes.go
func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) {
server.AddRoutes( //往rest.Server中添加路由
[]rest.Route{ //路由数组
{
Method: http.MethodGet,
Path: "/from/:name", //路由
Handler: BaseHandlerFunc(serverCtx,logic.GreetLogic{}),
},
{
Method: http.MethodGet,
Path: "/to/:name", //路由
Handler: BaseHandlerFunc(serverCtx,logic.CustomLogic{}),
},
},
)
}
其他地方不需要修改
我们启动一下
go run greet.go -f etc/greet-api.yaml
然后在浏览器中请求一下http://localhost:8888/from/you、http://localhost:8888/to/you、http://localhost:8888/too/you
现在，在添加新的logic做路由映射时，就可以直接简化掉添加xxxxhandler.go文件了，实际上是将这个Handler移动到了xxxxlogic.go中。
本文代码放在go-zero-monolithic-service-generics
来源：https://www.cnblogs.com/spatxos/p/16524742.html
0
投稿猜你喜欢
- LCase：转成小写 UCase：转成大写 下面是ASP中的代码，可以直接演示效果的。 代码如下:<％ dim s
- 修改MySql Server安装目录下的 my.ini 文件，在mysqld节下加入下面一行set-variable=lower_case_
- 本文实例分析了php字符串截取函数用法。分享给大家供大家参考。具体分析如下：php自带的截取字符串的函数只能处理英文,数字的不能截取中文混排
- Dreamweaver一直是不少网友钟情的网页设计工具，除了它强大的动态效果制作能力外，方便简洁的操作界面更是独具特色，下面我们将以最新版本
- 为了使一个MySQL系统安全，强烈要求你考虑下列建议……当你连接一个MySQL服务器时，你通常应
- 在一个群上看到好几次问到call和apply的作用，function这两个方法的效果大家都很容易理解，但一般很难让人深刻地理解使用它们的时机
- 又有人说设session.timeout=99999。这种同样不行，session有最大时间限制。我经过测试发现最大值为24小时，也就是说你
- WinHttp; // Microsoft WinHTTP Services, version 5.1Alias HTTPREQUEST_P
Software as a service 软件即服务，21世纪开始兴起的一种完全创新的软件应用模式。客户通过互联网向厂商定购所需的应用软件- 在内容为主的网站中，搜索框往往是最常用的设计元素之一。从可用性的角度来看，搜索功能是用户有了明确的内容想看的时候最后使用的功能。如果一个网站
- 有感于中国互联网设计界十几年的变化，从网页设计师变身界面设计师，和近一两年来兴起的转型交互设计师。大多数都是随着一个行业的兴起而前赴后继的投
- 由于系统自带的MySQL默认字符集不是gbk,因此给数据库的推广应用以及中文程序的开发带来极大的不便,在没完没了的GBK和UTF8的转换过程
- 安装完 Oracle11g 之后，想打开自带的 SQL Plus 来学习，然后按照提示用 sys 用户来连接数据库，可输了好几次都提示一个错
- 请问鼠标移过去就出现二级菜单代码怎么写啊 <head><style type="tex
- numpy作为python科学计算的基础模块，支撑起了pandas、matplotlib等使用。其中，ndarray作为numpy的重要
- 程序如下：＜％Function GetEmploymentStatusListDim dd = Ap
- 占位符通过占位符，可以指定格式进行输入或输出，以下为 fmt 标准库里的占位符：普通占位符占位符描述举例结果％v默认格式的值fmt.Prin
- asp学习入门经验介绍，本文初步介绍了初学asp的一些相关知识，如VBScript语法简介，循环控制语句的使用，asp数据库的简单操作查询，
- 很早之前就在PJ的blog上看到可以用VS2005调试ASP程序，但是没有写出具体的步骤，后来一次偶尔也让我找到了方法，但是一直没把它写出来
- 废话不多说，我就直接上代码让大家看看吧！#!/usr/bin/env python# -*- coding: utf-8 -*-# @Fil
