Dubbo解析(三)-動態代理與包裝

時間  2019-11-15
標籤 dubbo 解析 動態 代理 包裝 欄目 Dubbo 简体版
原文   原文鏈接

Dubbo做爲RPC框架，首先要完成的就是跨系統，跨網絡的服務調用。消費方與提供方遵循統一的接口定義，消費方調用接口時，Dubbo將其轉換成統一格式的數據結構，經過網絡傳輸，提供方根據規則找到接口實現，經過反射完成調用。也就是說，消費方獲取的是對遠程服務的一個代理(Proxy)，而提供方由於要支持不一樣的接口實現，須要一個包裝層(Wrapper)。調用的過程大概是這樣：java

消費方的Proxy和提供方的Wrapper得以讓Dubbo構建出複雜、統一的體系。而這種動態代理與包裝也是經過基於SPI的插件方式實現的，它的接口就是ProxyFactory。數組

1. ProxyFactory接口定義

@SPI("javassist")
public interface ProxyFactory {

    @Adaptive({Constants.PROXY_KEY})
    <T> T getProxy(Invoker<T> invoker) throws RpcException;

    @Adaptive({Constants.PROXY_KEY})
    <T> Invoker<T> getInvoker(T proxy, Class<T> type, URL url) throws RpcException;

}

ProxyFactory有兩個方法，分別是消費方生成代理(getProxy方法)和提供方建立Invoker對象(getInvoker方法)。緩存

它的類結構以下：網絡

image

ProxyFactory有兩種實現方式，一種是基於JDK的代理實現，一種是基於javassist的實現。ProxyFactory接口上定義了@SPI("javassist")，默認爲javassist的實現。爲何選擇了javassist，能夠看下動態代理方案性能對比。數據結構

2. getProxy方法

消費方基於調用接口獲取代理對象，使用動態代理，動態代理有多種方式，Dubbo默認支持了JDK和javassist兩種。app

AbstractProxyFactory是代理工廠的公共抽象，主要用來在getProxy方法中獲取代理接口，先從Invoker的url中獲取，若是沒有，則直接使用Invoker的接口。框架

AbstractProxyFactory.java

public <T> T getProxy(Invoker<T> invoker) throws RpcException {
    Class<?>[] interfaces = null;
    String config = invoker.getUrl().getParameter("interfaces");
    if (config != null && config.length() > 0) {
        String[] types = Constants.COMMA_SPLIT_PATTERN.split(config);
        if (types != null && types.length > 0) {
            interfaces = new Class<?>[types.length + 2];
            interfaces[0] = invoker.getInterface();
            interfaces[1] = EchoService.class;
            for (int i = 0; i < types.length; i++) {
                interfaces[i + 1] = ReflectUtils.forName(types[i]);
            }
        }
    }
    if (interfaces == null) {
        interfaces = new Class<?>[]{invoker.getInterface(), EchoService.class};
    }
    return getProxy(invoker, interfaces);
}

JdkProxyFactory中getProxy直接使用JDK的Proxy.newProxyInstance方法生成代理對象。這裏的Proxy是java.lang.reflect.Proxy。ide

public <T> T getProxy(Invoker<T> invoker, Class<?>[] interfaces) {
    return (T) Proxy.newProxyInstance(Thread.currentThread().getContextClassLoader(), interfaces, new InvokerInvocationHandler(invoker));
}

而JavassistProxyFactory的getProxy實現雖然也只有一行，但內部實現都是由Dubbo基於javassist實現的。而這裏的Proxy是com.alibaba.dubbo.common.bytecode.Proxy，是生成代理對象的工具類。工具

public <T> T getProxy(Invoker<T> invoker, Class<?>[] interfaces) {
    return (T) Proxy.getProxy(interfaces).newInstance(new InvokerInvocationHandler(invoker));
}

Dubbo的Proxy.getProxy源碼：性能

