AttributesMapUtil.java

/*!
 * Copyright 2010 - 2015 Pentaho Corporation.  All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */
package org.pentaho.di.repository.pur;

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import org.pentaho.di.core.AttributesInterface;
import org.pentaho.di.core.exception.KettleException;
import org.pentaho.platform.api.repository2.unified.data.node.DataNode;
import org.pentaho.platform.api.repository2.unified.data.node.DataProperty;

public class AttributesMapUtil {

  public static final String NODE_ATTRIBUTE_GROUPS = "ATTRIBUTE_GROUPS";

  public static final void saveAttributesMap( DataNode dataNode, AttributesInterface attributesInterface )
    throws KettleException {
    Map<String, Map<String, String>> attributesMap = attributesInterface.getAttributesMap();

    DataNode attributeNodes = dataNode.getNode( NODE_ATTRIBUTE_GROUPS );
    if ( attributeNodes == null ) {
      attributeNodes = dataNode.addNode( NODE_ATTRIBUTE_GROUPS );
    }
    for ( String groupName : attributesMap.keySet() ) {
      DataNode attributeNode = attributeNodes.getNode( groupName );
      if ( attributeNode == null ) {
        attributeNode = attributeNodes.addNode( groupName );
      }
      Map<String, String> attributes = attributesMap.get( groupName );
      for ( String key : attributes.keySet() ) {
        String value = attributes.get( key );
        if ( key != null && value != null ) {
          attributeNode.setProperty( key, attributes.get( key ) );
        }
      }
    }
  }

  public static final void loadAttributesMap( DataNode dataNode, AttributesInterface attributesInterface )
    throws KettleException {
    Map<String, Map<String, String>> attributesMap = new HashMap<String, Map<String, String>>();
    attributesInterface.setAttributesMap( attributesMap );

    DataNode groupsNode = dataNode.getNode( NODE_ATTRIBUTE_GROUPS );
    if ( groupsNode != null ) {
      Iterable<DataNode> nodes = groupsNode.getNodes();
      for ( Iterator<DataNode> groupsIterator = nodes.iterator(); groupsIterator.hasNext(); ) {
        DataNode groupNode = groupsIterator.next();
        HashMap<String, String> attributes = new HashMap<String, String>();
        attributesMap.put( groupNode.getName(), attributes );
        Iterable<DataProperty> properties = groupNode.getProperties();
        for ( Iterator<DataProperty> propertiesIterator = properties.iterator(); propertiesIterator.hasNext(); ) {
          DataProperty dataProperty = propertiesIterator.next();
          String key = dataProperty.getName();
          String value = dataProperty.getString();
          if ( key != null && value != null ) {
            attributes.put( key, value );
          }
        }
      }
    }
  }

}