Go+gRPC-Gateway(V2) 微服务实战,小程序登录鉴权服务(五):鉴权 gRPC-Interceptor 拦截器实战
拦截器(gRPC-Interceptor
)类似于 Gin
中间件(Middleware),让你在真正调用 RPC
服务前,进行身份认证、参数校验、限流等通用操作。
系列
- 云原生 API 网关,gRPC-Gateway V2 初探
- Go + gRPC-Gateway(V2) 构建微服务实战系列,小程序登录鉴权服务:第一篇
- Go + gRPC-Gateway(V2) 构建微服务实战系列,小程序登录鉴权服务:第二篇
- Go + gRPC-Gateway(V2) 构建微服务实战系列,小程序登录鉴权服务(三):RSA(RS512) 签名 JWT
- Go+gRPC-Gateway(V2) 微服务实战,小程序登录鉴权服务(四):自动生成 API TS 类型
grpc.UnaryInterceptor
从 VSCode
-> Go to Definition
开始,我们看到如下源码:
// UnaryInterceptor returns a ServerOption that sets the UnaryServerInterceptor for the
// server. Only one unary interceptor can be installed. The construction of multiple
// interceptors (e.g., chaining) can be implemented at the caller.
func UnaryInterceptor(i UnaryServerInterceptor) ServerOption {
return newFuncServerOption(func(o *serverOptions) {
if o.unaryInt != nil {
panic("The unary server interceptor was already set and may not be reset.")
}
o.unaryInt = i
})
}
注释很清晰:UnaryInterceptor
返回一个为 gRPC server
设置 UnaryServerInterceptor
的 ServerOption
。只能安装一个一元拦截器。多个拦截器的构造(例如,chaining
)可以在调用方实现。
这里我们需要实现具有如下定义的方法:
// UnaryServerInterceptor provides a hook to intercept the execution of a unary RPC on the server. info
// contains all the information of this RPC the interceptor can operate on. And handler is the wrapper
// of the service method implementation. It is the responsibility of the interceptor to invoke handler
// to complete the RPC.
type UnaryServerInterceptor func(ctx context.Context, req interface{}, info *UnaryServerInfo, handler UnaryHandler) (resp interface{}, err error)
注释很清晰:UnaryServerInterceptor
提供了一个钩子来拦截服务器上一元 RPC
的执行。info
包含拦截器可以操作的这个 RPC
的所有信息。handler
是 service
方法实现的包装器。拦截器的职责是调用 handler
来完成 RPC
方法的执行。在真正调用 RPC
服务前,进行各微服务的通用操作(如:authorization
)。
Auth Interceptor 编写
一句话描述业务:
- 从请求头(
header
) 中拿到authorization
字段传过来的token
,然后通过pubclic.key
验证是否合法。合法就把AccountID
(claims.subject
) 附加到当前请求上下文中(context
)。
核心拦截器代码如下:
type interceptor struct {
verifier tokenVerifier
}
func (i *interceptor) HandleReq(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) {
// 拿到 token
tkn, err := tokenFromContext(ctx)
if err != nil {
return nil, status.Error(codes.Unauthenticated, "")
}
// 验证 token
aid, err := i.verifier.Verify(tkn)
if err != nil {
return nil, status.Errorf(codes.Unauthenticated, "token not valid: %v", err)
}
// 调用真正的 RPC 方法
return handler(ContextWithAccountID(ctx, AccountID(aid)), req)
}
具体代码位于 /microsvcs/shared/auth/auth.go
Todo
微服务
一个 Todo-List
测试服务。
这里,我们加入一个新的微服务 Todo
,我们要做的是:访问 Todo
RPC Service 之前需要经过我们的鉴权 Interceptor
判断是否合法。
定义 proto
todo.proto
syntax = "proto3";
package todo.v1;
option go_package="server/todo/api/gen/v1;todopb";
message CreateTodoRequest {
string title = 1;
}
message CreateTodoResponse {
}
service TodoService {
rpc CreateTodo (CreateTodoRequest) returns (CreateTodoResponse);
}
简单起见(测试用
),这里就一个字段 title
。
定义 google.api.Service
todo.yaml
type: google.api.Service
config_version: 3
http:
rules:
- selector: todo.v1.TodoService.CreateTodo
post: /v1/todo
body: "*"
生成相关代码
microsvcs
目录下执行:
sh gen.sh
会生成如下文件:
microsvcs/todo/api/gen/v1/todo_grpc.pb.go
microsvcs/todo/api/gen/v1/todo.pb.go
microsvcs/todo/api/gen/v1/todo.pb.gw.go
client
目录下执行:
sh gen_ts.sh
会生成如下文件:
client/miniprogram/service/proto_gen/todo/todo_pb.js
client/miniprogram/service/proto_gen/todo/todo_pb.d.ts
实现 CreateTodo
Service
具体见:microsvcs/todo/todo/todo.go
type Service struct {
Logger *zap.Logger
todopb.UnimplementedTodoServiceServer
}
func (s *Service) CreateTodo(c context.Context, req *todopb.CreateTodoRequest) (*todopb.CreateTodoResponse, error) {
// 从 token 中解析出 accountId,确定身份后执行后续操作
aid, err := auth.AcountIDFromContext(c)
if err != nil {
return nil, err
}
s.Logger.Info("create trip", zap.String("title", req.Title), zap.String("account_id", aid.String()))
return nil, status.Error(codes.Unimplemented, "")
}
重构下 gRPC-Server 的启动
我们现在有多个服务了,Server
启动部分有很多重复的,重构一下:
具体代码位于:microsvcs/shared/server/grpc.go
func RunGRPCServer(c *GRPCConfig) error {
nameField := zap.String("name", c.Name)
lis, err := net.Listen("tcp", c.Addr)
if err != nil {
c.Logger.Fatal("cannot listen", nameField, zap.Error(err))
}
var opts []grpc.ServerOption
// 鉴权微服务是无需 auth 拦截器,这里做一下判断
if c.AuthPublicKeyFile != "" {
in, err := auth.Interceptor(c.AuthPublicKeyFile)
if err != nil {
c.Logger.Fatal("cannot create auth interceptor", nameField, zap.Error(err))
}
opts = append(opts, grpc.UnaryInterceptor(in))
}
s := grpc.NewServer(opts...)
c.RegisterFunc(s)
c.Logger.Info("server started", nameField, zap.String("addr", c.Addr))
return s.Serve(lis)
}
接下,其它微服务的gRPC-Server
启动代码就好看很多了:
具体代码位于:todo/main.go
logger.Sugar().Fatal(
server.RunGRPCServer(&server.GRPCConfig{
Name: "todo",
Addr: ":8082",
AuthPublicKeyFile: "shared/auth/public.key",
Logger: logger,
RegisterFunc: func(s *grpc.Server) {
todopb.RegisterTodoServiceServer(s, &todo.Service{
Logger: logger,
})
},
}),
)
具体代码位于:auth/main.go
logger.Sugar().Fatal(
server.RunGRPCServer(&server.GRPCConfig{
Name: "auth",
Addr: ":8081",
Logger: logger,
RegisterFunc: func(s *grpc.Server) {
authpb.RegisterAuthServiceServer(s, &auth.Service{
OpenIDResolver: &wechat.Service{
AppID: "your-appid",
AppSecret: "your-appsecret",
},
Mongo: dao.NewMongo(mongoClient.Database("grpc-gateway-auth")),
Logger: logger,
TokenExpire: 2 * time.Hour,
TokenGenerator: token.NewJWTTokenGen("server/auth", privKey),
})
},
}),
)
联调
重构下 gateway server
我们要反向代理到多个 gRPC server
端点了,整理下代码,弄成配置的形式:
具体代码位于:microsvcs/gateway/main.go
serverConfig := []struct {
name string
addr string
registerFunc func(ctx context.Context, mux *runtime.ServeMux, endpoint string, opts []grpc.DialOption) (err error)
}{
{
name: "auth",
addr: "localhost:8081",
registerFunc: authpb.RegisterAuthServiceHandlerFromEndpoint,
},
{
name: "todo",
addr: "localhost:8082",
registerFunc: todopb.RegisterTodoServiceHandlerFromEndpoint,
},
}
for _, s := range serverConfig {
err := s.registerFunc(
c, mux, s.addr,
[]grpc.DialOption{grpc.WithInsecure()},
)
if err != nil {
logger.Sugar().Fatalf("cannot register service %s : %v", s.name, err)
}
}
addr := ":8080"
logger.Sugar().Infof("grpc gateway started at %s", addr)
logger.Sugar().Fatal(http.ListenAndServe(addr, mux))
测试
Refs
- grpc-ecosystem/go-grpc-middleware
- API Security : API key is dead..Long live Distributed Token by value
- Demo: go-grpc-gateway-v2-microservice
- gRPC-Gateway
- gRPC-Gateway Docs
我是为少
微信:uuhells123
公众号:黑客下午茶
加我微信(互相学习交流),关注公众号(获取更多学习资料~)
Go+gRPC-Gateway(V2) 微服务实战,小程序登录鉴权服务(五):鉴权 gRPC-Interceptor 拦截器实战的更多相关文章
- Go+gRPC-Gateway(V2) 微服务实战,小程序登录鉴权服务(六):客户端基础库 TS 实战
小程序登录鉴权服务,客户端底层 SDK,登录鉴权.业务请求.鉴权重试模块 Typescript 实战. 系列 云原生 API 网关,gRPC-Gateway V2 初探 Go + gRPC-Gatew ...
- Go+gRPC-Gateway(V2) 微服务实战,小程序登录鉴权服务(四):客户端强类型约束,自动生成 API TS 类型定义
系列 云原生 API 网关,gRPC-Gateway V2 初探 Go + gRPC-Gateway(V2) 构建微服务实战系列,小程序登录鉴权服务:第一篇 Go + gRPC-Gateway(V2) ...
- JeeWx捷微3.1小程序版本发布,支持微信公众号,微信企业号,支付窗——JAVA版开源微信管家
支持小程序,JeeWx捷微3.1小程序版本发布^_^ JeeWx捷微V3.1——多触点小程序版本管理平台(支持微信公众号,微信企业号,支付窗) JeeWx捷微V3.1.0版本紧跟微信小程序更新,在 ...
- JAVA版开源微信管家—JeeWx捷微3.1小程序版本发布,支持微信公众号,微信企业号,支付窗
支持小程序,JeeWx捷微3.1小程序版本发布^_^ JeeWx捷微V3.1--多触点小程序版本管理平台(支持微信公众号,微信企业号,支付窗) JeeWx捷微V3.1.0版本紧跟微信小程序更新,在原有 ...
- 微慕WordPress小程序增强版
2017年1月9日,张小龙在2017微信公开课Pro上发布的微信小程序正式上线.在过去的2年多的时间里,微信小程序领头,各大互联网平台也不甘落后,陆续推出自己的小程序.2018年7月4日,百度智能小程 ...
- 小程序server-3-搭建WebSocket 服务
小程序server-3-搭建WebSocket 服务: 1.安装 Node 模块 使用 ws 模块来在服务器上支持 WebSocket 协议,下面使用 NPM 来安装: cd /var/www/wxp ...
- 微信小程序个人/企业开放服务类目一览表
微信小程序个人/企业开放服务类目一览表 微信小程序个人开放服务类目表 服务类目 类目分类一 类目分类二 引导描述 出行与交通 代驾 / / 生活服务 家政.丽人.摄影/扩印.婚庆服务.环保回收/废 ...
- 微信-小程序-开发文档-服务端-模板消息:templateMessage.send
ylbtech-微信-小程序-开发文档-服务端-模板消息:templateMessage.send 1.返回顶部 1. templateMessage.send 本接口应在服务器端调用,详细说明参见服 ...
- 微信-小程序-开发文档-服务端-模板消息:templateMessage.getTemplateList
ylbtech-微信-小程序-开发文档-服务端-模板消息:templateMessage.getTemplateList 1.返回顶部 1. templateMessage.getTemplateLi ...
随机推荐
- Java自学第8期——多线程
1.多线程: 操作系统支持同时运行多个任务,一个任务通常是一个程序,所有运行中的程序就是一个进程().程序内部包含多个顺序执行流,每个顺序执行流就是一个线程. 并发:两个或者多个事件在同一个时间段内交 ...
- 1095 Cars on Campus——PAT甲级真题
1095 Cars on Campus Zhejiang University has 6 campuses and a lot of gates. From each gate we can col ...
- ffmpeg:为视频添加静态水印
在ffmpeg中,添加水印需要用overlay滤镜,这是一个复杂滤镜,因为它需要两个输入,默认第一个输入是主画面,第二输入为水印,先执行一个简单的看看. 下面有两个文件,一个是可爱的大雄兔,一个是可爱 ...
- 代码生成器 springboot架构
import com.baomidou.mybatisplus.annotation.DbType;import com.baomidou.mybatisplus.annotation.IdType; ...
- spring中的依赖注入(DI)笔记
使用xml bean依赖注入有set注入和构造器注入 set注入用的比较多 <bean id="a" class="com.A"> <prop ...
- JS数组的常用属性或方法
1.length 数组长度 计算数组的长度 var arr=[1,2,3,4,5]; console.log(arr.length);//输出结果是5 2. push() 添加元素 向数组尾部添加新元 ...
- es6 快速入门 —— 函数
其他章节请看: es6 快速入门 系列 函数 函数是所有编程语言的重要组成部分,es6之前函数语法一直没什么变化,遗留了许多问题,javaScript开发者多年来不断抱怨,es6终于决定大力度更新函数 ...
- 【HTB系列】靶机Netmon的渗透测试
出品|MS08067实验室(www.ms08067.com) 本文作者:是大方子(Ms08067实验室核心成员) 总结和反思: win中执行powershell的远程代码下载执行注意双引号转义 对po ...
- HDOJ-3001(TSP+三进制状态压缩)
Traving HDOJ-3001 这题考察的是状态压缩dp和tsp问题的改编 需要和传统tsp问题区分的事,这题每个点最多可以经过两次故状态有3种:0,1,2 这里可以模仿tsp问题的二进制压缩方法 ...
- pytorch(00)
pytorch入门到项目(-) 一.pytorch的环境 本身项目采用win10系统+pycharm+anaconda+cuda. 其中版本为 python 3.7 anaconda 5.3.1 cu ...