D-Net

package eu.dnetlib.parthenos.registry;

import java.io.IOException;

import java.io.StringWriter;

import com.fasterxml.jackson.core.JsonFactory;

import com.fasterxml.jackson.core.JsonGenerator;

import org.springframework.stereotype.Component;

/**

 * Created by Alessia Bardi on 02/10/2017.

 *

 * @author Alessia Bardi

 */

@Component

public class RelWriter {

	protected String writeRelationship(final JsonFactory jsonFactory, final String relType, final String sourceUUID, final String sourceType, final String targetUUID, final String targetType)

			throws IOException {

		StringWriter sw = new StringWriter();

		JsonGenerator jg = jsonFactory.createGenerator(sw);

		jg.writeStartObject();

		jg.writeStringField("@class", relType);

		jg.writeObjectFieldStart("source");

		jg.writeObjectFieldStart("header");

		jg.writeStringField("uuid", sourceUUID);

		jg.writeEndObject();

		jg.writeStringField("@class", sourceType);

		jg.writeEndObject();

		jg.writeObjectFieldStart("target");

		jg.writeObjectFieldStart("header");

		jg.writeStringField("uuid", targetUUID);

		jg.writeEndObject();

		jg.writeStringField("@class", targetType);

		jg.writeEndObject();

		jg.close();

		return sw.toString();

	}

}