public static Proxy getProxy(ClassLoader cl, Class<?>... ics) {
	if (ics.length > 65535)
		throw new IllegalArgumentException("interface limit exceeded");

	// 遍歷代理接口，獲取接口的全限定名，並以分號分隔鏈接成字符串
	StringBuilder sb = new StringBuilder();
	for (int i = 0; i < ics.length; i++) {
		String itf = ics[i].getName();
		if (!ics[i].isInterface())
			throw new RuntimeException(itf + " is not a interface.");

		Class<?> tmp = null;
		try {
			tmp = Class.forName(itf, false, cl);
		} catch (ClassNotFoundException e) {
		}

		if (tmp != ics[i])
			throw new IllegalArgumentException(ics[i] + " is not visible from class loader");

		sb.append(itf).append(';');
	}

	// use interface class name list as key.
	String key = sb.toString();

	// get cache by class loader.
	Map<String, Object> cache;
	synchronized (ProxyCacheMap) {
		cache = ProxyCacheMap.get(cl);
		if (cache == null) {
			cache = new HashMap<String, Object>();
			ProxyCacheMap.put(cl, cache);
		}
	}

	// 查找緩存map，若是緩存存在，則獲取代理對象直接返回
	Proxy proxy = null;
	synchronized (cache) {
		do {
			Object value = cache.get(key);
			if (value instanceof Reference<?>) {
				proxy = (Proxy) ((Reference<?>) value).get();
				if (proxy != null)
					return proxy;
			}

			if (value == PendingGenerationMarker) {
				try {
					cache.wait();
				} catch (InterruptedException e) {
				}
			} else {
				cache.put(key, PendingGenerationMarker);
				break;
			}
		}
		while (true);
	}

	// AtomicLong自增生成代理類類名後綴id，防止衝突
	long id = PROXY_CLASS_COUNTER.getAndIncrement();
	String pkg = null;
	ClassGenerator ccp = null, ccm = null;
	try {
		ccp = ClassGenerator.newInstance(cl);

		Set<String> worked = new HashSet<String>();
		List<Method> methods = new ArrayList<Method>();

		for (int i = 0; i < ics.length; i++) {
			if (!Modifier.isPublic(ics[i].getModifiers())) {
				String npkg = ics[i].getPackage().getName();
				if (pkg == null) {
					pkg = npkg;
				} else {
					if (!pkg.equals(npkg))
						throw new IllegalArgumentException("non-public interfaces from different packages");
				}
			}
			ccp.addInterface(ics[i]);

			// 遍歷接口中的方法，獲取返回類型和參數類型，構建方法體
			for (Method method : ics[i].getMethods()) {
				String desc = ReflectUtils.getDesc(method);
				if (worked.contains(desc))
					continue;
				worked.add(desc);

				int ix = methods.size();
				Class<?> rt = method.getReturnType();
				Class<?>[] pts = method.getParameterTypes();

				StringBuilder code = new StringBuilder("Object[] args = new Object[").append(pts.length).append("];");
				for (int j = 0; j < pts.length; j++)
					code.append(" args[").append(j).append("] = ($w)$").append(j + 1).append(";");
				code.append(" Object ret = handler.invoke(this, methods[" + ix + "], args);");
				if (!Void.TYPE.equals(rt))
					code.append(" return ").append(asArgument(rt, "ret")).append(";");

				methods.add(method);
				ccp.addMethod(method.getName(), method.getModifiers(), rt, pts, method.getExceptionTypes(), code.toString());
			}
		}

		if (pkg == null)
			pkg = PACKAGE_NAME;

		// create ProxyInstance class.
		// 生成服務接口的代理對象字節碼
		String pcn = pkg + ".proxy" + id;
		ccp.setClassName(pcn);
		ccp.addField("public static java.lang.reflect.Method[] methods;");
		ccp.addField("private " + InvocationHandler.class.getName() + " handler;");
		ccp.addConstructor(Modifier.PUBLIC, new Class<?>[]{InvocationHandler.class}, new Class<?>[0], "handler=$1;");
		ccp.addDefaultConstructor();
		Class<?> clazz = ccp.toClass();
		clazz.getField("methods").set(null, methods.toArray(new Method[0]));

		// create Proxy class.
		// 生成Proxy的實現類
		String fcn = Proxy.class.getName() + id;
		ccm = ClassGenerator.newInstance(cl);
		ccm.setClassName(fcn);
		ccm.addDefaultConstructor();
		ccm.setSuperClass(Proxy.class);
		ccm.addMethod("public Object newInstance(" + InvocationHandler.class.getName() + " h){ return new " + pcn + "($1); }");
		Class<?> pc = ccm.toClass();
		proxy = (Proxy) pc.newInstance();
	} catch (RuntimeException e) {
		throw e;
	} catch (Exception e) {
		throw new RuntimeException(e.getMessage(), e);
	} finally {
		// release ClassGenerator
		if (ccp != null)
			ccp.release();
		if (ccm != null)
			ccm.release();
		synchronized (cache) {
			if (proxy == null)
				cache.remove(key);
			else
				// 加入緩存
				cache.put(key, new WeakReference<Proxy>(proxy));
			cache.notifyAll();
		}
	}
	return proxy;
}

