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
|
package org.zwobble.couscous.ast.visitors;
import net.bytebuddy.implementation.bytecode.member.MethodReturn;
import org.zwobble.couscous.ast.Node;
import java.util.HashMap;
import java.util.Map;
import java.util.function.BiFunction;
import java.util.function.Function;
import static org.zwobble.couscous.util.ExtraMaps.entry;
public class DynamicNodeMapper {
private static final Map<Map.Entry<Class<?>, String>, Function<?, Function>> VISITOR_BUILDERS
= new HashMap<>();
// TODO: check return type -- at the very least, check the methods are consistent.
public static <T, R> Function<Node, R> instantiate(T visitor, String methodName) {
Function<T, Function<Node, R>> builder = buildClassSupplier(visitor.getClass(), methodName);
return builder.apply(visitor);
}
public static <T, R> BiFunction<Node, T, R> visitor(Class<T> clazz, String methodName) {
Function<T, Function<Node, R>> classSupplier = buildClassSupplier(clazz, methodName);
return (node, visitor) -> classSupplier.apply(visitor).apply(node);
}
private static <T> Function buildClassSupplier(Class<?> visitorClass, String methodName) {
Map.Entry<Class<?>, String> key = entry(visitorClass, methodName);
Function builder = VISITOR_BUILDERS.get(key);
if (builder == null) {
builder = DynamicNodeVisitor.buildClassSupplier(visitorClass, Function.class, methodName, MethodReturn.REFERENCE);
VISITOR_BUILDERS.put(key, builder);
}
return builder;
}
}
|
