com.opengamma.strata.product.TradeInfoBuilder Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2016 - present by OpenGamma Inc. and the OpenGamma group of companies
 *
 * Please see distribution for license.
 */
package com.opengamma.strata.product;

import java.time.LocalDate;
import java.time.LocalTime;
import java.time.ZoneId;
import java.util.HashMap;
import java.util.Map;

import com.opengamma.strata.basics.StandardId;
import com.opengamma.strata.collect.ArgChecker;

/**
 * Builder to create {@code TradeInfo}.
 * 
 * This builder allows a {@link TradeInfo} to be created.
 */
public final class TradeInfoBuilder implements PortfolioItemInfoBuilder {

  /**
   * The primary identifier for the trade.
   * 

   * The identifier is used to identify the trade.
   */
  private StandardId id;
  /**
   * The counterparty identifier.
   * 

   * An identifier used to specify the counterparty of the trade.
   */
  private StandardId counterparty;
  /**
   * The trade date.
   */
  private LocalDate tradeDate;
  /**
   * The trade time.
   */
  private LocalTime tradeTime;
  /**
   * The trade time-zone.
   */
  private ZoneId zone;
  /**
   * The settlement date.
   */
  private LocalDate settlementDate;
  /**
   * The trade attributes.
   * 

   * Trade attributes, provide the ability to associate arbitrary information
   * with a trade in a key-value map.
   */
  private final Map, Object> attributes = new HashMap<>();

  // creates an empty instance
  TradeInfoBuilder() {
  }

  // creates a populated instance
  TradeInfoBuilder(
      StandardId id,
      StandardId counterparty,
      LocalDate tradeDate,
      LocalTime tradeTime,
      ZoneId zone,
      LocalDate settlementDate,
      Map, Object> attributes) {

    this.id = id;
    this.counterparty = counterparty;
    this.tradeDate = tradeDate;
    this.tradeTime = tradeTime;
    this.zone = zone;
    this.settlementDate = settlementDate;
    this.attributes.putAll(attributes);
  }

  //-----------------------------------------------------------------------
  /**
   * Sets the primary identifier for the trade, optional.
   * 

   * The identifier is used to identify the trade.
   * 
   * @param id  the identifier
   * @return this, for chaining
   */
  @Override
  public TradeInfoBuilder id(StandardId id) {
    this.id = id;
    return this;
  }

  /**
   * Sets the counterparty identifier, optional.
   * 

   * An identifier used to specify the counterparty of the trade.
   * 
   * @param counterparty  the counterparty
   * @return this, for chaining
   */
  public TradeInfoBuilder counterparty(StandardId counterparty) {
    this.counterparty = counterparty;
    return this;
  }

  /**
   * Sets the trade date, optional.
   * 
   * @param tradeDate  the trade date
   * @return this, for chaining
   */
  public TradeInfoBuilder tradeDate(LocalDate tradeDate) {
    this.tradeDate = tradeDate;
    return this;
  }

  /**
   * Sets the trade time, optional.
   * 
   * @param tradeTime  the trade time
   * @return this, for chaining
   */
  public TradeInfoBuilder tradeTime(LocalTime tradeTime) {
    this.tradeTime = tradeTime;
    return this;
  }

  /**
   * Sets the trade time-zone, optional.
   * 
   * @param zone  the trade zone
   * @return this, for chaining
   */
  public TradeInfoBuilder zone(ZoneId zone) {
    this.zone = zone;
    return this;
  }

  /**
   * Sets the settlement date, optional.
   * 
   * @param settlementDate  the settlement date
   * @return this, for chaining
   */
  public TradeInfoBuilder settlementDate(LocalDate settlementDate) {
    this.settlementDate = settlementDate;
    return this;
  }

  /**
   * Adds a trade attribute to the map of attributes.
   * 
   * The attribute is added using {@code Map.put(type, value)} semantics.
   * 
   * @param  the type of the value
   * @param attributeType  the type providing meaning to the value
   * @param attributeValue  the value
   * @return this, for chaining
   */
  @Override
  @SuppressWarnings("unchecked")
  public  TradeInfoBuilder addAttribute(AttributeType attributeType, T attributeValue) {
    ArgChecker.notNull(attributeType, "attributeType");
    ArgChecker.notNull(attributeValue, "attributeValue");
    attributes.put(attributeType, attributeType.toStoredForm(attributeValue));
    return this;
  }

  /**
   * Builds the trade information.
   * 
   * @return the trade information
   */
  @Override
  public TradeInfo build() {
    return new TradeInfo(
        id,
        counterparty,
        tradeDate,
        tradeTime,
        zone,
        settlementDate,
        attributes);
  }

}