getProxy生成代理對象步驟以下：

a. 遍歷代理接口，獲取接口的全限定名，並以分號分隔鏈接成字符串，以此字符串爲key，查找緩存map，若是緩存存在，則獲取代理對象直接返回。

b. 由一個AtomicLong自增生成代理類類名後綴id，防止衝突

c. 遍歷接口中的方法，獲取返回類型和參數類型，構建主要內容以下的方法體

return  ret= handler.invoke(this, methods[ix], args);

以一個DemoService爲例，它只有一個方法sayHello(String name)，生成的方法體以下：

public String sayHello(String arg0) {
    Object[] args = new Object[1];
    args[0] = arg0;

    Object ret = handler.invoke(this, methods[0], args);

    return (String) ret;
}

另外使用worked的Set集合對方法去重

d. 建立工具類ClassGenerator實例，添加靜態字段Method[] methods，添加實例對象InvokerInvocationHandler hanler，添加參數爲InvokerInvocationHandler的構造器，添加無參構造器，而後使用toClass方法生成對應的字節碼。仍是DemoService爲例，生成的字節碼對象以下：

import com.alibaba.dubbo.demo.DemoService;
import com.alibaba.dubbo.rpc.service.EchoService;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;

public class proxy0
  implements ClassGenerator.DC, EchoService, DemoService
{
  public static Method[] methods;
  private InvocationHandler handler;

  public String sayHello(String paramString)
  {
    Object[] arrayOfObject = new Object[1];
    arrayOfObject[0] = paramString;
    Object localObject = this.handler.invoke(this, methods[0], arrayOfObject);
    return ((String)localObject);
  }

  public Object $echo(Object paramObject)
  {
    Object[] arrayOfObject = new Object[1];
    arrayOfObject[0] = paramObject;
    Object localObject = this.handler.invoke(this, methods[1], arrayOfObject);
    return ((Object)localObject);
  }

  public proxy0()
  {
  }

  public proxy0(InvocationHandler paramInvocationHandler)
  {
    this.handler = paramInvocationHandler;
  }
}

e. d中生成的字節碼對象爲服務接口的代理對象，而Proxy類自己是抽象類，須要實現newInstance(InvocationHandler handler)方法，生成Proxy的實現類，其中proxy0即上面生成的服務接口的代理對象。

package com.alibaba.dubbo.common.bytecode;

public class Proxy0 implements Proxy {

    public void Proxy0() {
    }

    public Object newInstance(InvocationHandler h){
		return new proxy0(h);
	}
}

再回頭看看JavassistProxyFactory的getPrxoy方法， InvokerInvocationHandler封裝了Invoker對象，經過Proxy的newInstance方法，最終返回經過javassist生成的代理對象。

public <T> T getProxy(Invoker<T> invoker, Class<?>[] interfaces) {
    return (T) Proxy.getProxy(interfaces).newInstance(new InvokerInvocationHandler(invoker));
}

3. getInvoker方法

Invoker，Dubbo的核心模型，表明一個可執行體。在我理解，它就是一個指向最終服務實現的執行路徑，在Dubbo的實現中，提供方的服務實現首先被包裝成一個ProxyInvoker(代理執行器)，而後這個ProxyInvoker被Filter封裝成鏈式結構，被Listener包裝，被Cluster封裝，而消費方的Invoker也是一個執行遠程調用的執行器，區別在於執行路徑的不一樣。

