cn.wjybxx.dson.AbstractDsonObject Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dson-core Show documentation
Dson核心模块
There is a newer version: 2.2.0
package cn.wjybxx.dson;

import cn.wjybxx.base.CollectionUtils;
import cn.wjybxx.dson.internal.ValuesPolicy;

import java.util.*;

/**
 * @author wjybxx
 * date - 2023/6/18
 */
public abstract class AbstractDsonObject extends DsonValue implements Map {

    final Map valueMap;

    AbstractDsonObject(Map valueMap, ValuesPolicy policy) {
        Objects.requireNonNull(valueMap);
        this.valueMap = policy.applyMap(valueMap);
    }

    public Map getValueMap() {
        return Collections.unmodifiableMap(valueMap);
    }

    public DsonValue getOrThrow(K key) {
        Objects.requireNonNull(key);
        DsonValue value = valueMap.get(key);
        if (value == null) {
            throw new IllegalArgumentException("the value is absent, key " + key);
        }
        return value;
    }

    public DsonValue getOrElse(K key, DsonValue defaultValue) {
        Objects.requireNonNull(key);
        DsonValue value = valueMap.get(key);
        if (value == null) {
            return defaultValue;
        }
        return value;
    }

    @Override
    public DsonValue getOrDefault(Object key, DsonValue defaultValue) {
        return valueMap.getOrDefault(key, defaultValue);
    }

    /**
     * @throws NoSuchElementException 如果对象为空
     */
    public K firstKey() {
        return CollectionUtils.firstKey(valueMap);
    }

    public AbstractDsonObject append(K key, DsonValue value) {
        put(key, value);
        return this;
    }

    // region equals
    // 默认只比较valueMap

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        return o instanceof AbstractDsonObject that && valueMap.equals(that.valueMap);
    }

    @Override
    public int hashCode() {
        return valueMap.hashCode();
    }

    @Override
    public String toString() {
        return getClass().getSimpleName() + "{" +
                "valueMap=" + valueMap +
                '}';
    }

    // endregion

    // region 安全检查

    static  void checkKeyValue(K key, DsonValue value) {
        if (key == null) throw new IllegalArgumentException("key cant be null");
        if (value == null) throw new IllegalArgumentException("value cant be null");
        if (value.getDsonType() == DsonType.HEADER) throw new IllegalArgumentException("add Header");
    }

    @Override
    public DsonValue put(K key, DsonValue value) {
        checkKeyValue(key, value);
        return valueMap.put(key, value);
    }

    @Override
    public void putAll(Map m) {
        // 需要检测key-value的空
        for (Map.Entry entry : m.entrySet()) {
            put(entry.getKey(), entry.getValue());
        }
    }

    // endregion

    // region 代理实现

    @Override
    public int size() {
        return valueMap.size();
    }

    @Override
    public boolean isEmpty() {
        return valueMap.isEmpty();
    }

    @Override
    public boolean containsKey(Object key) {
        return valueMap.containsKey(key);
    }

    @Override
    public boolean containsValue(Object value) {
        return valueMap.containsValue(value);
    }

    @Override
    public DsonValue get(Object key) {
        return valueMap.get(key);
    }

    @Override
    public DsonValue remove(Object key) {
        return valueMap.remove(key);
    }

    @Override
    public void clear() {
        valueMap.clear();
    }

    @Override
    public Set keySet() {
        return valueMap.keySet();
    }

    @Override
    public Collection values() {
        return valueMap.values();
    }

    @Override
    public Set> entrySet() {
        return valueMap.entrySet(); // 这里也缺少对 entry.setValue 的校验
    }

    // endregion

}