D-Net

package eu.dnetlib.data.mdstore.mongo.codec;

import eu.dnetlib.data.mdstore.model.MDStoreRecord;

import org.apache.commons.lang3.StringUtils;

import org.bson.BsonReader;

import org.bson.BsonWriter;

import org.bson.codecs.Codec;

import org.bson.codecs.DecoderContext;

import org.bson.codecs.EncoderContext;

import org.springframework.stereotype.Component;

import static eu.dnetlib.data.mdstore.MDStoreConstants.*;

/**

 * Created by claudio on 20/03/2017.

 */

@Component

public class MDStoreRecordCodec implements Codec<MDStoreRecord> {

	@Override

	public MDStoreRecord decode(final BsonReader reader, final DecoderContext decoderContext) {

		reader.readStartDocument();

		reader.readObjectId();

		final MDStoreRecord record = MDStoreRecord.create()

				.setId(reader.readString(ID))

				.setOriginalId(reader.readString(ORIGINAL_ID))

				.setTimestamp(reader.readInt64(TIMESTAMP))

				.setBody(reader.readString(BODY));

		reader.readEndDocument();

		return record;

	}

	@Override

	public void encode(final BsonWriter writer, final MDStoreRecord r, final EncoderContext encoderContext) {

		writer.writeStartDocument();

		writer.writeString(ID, StringUtils.isNotBlank(r.getId()) ? r.getId() : "");

		writer.writeString(ORIGINAL_ID, StringUtils.isNotBlank(r.getOriginalId()) ? r.getOriginalId() : "");

		writer.writeInt64(TIMESTAMP, r.getTimestamp());

		writer.writeString(BODY, r.getBody());

		writer.writeEndDocument();

	}

	@Override

	public Class<MDStoreRecord> getEncoderClass() {

		return MDStoreRecord.class;

	}

}