ProxyFactory的getInvoker方法，所作的就是第一步，將服務實現封裝成代理執行器，有JdkProxyFactory和JavassistProxyFactory兩種實現，兩種都是建立匿名內部類AbstractProxyInvoker，實現doInvoke方法。

JDK的實現比較簡單，匹配到調用的方法，使用反射執行。

public <T> Invoker<T> getInvoker(T proxy, Class<T> type, URL url) {
    return new AbstractProxyInvoker<T>(proxy, type, url) {
        @Override
        protected Object doInvoke(T proxy, String methodName,
                                  Class<?>[] parameterTypes,
                                  Object[] arguments) throws Throwable {
            Method method = proxy.getClass().getMethod(methodName, parameterTypes);
            return method.invoke(proxy, arguments);
        }
    };
}

而JavassistProxyFactory則是經過javassist生成字節碼對象Wrapper類進行調用。

public <T> Invoker<T> getInvoker(T proxy, Class<T> type, URL url) {
    // TODO Wrapper cannot handle this scenario correctly: the classname contains '$'
    final Wrapper wrapper = Wrapper.getWrapper(proxy.getClass().getName().indexOf('$') < 0 ? proxy.getClass() : type);
    return new AbstractProxyInvoker<T>(proxy, type, url) {
        @Override
        protected Object doInvoke(T proxy, String methodName,
                                  Class<?>[] parameterTypes,
                                  Object[] arguments) throws Throwable {
            return wrapper.invokeMethod(proxy, methodName, parameterTypes, arguments);
        }
    };
}

Wrapper類自己是抽象類，是對java類的一種包裝，將類中的field和method抽象出propertyName和methodName，真正調用時根據傳入的方法名和參數進行匹配。

public abstract class Wrapper {

    abstract public String[] getPropertyNames();

    abstract public Class<?> getPropertyType(String pn);

    abstract public boolean hasProperty(String name);

    abstract public Object getPropertyValue(Object instance, String pn) throws NoSuchPropertyException, IllegalArgumentException;

    abstract public void setPropertyValue(Object instance, String pn, Object pv) throws NoSuchPropertyException, IllegalArgumentException;

    abstract public String[] getMethodNames();

    abstract public String[] getDeclaredMethodNames();

    abstract public Object invokeMethod(Object instance, String mn, Class<?>[] types, Object[] args) throws NoSuchMethodException, InvocationTargetException;
}

Wrapper.getWrapper方法基於不一樣的服務實現對象生成一個Wrapper實現對象。getWrapper方法先查詢緩存是否已存在，存在則返回，不然調用makeWrapper生成Wrapper對象。

public static Wrapper getWrapper(Class<?> c) {
    while (ClassGenerator.isDynamicClass(c)) // can not wrapper on dynamic class.
        c = c.getSuperclass();

    if (c == Object.class)
        return OBJECT_WRAPPER;

    Wrapper ret = WRAPPER_MAP.get(c);
    if (ret == null) {
        ret = makeWrapper(c);
        WRAPPER_MAP.put(c, ret);
    }
    return ret;
}

makeWrapper方遍歷傳入的Class對象的全部public field和public method，構建字節碼對象字符串。

  1. public字段組建getPropertyValue和setPropertyValue方法，好比有public字段name
// getPropertyValue：
// $2表明傳入的字段名稱，w爲原始對象，$w爲name的類型
if ($2.equals("name")) { return ($w) w.name;} 

// setPropertyValue
// $2表明傳入的字段名稱，w爲原始對象，$3爲傳入的字段值
if ($2.equals("name")) { w.name = ($w) $3; return;}
  1. 處理public方法，區分setter/getter方法，轉化成屬性加入到getPropertyValue和setPropertyValue方法中，另外全部方法加入到invokeMethod方法中。
// 以sayHello爲例，轉化到invokeMethod方法中
// $2爲傳入方法名稱，$3爲傳入方法參數類型數組，$4爲傳入方法參數值數組
if ("sayHello".equals($2) && ($3.length == 1)) {
    return ($w) w.sayHello((java.lang.String) $4[0]);
}

以DemoServiceImpl爲例

public class DemoServiceImpl implements DemoService {
	private String name;
	
