dubbo 令牌验证 源码分析

在ServiceConfig.java中在doExportUrlsFor1Protocol中：

 

        if (! ConfigUtils.isEmpty(token)) {//如果配置令牌功能
            if (ConfigUtils.isDefault(token)) {
                map.put("token", UUID.randomUUID().toString());
            } else {
                map.put("token", token);
            }
        }

如果服务提供者只是设置token=true，那使用系统使用UUID生成token令牌，否则直接使用token做令牌。

 

 

当消费者发起调用时，会运行到TokenFilter调用链进行处理：

 

@Activate(group = Constants.PROVIDER, value = Constants.TOKEN_KEY)
public class TokenFilter implements Filter {

	public Result invoke(Invoker<?> invoker, Invocation inv) throws RpcException {
	    String token = invoker.getUrl().getParameter(Constants.TOKEN_KEY);
	    if (ConfigUtils.isNotEmpty(token)) {
	        Class<?> serviceType = invoker.getInterface();
	        Map<String, String> attachments = inv.getAttachments();
    		String remoteToken = attachments == null ? null : attachments.get(Constants.TOKEN_KEY);
    		if (! token.equals(remoteToken)) {
    			throw new RpcException("Invalid token! Forbid invoke remote service " + serviceType + " method " + inv.getMethodName() + "() from consumer " + RpcContext.getContext().getRemoteHost() + " to provider "  + RpcContext.getContext().getLocalHost());
    		}
	    }
		return invoker.invoke(inv);
	}

}

group=Constants.PROVIDER表示当rpc请求到达服务端时才会运行这个调用链，value=Constants.TOKEN_KEY表示只有开启token令牌功能时，才运行这段代码逻辑。

 

最后欢迎大家访问我的个人网站：1024s