org.davidmoten.kool.json.JsonArray Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kool-json Show documentation

JSON support for kool

The newest version!

package org.davidmoten.kool.json;


import org.davidmoten.kool.Stream;

import com.fasterxml.jackson.core.JsonParser;
import com.fasterxml.jackson.core.JsonToken;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;


public final class JsonArray {

    private final Stream stream;
    private final ObjectMapper mapper;

    JsonArray(Stream flowable, ObjectMapper mapper) {
        this.stream = flowable;
        this.mapper = mapper;
    }

    public  Stream nodes() {
        return nodes_().map(p -> Util.MAPPER.readTree(p));
    }

    public Stream objectNodes() {
        return nodes_().map(p -> new LazyObjectNode(p));
    }

    public Stream valueNodes() {
        return nodes_().map(p -> new LazyValueNode(p));
    }

    public Stream arrayNodes() {
        return nodes_().map(p -> new LazyArrayNode(p, mapper));
    }

    public Stream field(String name) {
        return objectNodes().map(on -> on.get().get(name));
    }

    private Stream nodes_() {
        return Stream.defer(() -> {
            int[] depth = new int[1];
            return stream //
                    .doOnNext(p -> {
                        JsonToken t = p.currentToken();
                        if (t == JsonToken.START_OBJECT || t == JsonToken.START_ARRAY) {
                            depth[0]++;
                        } else if (t == JsonToken.END_OBJECT || t == JsonToken.END_ARRAY) {
                            depth[0]--;
                        }
                    }) //
                    .skipWhile(p -> p.currentToken() == JsonToken.FIELD_NAME
                            || p.currentToken() == JsonToken.START_ARRAY) //
                    .takeUntil(p -> depth[0] == 0) //
                    .filter(p -> p.currentToken() != JsonToken.END_ARRAY
                            && p.currentToken() != JsonToken.END_OBJECT); //
        });
    }

}