com.opengamma.engine.marketdata.manipulator.YieldCurveNodeExtractor.java Source code

Introduction

Here is the source code for com.opengamma.engine.marketdata.manipulator.YieldCurveNodeExtractor.java

Source

/**
 * Copyright (C) 2013 - present by OpenGamma Inc. and the OpenGamma group of companies
 *
 * Please see distribution for license.
 */
package com.opengamma.engine.marketdata.manipulator;

import java.util.Set;

import com.google.common.collect.Iterables;
import com.opengamma.core.marketdatasnapshot.YieldCurveKey;
import com.opengamma.engine.value.ValueProperties;
import com.opengamma.engine.value.ValuePropertyNames;
import com.opengamma.engine.value.ValueRequirementNames;
import com.opengamma.engine.value.ValueSpecification;
import com.opengamma.util.money.Currency;

/**
 * Extracts the structure id for a yield curve from a value specification.
 */
public class YieldCurveNodeExtractor extends NodeExtractor<YieldCurveKey> {

    /**
     * Constructs a yield curve extractor.
     */
    public YieldCurveNodeExtractor() {
        super(ValueRequirementNames.YIELD_CURVE);
    }

    @Override
    public StructureIdentifier<YieldCurveKey> getStructuredIdentifier(ValueSpecification spec) {
        Currency currency = Currency.of(spec.getTargetSpecification().getUniqueId().getValue());
        String curve = getSingleProperty(spec, ValuePropertyNames.CURVE);
        return StructureIdentifier.of(new YieldCurveKey(currency, curve));
    }

    private String getSingleProperty(final ValueSpecification spec, final String propertyName) {
        final ValueProperties properties = spec.getProperties();
        final Set<String> curves = properties.getValues(propertyName);
        return Iterables.getOnlyElement(curves);
    }
}