TestApiClassLoaderManager.java
4.06 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
package com.zhaoonline.support.gateway.apijar;
import java.io.IOException;
import java.lang.ref.WeakReference;
import java.net.URLClassLoader;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.nio.file.attribute.PosixFilePermission;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import org.junit.Assert;
import org.junit.Test;
import com.zhaoonline.support.gateway.utils.Utils;
import javassist.CannotCompileException;
import javassist.NotFoundException;
public class TestApiClassLoaderManager {
public static String classParentDir = "D://test/";
String jarDir=classParentDir + "testJar.jar";
@Test
public void testAddNewLoader() throws IOException, NotFoundException, CannotCompileException, ClassLoaderExistException{
ApiClassLoaderManager manager=ApiClassLoaderManager.INSTANCE;
Utils.deleteDirectory(Paths.get(classParentDir));
Set<PosixFilePermission> perms=new HashSet<PosixFilePermission>();
perms.add(PosixFilePermission.GROUP_EXECUTE);
perms.add(PosixFilePermission.GROUP_READ);
perms.add(PosixFilePermission.GROUP_WRITE);
Files.createDirectories(Paths.get(classParentDir));
Map<String, Class<?>> properties = new HashMap<String, Class<?>>();
properties.put("name", java.lang.String.class);
properties.put("id", java.lang.String.class);
TestJarLoader.createDynamicJarWithSimplePojo(properties, "com.test.User", jarDir);
URLClassLoader newLoader=JarLoader.loadFromJar(Thread.currentThread().getContextClassLoader(), jarDir);
manager.add("com.test.User",newLoader);
int count=manager.getApiClassLoaderCount();
Assert.assertEquals(1, count);
String className="com.test.User";
Class<?> targetClass;
try {
targetClass = manager.loadClass(className);
Assert.assertNotNull(targetClass);
} catch (ClassNotFoundException e) {
e.printStackTrace();
} catch (MultiClassFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
Class<?> targetClass2=null;
String className2="com.test.User2";
try {
targetClass2 = manager.loadClass(className2);
} catch (ClassNotFoundException e) {
e.printStackTrace();
Assert.assertNull(targetClass2);
} catch (MultiClassFoundException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
@Test
public void testUnloadLoader() throws IOException, NotFoundException, CannotCompileException, ClassNotFoundException{
Utils.deleteDirectory(Paths.get(classParentDir));
Set<PosixFilePermission> perms=new HashSet<PosixFilePermission>();
perms.add(PosixFilePermission.GROUP_EXECUTE);
perms.add(PosixFilePermission.GROUP_READ);
perms.add(PosixFilePermission.GROUP_WRITE);
Files.createDirectories(Paths.get(classParentDir));
Map<String, Class<?>> properties = new HashMap<String, Class<?>>();
properties.put("name", java.lang.String.class);
properties.put("id", java.lang.String.class);
TestJarLoader.createDynamicJarWithSimplePojo(properties, "com.test.User", jarDir);
WeakReference<URLClassLoader> weakReference=new WeakReference<URLClassLoader>(JarLoader.loadFromJar(Thread.currentThread().getContextClassLoader(), jarDir));
URLClassLoader newLoader=weakReference.get();
Assert.assertNotNull(newLoader);
Class<?> userClass = newLoader.loadClass("com.test.User");
Assert.assertNotNull(userClass);
newLoader.close();
newLoader=null;
Assert.assertNotNull(weakReference.get());
System.gc();
Class<?> userClassAfterClose = weakReference.get().loadClass("com.test.User");
Assert.assertNotNull(weakReference.get());
Assert.assertNotNull(userClassAfterClose);
// Class<?> userClass = newLoader.loadClass("com.test.User");
// Assert.assertNotNull(userClass);
// newLoader.close();
//
// Utils.deleteDirectory(Paths.get(classParentDir));
//
// Class<?> userClassAfterUnLoad = newLoader.loadClass("com.test.User");
//
// Assert.assertNotNull(userClassAfterUnLoad);
// Class<?> userClassAfterUnLoad2 = newLoader.loadClass("com.test.User");
//
// Assert.assertNotNull(userClassAfterUnLoad2);
}
}