	public String sayHello(String name) {
        return "Hello";
    }

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}
}

生成的Wrapper類以下：

public class Wrapper0 extends Wrapper {
    public static String[] pns = new String[] { "name" };
    public static Map pts = new HashMap() {

            {
                put("name", "java.lang.String");
            }
        };

    public static String[] mns = new String[] { "sayHello" };
    public static String[] dmns = new String[] { "sayHello" };
    public static Class[] mts0 = new Class[] { String.class };

    public String[] getPropertyNames() {
        return pns;
    }

    public boolean hasProperty(String n) {
        return pts.containsKey(n);
    }

    public Class getPropertyType(String n) {
        return (Class) pts.get(n);
    }

    public String[] getMethodNames() {
        return mns;
    }

    public String[] getDeclaredMethodNames() {
        return dmns;
    }

    public void setPropertyValue(Object o, String n, Object v) {
        com.alibaba.dubbo.demo.provider.DemoServiceImpl w;

        try {
            w = ((com.alibaba.dubbo.demo.provider.DemoServiceImpl) $1);
        } catch (Throwable e) {
            throw new IllegalArgumentException(e);
        }

        if ($2.equals("name")) {
            w.setName((java.lang.String) $3);

            return;
        }

        throw new com.alibaba.dubbo.common.bytecode.NoSuchPropertyException(
            "Not found property \"" + $2 +
            "\" filed or setter method in class com.alibaba.dubbo.demo.provider.DemoServiceImpl.");
    }

    public Object getPropertyValue(Object o, String n) {
        com.alibaba.dubbo.demo.provider.DemoServiceImpl w;

        try {
            w = ((com.alibaba.dubbo.demo.provider.DemoServiceImpl) $1);
        } catch (Throwable e) {
            throw new IllegalArgumentException(e);
        }

        if ($2.equals("name")) {
            return ($w) w.getName();
        }

        throw new com.alibaba.dubbo.common.bytecode.NoSuchPropertyException(
            "Not found property \"" + $2 +
            "\" filed or setter method in class com.alibaba.dubbo.demo.provider.DemoServiceImpl.");
    }

    public Object invokeMethod(Object o, String n, Class[] p, Object[] v)
        throws java.lang.reflect.InvocationTargetException {
        com.alibaba.dubbo.demo.provider.DemoServiceImpl w;

        try {
            w = ((com.alibaba.dubbo.demo.provider.DemoServiceImpl) $1);
        } catch (Throwable e) {
            throw new IllegalArgumentException(e);
        }

        try {
            if ("getName".equals($2) && ($3.length == 0)) {
                return ($w) w.getName();
            }

            if ("setName".equals($2) && ($3.length == 1)) {
                w.setName((java.lang.String) $4[0]);

                return null;
            }

            if ("sayHello".equals($2) && ($3.length == 1)) {
                return ($w) w.sayHello((java.lang.String) $4[0]);
            }
        } catch (Throwable e) {
            throw new java.lang.reflect.InvocationTargetException(e);
        }

        throw new com.alibaba.dubbo.common.bytecode.NoSuchMethodException(
            "Not found method \"" + $2 +
            "\" in class com.alibaba.dubbo.demo.provider.DemoServiceImpl.");
    }
}

在JavassistProxyFactory的getInvoker方法建立的AbstractProxyInvoker中，doInvoke方法最終經過Wrapper的invokeMethod方法完成對服務實現原始對象的調用。

wrapper.invokeMethod(proxy, methodName, parameterTypes, arguments)

如今你能夠回頭看下文章開始前的那張調用圖，Dubbo使用ProxyFactory對消費方的調用和提供者的執行進行了代理和包裝，從而爲Dubbo的內部邏輯實現奠基了基礎。

相關文章
相關標籤/搜索
每日一句
    每一个你不满意的现在，都有一个你没有努力的曾经。
本站公眾號
   歡迎關注本站公眾號,獲取更多信息
聯繫我們 最近搜索 最新文章 沪ICP备13005482号-10 MyBatis教程 SQL 教程 MySQL教程 Java 教程 Thymeleaf 教程 Hibernate教程 Spring教程 Redis教程