InvocationSequenceCustomCompatibleFieldSerializer.java

package rocks.inspectit.shared.all.serializer.impl;

import com.esotericsoftware.kryo.Kryo;
import com.esotericsoftware.kryo.io.Input;

import rocks.inspectit.shared.all.communication.data.InvocationSequenceData;
import rocks.inspectit.shared.all.serializer.schema.ClassSchemaManager;

/**
 * {@link CustomCompatibleFieldSerializer} for the {@link InvocationSequenceData} that in the
 * de-serialization process, connects the parents with the nested sequences.
 *
 * @author Ivan Senic
 *
 */
public class InvocationSequenceCustomCompatibleFieldSerializer extends CustomCompatibleFieldSerializer<InvocationSequenceData> {

	/**
	 * Default constructor.
	 *
	 * @param kryo
	 *            Kryo object.
	 * @param type
	 *            Type of class.
	 * @param schemaManager
	 *            Schema manager holding the schema for the given type.
	 *
	 * @see CustomCompatibleFieldSerializer#CustomCompatibleFieldSerializer(Kryo, Class,
	 *      ClassSchemaManager)
	 */
	public InvocationSequenceCustomCompatibleFieldSerializer(Kryo kryo, Class<? extends InvocationSequenceData> type, ClassSchemaManager schemaManager) {
		super(kryo, type, schemaManager);
	}

	/**
	 * {@inheritDoc}
	 */
	@Override
	public InvocationSequenceData read(Kryo kryo, Input input, Class<InvocationSequenceData> type) {
		InvocationSequenceData invocation = super.read(kryo, input, type);
		connectChildren(invocation);
		return invocation;
	}

	/**
	 * Sets the parent to all nested sequences of the invocation to the correct one.
	 *
	 * @param parent
	 *            Parent to start from.
	 */
	private void connectChildren(InvocationSequenceData parent) {
		if (null != parent.getNestedSequences()) {
			for (InvocationSequenceData child : parent.getNestedSequences()) {
				child.setParentSequence(parent);
				connectChildren(child);
			}
		}
	}

}