com.opengamma.integration.tool.portfolio.xml.v1_0.jaxb.AttributeMapAdapter.java Source code

Introduction

Here is the source code for com.opengamma.integration.tool.portfolio.xml.v1_0.jaxb.AttributeMapAdapter.java

Source

/**
 * Copyright (C) 2013 - present by OpenGamma Inc. and the OpenGamma group of companies
 *
 * Please see distribution for license.
 */
package com.opengamma.integration.tool.portfolio.xml.v1_0.jaxb;

import java.util.List;
import java.util.Map;

import javax.xml.bind.annotation.adapters.XmlAdapter;

import com.google.common.collect.Lists;
import com.google.common.collect.Maps;

public class AttributeMapAdapter extends XmlAdapter<AdditionalAttributes, Map<String, String>> {

    @Override
    public Map<String, String> unmarshal(AdditionalAttributes attrs) throws Exception {

        Map<String, String> map = Maps.newHashMap();

        for (Attribute attribute : attrs.getAttributes()) {
            map.put(attribute.getName(), attribute.getValue());
        }
        return map;
    }

    @Override
    public AdditionalAttributes marshal(Map<String, String> map) throws Exception {

        List<Attribute> attrs = Lists.newArrayList();

        for (Map.Entry<String, String> entry : map.entrySet()) {
            Attribute attribute = new Attribute();
            attribute.setName(entry.getKey());
            attribute.setValue(entry.getValue());
            attrs.add(attribute);
        }

        AdditionalAttributes attributes = new AdditionalAttributes();
        attributes.setAttributes(attrs);
        return attributes;
    }
}