forked from json-iterator/java
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathCodegen.java
More file actions
107 lines (100 loc) · 4.05 KB
/
Codegen.java
File metadata and controls
107 lines (100 loc) · 4.05 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
package com.jsoniter;
import com.jsoniter.spi.*;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.CtNewMethod;
import java.lang.reflect.*;
import java.util.*;
class Codegen {
static boolean strictMode = false;
static ClassPool pool = ClassPool.getDefault();
public static void enableStrictMode() {
strictMode = true;
}
static Decoder getDecoder(String cacheKey, Type type) {
Decoder decoder = ExtensionManager.getDecoder(cacheKey);
if (decoder != null) {
return decoder;
}
return gen(cacheKey, type);
}
private synchronized static Decoder gen(String cacheKey, Type type) {
Decoder decoder = ExtensionManager.getDecoder(cacheKey);
if (decoder != null) {
return decoder;
}
for (Extension extension : ExtensionManager.getExtensions()) {
decoder = extension.createDecoder(cacheKey, type);
if (decoder != null) {
ExtensionManager.addNewDecoder(cacheKey, decoder);
return decoder;
}
}
Type[] typeArgs = new Type[0];
Class clazz;
if (type instanceof ParameterizedType) {
ParameterizedType pType = (ParameterizedType) type;
clazz = (Class) pType.getRawType();
typeArgs = pType.getActualTypeArguments();
} else {
clazz = (Class) type;
}
String source = genSource(cacheKey, clazz, typeArgs);
if ("true".equals(System.getenv("JSONITER_DEBUG"))) {
System.out.println(">>> " + cacheKey);
System.out.println(source);
}
try {
CtClass ctClass = pool.makeClass(cacheKey);
ctClass.setInterfaces(new CtClass[]{pool.get(Decoder.class.getName())});
CtMethod staticMethod = CtNewMethod.make(source, ctClass);
ctClass.addMethod(staticMethod);
CtMethod interfaceMethod = CtNewMethod.make("" +
"public Object decode(com.jsoniter.JsonIterator iter) {" +
"return decode_(iter);" +
"}", ctClass);
ctClass.addMethod(interfaceMethod);
decoder = (Decoder) ctClass.toClass().newInstance();
ExtensionManager.addNewDecoder(cacheKey, decoder);
return decoder;
} catch (Exception e) {
System.err.println("failed to generate decoder for: " + type + " with " + Arrays.toString(typeArgs));
System.err.println(source);
throw new JsonException(e);
}
}
private static String genSource(String cacheKey, Class clazz, Type[] typeArgs) {
if (CodegenImplNative.NATIVE_READS.containsKey(clazz.getName())) {
return CodegenImplNative.genNative(clazz.getName());
}
if (clazz.isArray()) {
return CodegenImplArray.genArray(clazz);
}
if (Map.class.isAssignableFrom(clazz)) {
return CodegenImplMap.genMap(clazz, typeArgs);
}
if (Collection.class.isAssignableFrom(clazz)) {
return CodegenImplArray.genCollection(clazz, typeArgs);
}
ClassDescriptor desc = ExtensionManager.getClassDescriptor(clazz, false);
List<Binding> allBindings = desc.allDecoderBindings();
for (Binding binding : allBindings) {
if (binding.failOnMissing || binding.failOnPresent) {
// only slice support mandatory tracking
return CodegenImplObject.genObjectUsingSlice(clazz, cacheKey, desc);
}
}
if (desc.failOnUnknownFields) {
// only slice support unknown field tracking
return CodegenImplObject.genObjectUsingSlice(clazz, cacheKey, desc);
}
if (allBindings.isEmpty()) {
return CodegenImplObject.genObjectUsingSkip(clazz, desc.ctor);
}
if (strictMode) {
return CodegenImplObject.genObjectUsingSlice(clazz, cacheKey, desc);
}
return CodegenImplObject.genObjectUsingHash(clazz, cacheKey, desc);
}
}