io.github.mmm.marshall.snakeyaml.impl.state.SnakeYamlObjectState Maven / Gradle / Ivy

Go to download
package io.github.mmm.marshall.snakeyaml.impl.state;

import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;

import io.github.mmm.marshall.StructuredReader.State;

/**
 * {@link SnakeYamlParentState} for objects as key/value pairs.
 *
 * @since 1.0.0
 */
public class SnakeYamlObjectState extends SnakeYamlParentState {

  final Map map;

  private Iterator> iterator;

  SnakeYamlObjectState(String name, SnakeYamlParentState parent, Map map) {

    super(name, parent);
    this.map = map;
    if (this.state == null) {
      next();
    }
  }

  @Override
  public SnakeYamlState next() {

    if (this.iterator == null) {
      if (this.map == null) {
        return null;
      }
      this.state = State.START_OBJECT;
      this.iterator = this.map.entrySet().iterator();
      return this;
    } else if (this.iterator.hasNext()) {
      this.state = null;
      Entry entry = this.iterator.next();
      return of(entry.getKey(), entry.getValue());
    } else if (this.state == State.END_OBJECT) {
      return this.parent;
    } else {
      this.state = State.END_OBJECT;
      return this;
    }
  }

  @Override
  public Map getValue() {

    return this.map;
  }

  @Override
  public void addValue(Object value) {

    String childName = getChildName();
    if (childName == null) {
      throw new IllegalStateException("Missing name");
    }
    this.map.put(childName, value);
  }

  @Override
  public boolean isObject() {

    return true;
  }

}