azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

More callbacks in preloader

Browse Source
This commit is contained in:
Andrey Breslav
2013-04-25 21:39:07 +04:00
parent 88112e500c
commit 4f967363a6
1 changed files with 27 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
compiler/preloader/src/org/jetbrains/jet/preloading/ClassPreloadingUtils.java
+27 -5
View File
@@ -29,8 +29,12 @@ import java.util.zip.ZipInputStream;
public class ClassPreloadingUtils {
public interface ClassHandler {
void beforeDefineClass(String name, int sizeInBytes);
public static abstract class ClassHandler {
public void beforeDefineClass(String name, int sizeInBytes) {}
public void afterDefineClass(String name) {}
public void beforeLoadJar(File jarFile) {}
public void afterLoadJar(File jarFile) {}
}
/**
@@ -48,7 +52,7 @@ public class ClassPreloadingUtils {
public static ClassLoader preloadClasses(
Collection<File> jarFiles, int classCountEstimation, ClassLoader parent, ClassHandler handler
) throws IOException {
Map<String, ResourceData> entries = loadAllClassesFromJars(jarFiles, classCountEstimation);
Map<String, ResourceData> entries = loadAllClassesFromJars(jarFiles, classCountEstimation, handler);
return createMemoryBasedClassLoader(parent, entries, handler);
}
@@ -86,7 +90,13 @@ public class ClassPreloadingUtils {
handler.beforeDefineClass(name, sizeInBytes);
}
return defineClass(name, resourceData.bytes, 0, sizeInBytes);
Class<?> definedClass = defineClass(name, resourceData.bytes, 0, sizeInBytes);
if (handler != null) {
handler.afterDefineClass(name);
}
return definedClass;
}
@Override
@@ -108,10 +118,18 @@ public class ClassPreloadingUtils {
};
}
private static Map<String, ResourceData> loadAllClassesFromJars(Collection<File> jarFiles, int classNumberEstimate) throws IOException {
private static Map<String, ResourceData> loadAllClassesFromJars(
Collection<File> jarFiles,
int classNumberEstimate,
ClassHandler handler
) throws IOException {
Map<String, ResourceData> resources = new HashMap<String, ResourceData>(classNumberEstimate);
for (File jarFile : jarFiles) {
if (handler != null) {
handler.beforeLoadJar(jarFile);
}
FileInputStream fileInputStream = new FileInputStream(jarFile);
try {
byte[] buffer = new byte[10 * 1024];
@@ -144,6 +162,10 @@ public class ClassPreloadingUtils {
// Ignore
}
}
if (handler != null) {
handler.afterLoadJar(jarFile);
}
}
return resources;
}