org.aksw.jenax.io.kryo.jena.RDFNodeSerializer Maven / Gradle / Ivy

Go to download
package org.aksw.jenax.io.kryo.jena;

import java.util.function.Function;

import org.apache.jena.graph.Node;
import org.apache.jena.rdf.model.Model;
import org.apache.jena.rdf.model.RDFNode;
import org.apache.jena.sparql.util.ModelUtils;

import com.esotericsoftware.kryo.Kryo;
import com.esotericsoftware.kryo.Serializer;
import com.esotericsoftware.kryo.io.Input;
import com.esotericsoftware.kryo.io.Output;


public class RDFNodeSerializer
        extends Serializer {

    protected Function fn;

    public RDFNodeSerializer(Function fn) {
        super();
        this.fn = fn;
    }

    @Override
    public T read(Kryo kryo, Input input, Class clazz) {
        Node node = (Node)kryo.readClassAndObject(input);
        Model model = (Model)kryo.readClassAndObject(input);
        RDFNode rdfNode = ModelUtils.convertGraphNodeToRDFNode(node, model);
        T result = fn.apply(rdfNode);
        return result;
    }

    @Override
    public void write(Kryo kryo, Output output, T rdfNode) {
        Node node = rdfNode.asNode();
        Model model = rdfNode.getModel();
        kryo.writeClassAndObject(output, node);
        kryo.writeClassAndObject(output, model);
    }
}