tech.ytsaurus.ysontree.YTreeNodeImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of yson-tree Show documentation
YTsaurus yson tree library
The newest version!
package tech.ytsaurus.ysontree;

import java.io.ByteArrayOutputStream;
import java.util.HashMap;
import java.util.Map;
import java.util.NoSuchElementException;
import java.util.Optional;
import java.util.function.Supplier;

import javax.annotation.Nullable;

public abstract class YTreeNodeImpl implements YTreeNode {

    private Map attributes;

    public YTreeNodeImpl(@Nullable Map attributes) {
        this.attributes = attributes == null ? null : new HashMap<>(attributes);
    }

    @Override
    public Map getAttributes() {
        init();
        return attributes;
    }

    @Override
    public void clearAttributes() {
        if (attributes != null) {
            attributes.clear();
        }
    }

    @Override
    public boolean containsAttributes() {
        return (attributes != null && !attributes.isEmpty());
    }

    @Override
    public boolean containsAttribute(String key) {
        return (attributes != null && attributes.containsKey(key));
    }

    @Override
    public Optional removeAttribute(String key) {
        if (attributes == null) {
            return Optional.empty();
        }
        return Optional.ofNullable(attributes.remove(key));
    }

    @Override
    public Optional putAttribute(String key, YTreeNode value) {
        init();
        return Optional.ofNullable(attributes.put(key, value));
    }

    @Override
    public Optional getAttribute(String key) {
        if (attributes == null) {
            return Optional.empty();
        }
        return Optional.ofNullable(attributes.get(key));
    }

    @Override
    public YTreeNode getAttributeOrThrow(String key) {
        if (attributes == null || !attributes.containsKey(key)) {
            throw new NoSuchElementException("Key " + key);
        }
        return attributes.get(key);
    }

    @Override
    public YTreeNode getAttributeOrThrow(String key, Supplier createMessage) {
        if (attributes == null || !attributes.containsKey(key)) {
            throw new NoSuchElementException(createMessage.get());
        }
        return attributes.get(key);
    }

    @Override
    public String toString() {
        return YTreeTextSerializer.serialize(this);
    }

    @Override
    public byte[] toBinary() {
        ByteArrayOutputStream output = new ByteArrayOutputStream();
        YTreeBinarySerializer.serialize(this, output);
        return output.toByteArray();
    }

    protected int hashCodeBase() {
        if (attributes == null) {
            return 0;
        }
        return attributes.hashCode();
    }

    protected boolean equalsBase(YTreeNode another0) {
        YTreeNodeImpl another = (YTreeNodeImpl) another0;
        if (attributes == null) {
            return another.attributes == null || another.attributes.isEmpty();
        }
        if (another.attributes == null) {
            return attributes.isEmpty();
        }
        return attributes.equals(another.attributes);
    }

    private void init() {
        if (attributes == null) {
            attributes = new HashMap<>();
        }
    }
}