org.neo4j.jdbc.utils.PathSerializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of neo4j-jdbc-bolt Show documentation

Bolt implementation for the Neo4j JDBC Driver

There is a newer version: 5.0.0

package org.neo4j.jdbc.utils;

import com.fasterxml.jackson.core.JsonGenerator;
import com.fasterxml.jackson.databind.JsonSerializer;
import com.fasterxml.jackson.databind.SerializerProvider;
import org.neo4j.driver.v1.types.Path;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import static org.neo4j.jdbc.utils.DataConverterUtils.nodeToMap;
import static org.neo4j.jdbc.utils.DataConverterUtils.relationshipToMap;

public class PathSerializer extends JsonSerializer {
    @Override
    public void serialize(Path value, JsonGenerator jsonGenerator, SerializerProvider serializers) throws IOException {
        List> list = toPath(value);
        jsonGenerator.writeObject(list);
    }

    public static List> toPath(Path value) {
        List> list = new ArrayList<>();
        list.add(nodeToMap(value.start()));
        for (Path.Segment s : value) {
            list.add(relationshipToMap(s.relationship()));
            list.add(nodeToMap(s.end()));
        }
        return list;
    }
}