关于dubbo:聊聊dubbogo的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