Class YamlSerializer

src/main/java/org/rumbledb/serialization/YamlSerializer.java:12–115

import java.io.IOException;

public class YamlSerializer implements Serializer, java.io.Serializable {

    private static final long serialVersionUID = 1L;

    @SuppressWarnings("unused")
    private final org.rumbledb.serialization.SerializationParameters params;

    public YamlSerializer(SerializationParameters params) {
        this.params = params;
    }

    @Override
    public String serialize(Item i) {
        StringBuffer sb = new StringBuffer();
        serialize(i, sb, "", true);
        return sb.toString();
    }

    @Override
    public void serialize(Item item, StringBuffer sb, String indent, boolean isTopLevel) {
        YAMLFactory yamlFactory = new YAMLFactory();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        try {
            YAMLGenerator yamlGenerator = yamlFactory.createGenerator(baos);
            generateYAML(item, yamlGenerator);
            yamlGenerator.flush();
        } catch (IOException ioe) {
            RuntimeException e = new OurBadException("Not able to output YAML.");
            e.initCause(ioe);
            throw e;
        }
        sb.append(baos.toString());
    }

    private void generateYAML(Item item, YAMLGenerator yamlGenerator) throws IOException {
        if (item.isFunction()) {
            throw new FunctionsNonSerializableException();
        }
        if (item.isAtomic()) {
            generateYAMLAtomicValue(item, yamlGenerator);
            return;
        }
        if (item.isArray()) {
            yamlGenerator.writeStartArray();
            for (Item member : item.getItemMembers()) {
                generateYAML(member, yamlGenerator);
            }
            yamlGenerator.writeEndArray();
            return;
        }
        if (item.isMap() && !item.isObject()) {
            yamlGenerator.writeStartObject();
            for (Item key : item.getItemKeys()) {
                yamlGenerator.writeFieldName(key.getStringValue());
                appendMapValue(item, key, yamlGenerator);
            }
            yamlGenerator.writeEndObject();
            return;