Skip to content

Commit

Permalink
Support protobuf serialization in the env of multiple class loader. (#…
Browse files Browse the repository at this point in the history
…49)

* Add ClassUtils.isAssignableFrom() which can cross multiple classloader.
* Support protobuf serialization in the env of multiple class loader.
  • Loading branch information
ujjboy committed Apr 28, 2018
1 parent 58c85a5 commit f0ad19f
Show file tree
Hide file tree
Showing 6 changed files with 344 additions and 60 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -194,10 +194,10 @@ public static <T> T newInstance(Class<T> clazz) throws SofaRpcRuntimeException {
/**
* 实例化一个对象(根据参数自动检测构造方法)
*
* @param clazz 对象类
* @param argTypes 构造函数需要的参数
* @param args 构造函数需要的参数
* @param <T> 对象具体类
* @param clazz 对象类
* @param argTypes 构造函数需要的参数
* @param args 构造函数需要的参数
* @param <T> 对象具体类
* @return 对象实例
* @throws SofaRpcRuntimeException 没有找到方法,或者无法处理,或者初始化方法异常等
*/
Expand Down Expand Up @@ -283,4 +283,51 @@ public static Object getDefaultArg(Class clazz) {
public static String getMethodKey(String interfaceName, String methodName) {
return interfaceName + "#" + methodName;
}

/**
* The isAssignableFrom method which can cross multiple classloader.
*
* @param interfaceClass 接口类
* @param implementClass 实现类
* @return 是否指定类型的实现类
* @see Class#isAssignableFrom(Class)
*/
public static boolean isAssignableFrom(Class<?> interfaceClass, Class<?> implementClass) {
if (interfaceClass.isAssignableFrom(implementClass)) {
return true;
}
// 跨ClassLoader的情况
String interfaceName = interfaceClass.getCanonicalName();
return implementClass.getCanonicalName().equals(interfaceName)
|| isImplementOrSubclass(interfaceName, implementClass);
}

private static boolean isImplementOrSubclass(String interfaceName, Class<?> implementClass) {
// First, get all direct interface
Class<?>[] interfaces = implementClass.getInterfaces();
if (interfaces.length > 0) {
for (Class<?> oneInterface : interfaces) {
if (interfaceName.equals(oneInterface.getCanonicalName())) {
return true;
}
if (isImplementOrSubclass(interfaceName, oneInterface)) {
return true;
}
}
}
while (!Object.class.equals(implementClass)) {
// Add the super class
Class<?> superClass = implementClass.getSuperclass();
// Interfaces does not have java.lang.Object as superclass, they have null, so break the cycle and return
if (superClass == null) {
break;
}
// Now inspect the superclass
implementClass = superClass;
if (isImplementOrSubclass(interfaceName, implementClass)) {
return true;
}
}
return false;
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,9 @@

import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.net.MalformedURLException;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.List;

/**
Expand Down Expand Up @@ -219,4 +222,48 @@ private TestMemberClass6(String s, int i) {
}
}

@Test
public void testIsAssignableFrom() throws MalformedURLException, ClassNotFoundException {
// single class loader
testIsAssignableFrom0();

// over different class loader
String codebase = ReflectUtils.getCodeBase(ClassUtilsTest.class);
String url = "file://" + codebase;
loader = new URLClassLoader(new URL[] { new URL(url) }, null);
try {
testIsAssignableFrom0();
} finally {
loader = null;
}
}

private void testIsAssignableFrom0() throws ClassNotFoundException {
Assert.assertTrue(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignOk.class)));
Assert.assertTrue(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignOk0.class)));
Assert.assertTrue(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignOk1.class)));
Assert.assertTrue(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignOk2.class)));
Assert.assertTrue(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignOk3.class)));
Assert.assertTrue(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignOk4.class)));
Assert.assertTrue(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignOk5.class)));
Assert.assertTrue(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignOk6.class)));
Assert.assertTrue(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignOk7.class)));
Assert.assertTrue(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignOk8.class)));
Assert.assertTrue(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignOk9.class)));

Assert.assertFalse(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignNotOk.class)));
Assert.assertFalse(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignNotOk0.class)));
Assert.assertFalse(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignNotOk1.class)));
Assert.assertFalse(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignNotOk2.class)));
Assert.assertFalse(ClassUtils.isAssignableFrom(TestAssignOk.class, getClass(TestAssignNotOk3.class)));
}

private static ClassLoader loader;

private static Class getClass(Class clazz) throws ClassNotFoundException {
if (loader != null) {
return loader.loadClass(clazz.getCanonicalName());
}
return clazz;
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.alipay.sofa.rpc.common.utils;

/**
* @author <a href="mailto:zhanggeng.zg@antfin.com">GengZhang</a>
*/
interface TestAssignOk {
}

abstract class TestAssignOk0 implements TestAssignOk {
}

interface TestAssignOk1 extends TestAssignOk {
}

class TestAssignOk2 implements TestAssignOk {
}

class TestAssignOk3 implements TestAssignOk1 {
}

class TestAssignOk4 extends TestAssignOk0 {
}

interface TestAssignOk5 extends TestAssignOk1 {
}

interface TestAssignNotOk {
}

abstract class TestAssignNotOk0 implements TestAssignNotOk {
}

class TestAssignNotOk1 extends TestAssignNotOk0 {
}

class TestAssignNotOk2 implements TestAssignNotOk {
}

interface TestAssignNotOk3 extends TestAssignNotOk {
}

class TestAssignOk6 implements TestAssignOk, TestAssignNotOk {
}

class TestAssignOk7 implements TestAssignOk1, TestAssignNotOk3 {
}

class TestAssignOk8 extends TestAssignNotOk1 implements TestAssignOk5, TestAssignNotOk3 {
}

class TestAssignOk9 extends TestAssignOk6 {
}
Loading

0 comments on commit f0ad19f

Please sign in to comment.