聊聊dubbo-go的TokenFilter

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

TokenFilter

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

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

// GetTokenFilter ...
func GetTokenFilter() filter.Filter {
    return &TokenFilter{}
}
  • GetTokenFilter创建了TokenFilter

Invoke

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

// 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

// 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

  • token_filter

原文 

https://segmentfault.com/a/1190000023289327

本站部分文章源于互联网,本着传播知识、有益学习和研究的目的进行的转载,为网友免费提供。如有著作权人或出版方提出异议,本站将立即删除。如果您对文章转载有任何疑问请告之我们,以便我们及时纠正。

PS:推荐一个微信公众号: askHarries 或者qq群:474807195,里面会分享一些资深架构师录制的视频录像:有Spring,MyBatis,Netty源码分析,高并发、高性能、分布式、微服务架构的原理,JVM性能优化这些成为架构师必备的知识体系。还能领取免费的学习资源,目前受益良多

转载请注明原文出处:Harries Blog™ » 聊聊dubbo-go的TokenFilter

赞 (0)
分享到:更多 ()

评论 0

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址