org.dsa.iot.dslink.util.json.JsonObject Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dslink Show documentation
SDK for the IoT DSA protocol
There is a newer version: 0.24.2
package org.dsa.iot.dslink.util.json;

import io.netty.util.CharsetUtil;

import java.util.*;

/**
 * @author Samuel Grenier
 */
@SuppressWarnings("unchecked")
public class JsonObject implements Iterable> {

    private final Map map;

    public JsonObject() {
        this(new LinkedHashMap());
    }

    public JsonObject(String json) {
        this(EncodingFormat.JSON, json.getBytes(CharsetUtil.UTF_8));
    }

    public JsonObject(EncodingFormat format,
                      byte[] json) {
        this(format, json, 0, json.length);
    }

    public JsonObject(EncodingFormat format,
                      byte[] json,
                      int offset,
                      int length) {
        this(Json.decodeMap(format, json, offset, length));
    }

    public JsonObject(Map map) {
        if (map == null) {
            throw new NullPointerException("map");
        }
        this.map = map;
    }

    public byte[] encode() {
        return encode(EncodingFormat.JSON);
    }

    public byte[] encode(EncodingFormat format) {
        return Json.encode(format, this);
    }

    public byte[] encodePrettily() {
        return encodePrettily(EncodingFormat.JSON);
    }

    public byte[] encodePrettily(EncodingFormat format) {
        return Json.encodePrettily(format, this);
    }

    public boolean contains(String key) {
        return map.containsKey(key);
    }

    public  T remove(String key) {
        return (T) Json.update(map.remove(key));
    }

    public  T get(String key) {
        return get(key, null);
    }


    public  T get(String key, T def) {
        T t = (T) Json.update(map.get(key));
        return t != null ? t : def;
    }

    public JsonObject put(String key, Object value) {
        Objects.requireNonNull(key);
        value = Json.checkAndUpdate(value);
        map.put(key, value);
        return this;
    }

    public int size() {
        return map.size();
    }

    public void mergeIn(JsonObject other) {
        mergeIn(other, false);
    }

    public void mergeIn(JsonObject other, boolean deep) {
        if (deep) {
            for (Map.Entry entry : other.map.entrySet()) {
                String name = entry.getKey();
                Object instance = entry.getValue();
                if (instance instanceof JsonArray) {
                    Object thisInst = map.get(name);
                    if (thisInst instanceof JsonArray) {
                        ((JsonArray) thisInst).mergeIn((JsonArray) instance);
                    }
                } else if (instance instanceof JsonObject) {
                    Object thisInst = map.get(name);
                    if (thisInst instanceof JsonObject) {
                        JsonObject jsonObject = (JsonObject) thisInst;
                        jsonObject.mergeIn((JsonObject) instance, true);
                    }
                } else {
                    map.put(name, instance);
                }
            }
            return;
        }
        map.putAll(other.map);
    }

    public Map getMap() {
        return Collections.unmodifiableMap(map);
    }

    @Override
    public String toString() {
        return new String(encode(EncodingFormat.JSON), CharsetUtil.UTF_8);
    }

    @Override
    public Iterator> iterator() {
        return new JsonIterator(map.entrySet().iterator());
    }

    private static class JsonIterator
            implements Iterator> {

        private final Iterator> it;

        public JsonIterator(Iterator> it) {
            this.it = it;
        }

        @Override
        public boolean hasNext() {
            return it.hasNext();
        }

        @Override
        public Map.Entry next() {
            Map.Entry entry = it.next();
            entry.setValue(Json.update(entry.getValue()));
            return entry;
        }

        @Override
        public void remove() {
            it.remove();
        }
    }
}