EObjectIdentityProperty.java

package org.emfjson.jackson.databind.property;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.core.JsonToken;
import com.fasterxml.jackson.databind.DeserializationContext;
import com.fasterxml.jackson.databind.SerializerProvider;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.Resource;
import org.emfjson.jackson.utils.ValueReader;
import org.emfjson.jackson.utils.ValueWriter;
import org.emfjson.jackson.annotations.EcoreIdentityInfo;
import org.emfjson.jackson.resource.JsonResource;

import java.io.IOException;

public class EObjectIdentityProperty extends EObjectProperty {

	private final ValueReader<Object, String> valueReader;
	private final ValueWriter<EObject, Object> valueWriter;

	public EObjectIdentityProperty(EcoreIdentityInfo info) {
		super(info.getProperty());

		this.valueReader = info.getValueReader();
		this.valueWriter = info.getValueWriter();
	}

	public void serialize(EObject bean, JsonGenerator jg, SerializerProvider provider) throws IOException {
		jg.writeObjectField(getFieldName(), valueWriter.writeValue(bean, provider));
	}

	@Override
	public EObject deserialize(JsonParser jp, DeserializationContext ctxt) throws IOException {
		return null;
	}

	@Override
	public void deserializeAndSet(JsonParser jp, EObject current, DeserializationContext ctxt, Resource resource) throws IOException {
		if (jp.getCurrentToken() == JsonToken.FIELD_NAME) {
			jp.nextToken();
		}

		Object value;
		switch (jp.getCurrentToken()) {
			case VALUE_STRING:
				value = jp.getValueAsString();
				break;
			case VALUE_NUMBER_INT:
				value = jp.getValueAsInt();
				break;
			case VALUE_NUMBER_FLOAT:
				value = jp.getValueAsLong();
				break;
			default:
				value = null;
		}

		if (value != null) {
			String id = valueReader.readValue(value, ctxt);
			if (resource instanceof JsonResource && id != null) {
				((JsonResource) resource).setID(current, id);
			}
		}
	}

}