net.amygdalum.testrecorder.serializers.CollectionsSetSerializer Maven / Gradle / Ivy

Go to download
package net.amygdalum.testrecorder.serializers;

import static java.util.stream.Collectors.toList;
import static net.amygdalum.testrecorder.util.TypeFilters.startingWith;
import static net.amygdalum.testrecorder.util.Types.inferType;
import static net.amygdalum.testrecorder.util.Types.parameterized;
import static net.amygdalum.testrecorder.util.Types.typeArgument;

import java.lang.reflect.Type;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Set;
import java.util.stream.Stream;

import net.amygdalum.testrecorder.types.SerializerSession;
import net.amygdalum.testrecorder.util.Reflections;
import net.amygdalum.testrecorder.values.SerializedSet;

public class CollectionsSetSerializer extends HiddenInnerClassSerializer {

	public CollectionsSetSerializer() {
		super(Collections.class);
	}

	@Override
	public List> getMatchingClasses() {
		return innerClasses().stream()
			.filter(startingWith("Unmodifiable", "Synchronized", "Checked", "Empty", "Singleton"))
			.filter(clazz -> Set.class.isAssignableFrom(clazz))
			.collect(toList());
	}

	@Override
	public Stream components(Object object, SerializerSession session) {
		return ((Set) object).stream();
	}

	@Override
	public SerializedSet generate(Class type, SerializerSession session) {
		return new SerializedSet(type);
	}

	@Override
	public void populate(SerializedSet serializedObject, Object object, SerializerSession session) {
		Type componentType = computeComponentType(serializedObject, object);

		for (Object element : (Set) object) {
			serializedObject.add(resolvedValueOf(session, componentType, element));
		}
		serializedObject.useAs(parameterized(Set.class, null, componentType));
	}

	private Type computeComponentType(SerializedSet serializedObject, Object object) {
		if (object.getClass().getSimpleName().contains("Checked")) {
			return getTypeField(object);
		}
		Stream definedTypes = Arrays.stream(serializedObject.getUsedTypes())
			.map(type -> typeArgument(type, 0).orElse(Object.class));
		Stream elementTypes = ((Set) object).stream()
			.filter(Objects::nonNull)
			.map(element -> element.getClass());
		Stream usedTypes = Stream.concat(definedTypes, elementTypes);
		
		return inferType(usedTypes.toArray(Type[]::new));
	}

	private Class getTypeField(Object object) {
		try {
			return (Class) Reflections.getValue("type", object);
		} catch (ReflectiveOperationException e) {
			return Object.class;
		}
	}

}