1 | |
package org.jbehave.core.configuration.groovy; |
2 | |
|
3 | |
import groovy.lang.GroovyClassLoader; |
4 | |
import groovyjarjarasm.asm.ClassWriter; |
5 | |
import org.codehaus.groovy.ast.ClassNode; |
6 | |
import org.codehaus.groovy.control.CompilationUnit; |
7 | |
import org.codehaus.groovy.control.SourceUnit; |
8 | |
|
9 | |
import java.io.ByteArrayInputStream; |
10 | |
import java.io.InputStream; |
11 | |
import java.security.AccessController; |
12 | |
import java.security.PrivilegedAction; |
13 | |
import java.util.HashMap; |
14 | |
import java.util.Map; |
15 | |
|
16 | |
|
17 | |
|
18 | |
|
19 | |
|
20 | |
|
21 | |
|
22 | |
|
23 | |
|
24 | |
|
25 | |
|
26 | |
|
27 | 5 | public class BytecodeGroovyClassLoader extends GroovyClassLoader { |
28 | |
|
29 | 5 | private Map<String, byte[]> classBytes = new HashMap<String, byte[]>(); |
30 | |
|
31 | |
@Override |
32 | |
public InputStream getResourceAsStream(String name) { |
33 | 1 | if (classBytes.containsKey(name)) { |
34 | 1 | return new ByteArrayInputStream(classBytes.get(name)); |
35 | |
} |
36 | 0 | return super.getResourceAsStream(name); |
37 | |
} |
38 | |
|
39 | |
@Override |
40 | |
protected ClassCollector createCollector(CompilationUnit unit, SourceUnit su) { |
41 | |
|
42 | |
|
43 | 12 | InnerLoader loader = AccessController.doPrivileged(new PrivilegedAction<InnerLoader>() { |
44 | |
public InnerLoader run() { |
45 | 6 | return new InnerLoader(BytecodeGroovyClassLoader.this); |
46 | |
} |
47 | |
}); |
48 | 6 | return new BytecodeClassCollector(classBytes, loader, unit, su); |
49 | |
} |
50 | |
|
51 | 5 | public static class BytecodeClassCollector extends ClassCollector { |
52 | |
private final Map<String, byte[]> classBytes; |
53 | |
|
54 | |
public BytecodeClassCollector(Map<String, byte[]> classBytes, InnerLoader loader, CompilationUnit unit, |
55 | |
SourceUnit su) { |
56 | 6 | super(loader, unit, su); |
57 | 6 | this.classBytes = classBytes; |
58 | 6 | } |
59 | |
|
60 | |
@Override |
61 | |
protected Class<?> onClassNode(ClassWriter classWriter, ClassNode classNode) { |
62 | 6 | classBytes.put(classNode.getName() + ".class", classWriter.toByteArray()); |
63 | 6 | return super.onClassNode(classWriter, classNode); |
64 | |
} |
65 | |
} |
66 | |
|
67 | |
} |