88 lines
3.4 KiB
Java
88 lines
3.4 KiB
Java
package it.cavallium.dbengine.client.query;
|
|
|
|
import com.squareup.moshi.JsonAdapter;
|
|
import it.cavallium.dbengine.buffers.Buf;
|
|
import it.cavallium.dbengine.client.IntOpenHashSetJsonAdapter;
|
|
import it.cavallium.dbengine.client.query.current.CurrentVersion;
|
|
import it.cavallium.dbengine.client.query.current.IBaseType;
|
|
import it.cavallium.dbengine.client.query.current.IType;
|
|
import it.cavallium.dbengine.utils.BooleanListJsonAdapter;
|
|
import it.cavallium.dbengine.utils.ByteListJsonAdapter;
|
|
import it.cavallium.dbengine.utils.CharListJsonAdapter;
|
|
import it.cavallium.dbengine.utils.IntListJsonAdapter;
|
|
import it.cavallium.dbengine.utils.LongListJsonAdapter;
|
|
import it.cavallium.dbengine.utils.MoshiPolymorphic;
|
|
import it.cavallium.dbengine.utils.ShortListJsonAdapter;
|
|
import it.unimi.dsi.fastutil.booleans.BooleanList;
|
|
import it.unimi.dsi.fastutil.chars.CharList;
|
|
import it.unimi.dsi.fastutil.ints.IntList;
|
|
import it.unimi.dsi.fastutil.ints.IntOpenHashSet;
|
|
import it.unimi.dsi.fastutil.longs.LongList;
|
|
import it.unimi.dsi.fastutil.objects.Object2ObjectMap;
|
|
import it.unimi.dsi.fastutil.objects.Object2ObjectMaps;
|
|
import it.unimi.dsi.fastutil.objects.Object2ObjectOpenHashMap;
|
|
import it.unimi.dsi.fastutil.shorts.ShortList;
|
|
import java.util.HashSet;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
|
|
public class QueryMoshi extends MoshiPolymorphic<IType> {
|
|
|
|
private final Set<Class<IType>> abstractClasses;
|
|
private final Set<Class<IType>> concreteClasses;
|
|
private final Map<Class<?>, JsonAdapter<?>> extraAdapters;
|
|
|
|
@SuppressWarnings({"unchecked", "RedundantCast", "rawtypes"})
|
|
public QueryMoshi() {
|
|
super(true, GetterStyle.RECORDS_GETTERS);
|
|
HashSet<Class<IType>> abstractClasses = new HashSet<>();
|
|
HashSet<Class<IType>> concreteClasses = new HashSet<>();
|
|
|
|
// Add all super types with their implementations
|
|
for (var superTypeClass : CurrentVersion.getSuperTypeClasses()) {
|
|
for (Class<? extends IBaseType> superTypeSubtypesClass : CurrentVersion.getSuperTypeSubtypesClasses(
|
|
superTypeClass)) {
|
|
concreteClasses.add((Class<IType>) (Class) superTypeSubtypesClass);
|
|
}
|
|
abstractClasses.add((Class<IType>) (Class) superTypeClass);
|
|
}
|
|
|
|
// Add IBaseType with all basic types
|
|
abstractClasses.add((Class<IType>) (Class) IBaseType.class);
|
|
for (BaseType BaseType : BaseType.values()) {
|
|
concreteClasses.add((Class<IType>) (Class) CurrentVersion.getClass(BaseType));
|
|
}
|
|
|
|
this.abstractClasses = abstractClasses;
|
|
this.concreteClasses = concreteClasses;
|
|
Object2ObjectMap<Class<?>, JsonAdapter<?>> extraAdapters = new Object2ObjectOpenHashMap<>();
|
|
extraAdapters.put(BooleanList.class, new BooleanListJsonAdapter());
|
|
extraAdapters.put(Buf.class, new ByteListJsonAdapter());
|
|
extraAdapters.put(ShortList.class, new ShortListJsonAdapter());
|
|
extraAdapters.put(CharList.class, new CharListJsonAdapter());
|
|
extraAdapters.put(IntList.class, new IntListJsonAdapter());
|
|
extraAdapters.put(LongList.class, new LongListJsonAdapter());
|
|
extraAdapters.put(IntOpenHashSet.class, new IntOpenHashSetJsonAdapter());
|
|
this.extraAdapters = Object2ObjectMaps.unmodifiable(extraAdapters);
|
|
}
|
|
|
|
@Override
|
|
public Map<Class<?>, JsonAdapter<?>> getExtraAdapters() {
|
|
return extraAdapters;
|
|
}
|
|
|
|
@Override
|
|
protected Set<Class<IType>> getAbstractClasses() {
|
|
return abstractClasses;
|
|
}
|
|
|
|
@Override
|
|
protected Set<Class<IType>> getConcreteClasses() {
|
|
return concreteClasses;
|
|
}
|
|
|
|
@Override
|
|
protected boolean shouldIgnoreField(String fieldName) {
|
|
return fieldName.contains("$");
|
|
}
|
|
} |