com.harium.suneidesis.serialization.jackson.KnowledgeBaseJacksonSerializer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of core Show documentation

Project to represent knowledge

The newest version!

package com.harium.suneidesis.serialization.jackson;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.module.SimpleModule;
import com.harium.suneidesis.concept.Concept;
import com.harium.suneidesis.concept.attribute.Inheritance;
import com.harium.suneidesis.repository.KnowledgeBase;
import com.harium.suneidesis.serialization.KnowledgeBaseSerializer;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStream;

public class KnowledgeBaseJacksonSerializer implements KnowledgeBaseSerializer {

    private final ObjectMapper objectMapper;

    public KnowledgeBaseJacksonSerializer() {
        objectMapper = new ObjectMapper();
        SimpleModule module = new SimpleModule();
        module.addSerializer(KnowledgeBase.class, new CustomKnowledgeBaseSerializer());
        module.addSerializer(Concept.class, new ConceptSerializer());
        module.addSerializer(Inheritance.class, new InheritanceSerializer());
        objectMapper.registerModule(module);
    }

    public String serialize(KnowledgeBase knowledgeBase) throws IOException {
        ByteArrayOutputStream stream = new ByteArrayOutputStream();
        return serialize(knowledgeBase, stream).toString();
    }

    @Override
    public OutputStream serialize(KnowledgeBase knowledgeBase, OutputStream stream) throws IOException {
        objectMapper.writerWithDefaultPrettyPrinter().writeValue(stream, knowledgeBase);
        return stream;
    }

}