/**
* Copyright (C) 2009 - present by OpenGamma Inc. and the OpenGamma group of companies
*
* Please see distribution for license.
*/
package com.opengamma.core.position;
import java.math.BigDecimal;
import java.util.Map.Entry;
import org.fudgemsg.FudgeField;
import org.fudgemsg.FudgeMsg;
import org.fudgemsg.MutableFudgeMsg;
import org.fudgemsg.mapping.FudgeBuilder;
import org.fudgemsg.mapping.FudgeDeserializer;
import org.fudgemsg.mapping.FudgeSerializer;
import org.fudgemsg.mapping.GenericFudgeBuilderFor;
import org.fudgemsg.wire.types.FudgeWireType;
import org.threeten.bp.LocalDate;
import org.threeten.bp.OffsetTime;
import com.opengamma.core.position.impl.SimpleCounterparty;
import com.opengamma.core.position.impl.SimpleTrade;
import com.opengamma.core.security.Security;
import com.opengamma.core.security.impl.SimpleSecurityLink;
import com.opengamma.id.ExternalId;
import com.opengamma.id.ExternalIdBundle;
import com.opengamma.id.ObjectId;
import com.opengamma.id.UniqueId;
import com.opengamma.util.money.Currency;
/**
* Fudge message builder for {@code Trade}.
*/
@GenericFudgeBuilderFor(Trade.class)
public class TradeFudgeBuilder implements FudgeBuilder<Trade> {
/** Field name. */
public static final String ATTRIBUTES_FIELD_NAME = "attributes";
/** Field name. */
public static final String PREMIUM_TIME_FIELD_NAME = "premiumTime";
/** Field name. */
public static final String PREMIUM_DATE_FIELD_NAME = "premiumDate";
/** Field name. */
public static final String PREMIUM_CURRENCY_FIELD_NAME = "premiumCurrency";
/** Field name. */
public static final String PREMIUM_FIELD_NAME = "premium";
/** Field name. */
public static final String UNIQUE_ID_FIELD_NAME = "uniqueId";
/** Field name. */
public static final String QUANTITY_FIELD_NAME = "quantity";
/** Field name. */
protected static final String SECURITY_KEY_FIELD_NAME = "securityKey";
/** Field name. */
protected static final String SECURITY_ID_FIELD_NAME = "securityId";
/** Field name. */
protected static final String SECURITY_TARGET = "securityTarget";
/** Field name. */
public static final String COUNTERPARTY_FIELD_NAME = "counterpartyKey";
/** Field name. */
public static final String TRADE_DATE_FIELD_NAME = "tradeDate";
/** Field name. */
public static final String TRADE_TIME_FIELD_NAME = "tradeTime";
protected static MutableFudgeMsg buildMessageImpl(final FudgeSerializer serializer, final Trade trade) {
final MutableFudgeMsg message = serializer.newMessage();
if (trade.getUniqueId() != null) {
serializer.addToMessage(message, UNIQUE_ID_FIELD_NAME, null, trade.getUniqueId());
}
if (trade.getQuantity() != null) {
message.add(QUANTITY_FIELD_NAME, null, trade.getQuantity());
}
if (trade.getSecurityLink().getExternalId().size() > 0) {
serializer.addToMessage(message, SECURITY_KEY_FIELD_NAME, null, trade.getSecurityLink().getExternalId());
}
if (trade.getSecurityLink().getObjectId() != null) {
serializer.addToMessage(message, SECURITY_ID_FIELD_NAME, null, trade.getSecurityLink().getObjectId());
}
if (trade.getSecurityLink().getTarget() != null) {
serializer.addToMessageWithClassHeaders(message, SECURITY_TARGET, null, trade.getSecurityLink().getTarget());
}
if (trade.getCounterparty() != null) {
serializer.addToMessage(message, COUNTERPARTY_FIELD_NAME, null, trade.getCounterparty().getExternalId());
}
if (trade.getTradeDate() != null) {
message.add(TRADE_DATE_FIELD_NAME, null, trade.getTradeDate());
}
if (trade.getTradeTime() != null) {
message.add(TRADE_TIME_FIELD_NAME, null, trade.getTradeTime());
}
if (trade.getPremium() != null) {
message.add(PREMIUM_FIELD_NAME, null, trade.getPremium());
}
if (trade.getPremiumCurrency() != null) {
message.add(PREMIUM_CURRENCY_FIELD_NAME, null, trade.getPremiumCurrency().getCode());
}
if (trade.getPremiumDate() != null) {
message.add(PREMIUM_DATE_FIELD_NAME, null, trade.getPremiumDate());
}
if (trade.getPremiumTime() != null) {
message.add(PREMIUM_TIME_FIELD_NAME, null, trade.getPremiumTime());
}
if (haveAttributes(trade)) {
final MutableFudgeMsg attributesMsg = serializer.newMessage();
for (Entry<String, String> entry : trade.getAttributes().entrySet()) {
attributesMsg.add(entry.getKey(), entry.getValue());
}
serializer.addToMessage(message, ATTRIBUTES_FIELD_NAME, null, attributesMsg);
}
return message;
}
private static boolean haveAttributes(final Trade trade) {
return trade.getAttributes() != null && !trade.getAttributes().isEmpty();
}
@Override
public MutableFudgeMsg buildMessage(final FudgeSerializer serializer, final Trade trade) {
final MutableFudgeMsg tradeMsg = buildMessageImpl(serializer, trade);
tradeMsg.add(null, FudgeSerializer.TYPES_HEADER_ORDINAL, FudgeWireType.STRING, Trade.class.getName());
return tradeMsg;
}
@Override
public Trade buildObject(final FudgeDeserializer deserializer, final FudgeMsg message) {
return buildObjectImpl(deserializer, message);
}
protected static SimpleTrade buildObjectImpl(final FudgeDeserializer deserializer, final FudgeMsg message) {
SimpleSecurityLink secLink = new SimpleSecurityLink();
if (message.hasField(SECURITY_KEY_FIELD_NAME)) {
FudgeField secKeyField = message.getByName(SECURITY_KEY_FIELD_NAME);
if (secKeyField != null) {
secLink.setExternalId(deserializer.fieldValueToObject(ExternalIdBundle.class, secKeyField));
}
}
if (message.hasField(SECURITY_ID_FIELD_NAME)) {
FudgeField secIdField = message.getByName(SECURITY_ID_FIELD_NAME);
if (secIdField != null) {
secLink.setObjectId(deserializer.fieldValueToObject(ObjectId.class, secIdField));
}
}
if (message.hasField(SECURITY_TARGET)) {
FudgeField secTargetField = message.getByName(SECURITY_TARGET);
if (secTargetField != null) {
secLink.setTarget(deserializer.fieldValueToObject(Security.class, secTargetField));
}
}
SimpleTrade trade = new SimpleTrade();
trade.setSecurityLink(secLink);
if (message.hasField(UNIQUE_ID_FIELD_NAME)) {
FudgeField uniqueIdField = message.getByName(UNIQUE_ID_FIELD_NAME);
if (uniqueIdField != null) {
trade.setUniqueId(deserializer.fieldValueToObject(UniqueId.class, uniqueIdField));
}
}
if (message.hasField(QUANTITY_FIELD_NAME)) {
FudgeField quantityField = message.getByName(QUANTITY_FIELD_NAME);
if (quantityField != null) {
trade.setQuantity(message.getFieldValue(BigDecimal.class, quantityField));
}
}
if (message.hasField(COUNTERPARTY_FIELD_NAME)) {
FudgeField counterpartyField = message.getByName(COUNTERPARTY_FIELD_NAME);
if (counterpartyField != null) {
trade.setCounterparty(new SimpleCounterparty(deserializer.fieldValueToObject(ExternalId.class, counterpartyField)));
}
}
if (message.hasField(TRADE_DATE_FIELD_NAME)) {
FudgeField tradeDateField = message.getByName(TRADE_DATE_FIELD_NAME);
if (tradeDateField != null) {
trade.setTradeDate(message.getFieldValue(LocalDate.class, tradeDateField));
}
}
if (message.hasField(TRADE_TIME_FIELD_NAME)) {
FudgeField tradeTimeField = message.getByName(TRADE_TIME_FIELD_NAME);
if (tradeTimeField != null) {
trade.setTradeTime(message.getFieldValue(OffsetTime.class, tradeTimeField));
}
}
if (message.hasField(PREMIUM_FIELD_NAME)) {
trade.setPremium(message.getDouble(PREMIUM_FIELD_NAME));
}
if (message.hasField(PREMIUM_CURRENCY_FIELD_NAME)) {
String currencyCode = message.getString(PREMIUM_CURRENCY_FIELD_NAME);
if (currencyCode != null) {
trade.setPremiumCurrency(Currency.of(currencyCode));
}
}
if (message.hasField(PREMIUM_DATE_FIELD_NAME)) {
FudgeField premiumDate = message.getByName(PREMIUM_DATE_FIELD_NAME);
if (premiumDate != null) {
trade.setPremiumDate(message.getFieldValue(LocalDate.class, premiumDate));
}
}
if (message.hasField(PREMIUM_TIME_FIELD_NAME)) {
FudgeField premiumTime = message.getByName(PREMIUM_TIME_FIELD_NAME);
if (premiumTime != null) {
trade.setPremiumTime(message.getFieldValue(OffsetTime.class, premiumTime));
}
}
if (message.hasField(ATTRIBUTES_FIELD_NAME)) {
FudgeMsg attributesMsg = message.getMessage(ATTRIBUTES_FIELD_NAME);
for (FudgeField fudgeField : attributesMsg) {
String key = fudgeField.getName();
Object value = fudgeField.getValue();
if (key != null && value != null) {
trade.addAttribute(key, (String) value);
}
}
}
return trade;
}
}