package org.apache.avro.reflect;
import org.apache.avro.AvroRuntimeException;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.lang.reflect.TypeVariable;
import java.util.IdentityHashMap;
import java.util.Map;
class ReflectionUtil {
private ReflectionUtil() {
}
private static FieldAccess fieldAccess;
static {
resetFieldAccess();
}
static void resetFieldAccess() {
FieldAccess access = null;
try {
if (null == System.getProperty("avro.disable.unsafe")) {
FieldAccess unsafeAccess = load("org.apache.avro.reflect.FieldAccessUnsafe", FieldAccess.class);
if (validate(unsafeAccess)) {
access = unsafeAccess;
}
}
} catch (Throwable ignored) {
}
if (access == null) {
try {
FieldAccess reflectAccess = load("org.apache.avro.reflect.FieldAccessReflect", FieldAccess.class);
if (validate(reflectAccess)) {
access = reflectAccess;
}
} catch (Throwable oops) {
throw new AvroRuntimeException("Unable to load a functional FieldAccess class!");
}
}
fieldAccess = access;
}
private static <T> T load(String name, Class<T> type) throws Exception {
return ReflectionUtil.class.getClassLoader().loadClass(name).asSubclass(type).getDeclaredConstructor()
.newInstance();
}
public static FieldAccess getFieldAccess() {
return fieldAccess;
}
private static boolean validate(FieldAccess access) throws Exception {
return new AccessorTestClass().validate(access);
}
private static final class AccessorTestClass {
private boolean b = true;
protected byte by = 0xf;
public char c = 'c';
short s = 123;
int i = 999;
long l = 12345L;
float f = 2.2f;
double d = 4.4d;
Object o = "foo";
Integer i2 = 555;
private boolean validate(FieldAccess access) throws Exception {
boolean valid = true;
valid &= validField(access, "b", b, false);
valid &= validField(access, "by", by, (byte) 0xaf);
valid &= validField(access, "c", c, 'C');
valid &= validField(access, "s", s, (short) 321);
valid &= validField(access, "i", i, 111);
valid &= validField(access, "l", l, 54321L);
valid &= validField(access, "f", f, 0.2f);
valid &= validField(access, "d", d, 0.4d);
valid &= validField(access, "o", o, new Object());
valid &= validField(access, "i2", i2, -555);
return valid;
}
private boolean validField(FieldAccess access, String name, Object original, Object toSet) throws Exception {
FieldAccessor a;
boolean valid = true;
a = accessor(access, name);
valid &= original.equals(a.get(this));
a.set(this, toSet);
valid &= !original.equals(a.get(this));
return valid;
}
private FieldAccessor accessor(FieldAccess access, String name) throws Exception {
return access.getAccessor(this.getClass().getDeclaredField(name));
}
}
protected static Map<TypeVariable<?>, Type> resolveTypeVariables(Class<?> iface) {
return resolveTypeVariables(iface, new IdentityHashMap<>());
}
private static Map<TypeVariable<?>, Type> resolveTypeVariables(Class<?> iface, Map<TypeVariable<?>, Type> reuse) {
for (Type type : iface.getGenericInterfaces()) {
if (type instanceof ParameterizedType) {
ParameterizedType parameterizedType = (ParameterizedType) type;
Type rawType = parameterizedType.getRawType();
if (rawType instanceof Class<?>) {
Class<?> classType = (Class<?>) rawType;
TypeVariable<? extends Class<?>>[] typeParameters = classType.getTypeParameters();
Type[] actualTypeArguments = parameterizedType.getActualTypeArguments();
for (int i = 0; i < typeParameters.length; i++) {
reuse.putIfAbsent(typeParameters[i], reuse.getOrDefault(actualTypeArguments[i], actualTypeArguments[i]));
}
resolveTypeVariables(classType, reuse);
}
}
}
return reuse;
}
}