前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >聊聊dubbo-go的TokenFilter

聊聊dubbo-go的TokenFilter

原创
作者头像
code4it
修改2020-07-20 10:07:53
2800
修改2020-07-20 10:07:53
举报
文章被收录于专栏:码匠的流水账

本文主要研究一下dubbo-go的TokenFilter

TokenFilter

dubbo-go-v1.4.2/filter/filter_impl/token_filter.go

代码语言:javascript
复制
const (
    // TOKEN ...
    TOKEN = "token"
)
​
func init() {
    extension.SetFilter(TOKEN, GetTokenFilter)
}
​
// TokenFilter ...
type TokenFilter struct{}
  • TokenFilter的init方法设置了GetTokenFilter

GetTokenFilter

dubbo-go-v1.4.2/filter/filter_impl/token_filter.go

代码语言:javascript
复制
// GetTokenFilter ...
func GetTokenFilter() filter.Filter {
    return &TokenFilter{}
}
  • GetTokenFilter创建了TokenFilter

Invoke

dubbo-go-v1.4.2/filter/filter_impl/token_filter.go

代码语言:javascript
复制
// Invoke ...
func (tf *TokenFilter) Invoke(ctx context.Context, invoker protocol.Invoker, invocation protocol.Invocation) protocol.Result {
    invokerTkn := invoker.GetUrl().GetParam(constant.TOKEN_KEY, "")
    if len(invokerTkn) > 0 {
        attachs := invocation.Attachments()
        remoteTkn, exist := attachs[constant.TOKEN_KEY]
        if exist && strings.EqualFold(invokerTkn, remoteTkn) {
            return invoker.Invoke(ctx, invocation)
        }
        return &protocol.RPCResult{Err: perrors.Errorf("Invalid token! Forbid invoke remote service %v method %s ",
            invoker, invocation.MethodName())}
    }
​
    return invoker.Invoke(ctx, invocation)
}
  • Invoke方法读取constant.TOKEN_KEY的值invokerTkn,在该值有配置的前提下,从invocation.Attachments()获取对应的remoteTkn,之后对比invokerTkn与remoteTkn,相等则执行invoker.Invoke(ctx, invocation),否则返回错误结果

OnResponse

dubbo-go-v1.4.2/filter/filter_impl/token_filter.go

代码语言:javascript
复制
// OnResponse ...
func (tf *TokenFilter) OnResponse(ctx context.Context, result protocol.Result, invoker protocol.Invoker, invocation protocol.Invocation) protocol.Result {
    return result
}
  • OnResponse方法这里直接返回result

小结

TokenFilter的Invoke方法读取constant.TOKEN_KEY的值invokerTkn,在该值有配置的前提下,从invocation.Attachments()获取对应的remoteTkn,之后对比invokerTkn与remoteTkn,相等则执行invoker.Invoke(ctx, invocation),否则返回错误结果

doc

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • TokenFilter
  • GetTokenFilter
  • Invoke
  • OnResponse
  • 小结
  • doc
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档