com.harium.suneidesis.repository.KnowledgeBase 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.repository;

import com.harium.suneidesis.concept.Concept;
import com.harium.suneidesis.repository.decorator.EntryDecorator;
import com.harium.suneidesis.repository.generator.BaseIdGenerator;
import com.harium.suneidesis.repository.generator.IdGenerator;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

public abstract class KnowledgeBase implements KnowledgeBaseRepository {

    // Reserved word for introspection
    public static final String SELF = "_self";

    protected String name;

    protected IdGenerator idGenerator;

    protected List decorators = new ArrayList<>();

    public KnowledgeBase() {
        super();
        initGenerators();
    }

    public KnowledgeBase(String name) {
        super();
        this.name = name;
        initGenerators();
    }

    public KnowledgeBase(IdGenerator idGenerator) {
        super();
        this.idGenerator = idGenerator;
    }

    public void initGenerators() {
        idGenerator = new BaseIdGenerator();
    }

    public abstract Concept save(String key, Concept concept);

    @Override
    public String delete(Concept concept) {
        String id = concept.getId();
        delete(id);
        return id;
    }

    @Override
    public String save(Concept concept) {
        Concept id = concept.getIdConcept();
        String idText;
        if (id.isUnknown()) {
            idText = idGenerator.generateId(concept);
        } else {
            idText = id.getName();
        }

        decorate(concept);

        save(idText, concept);
        return idText;
    }

    public void merge(KnowledgeBase knowledgeBase) {
        Iterator iterator = knowledgeBase.iterator();
        while (iterator.hasNext()) {
            Concept concept = iterator.next();
            save(concept.getId(), concept);
        }
    }

    @Override
    public Iterator iterator() {
        return getAll().values().iterator();
    }

    protected Concept decorate(Concept concept) {
        for (EntryDecorator decorator : decorators) {
            decorator.decorate(concept);
        }
        return concept;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public IdGenerator getIdGenerator() {
        return idGenerator;
    }

    public void setIdGenerator(IdGenerator idGenerator) {
        this.idGenerator = idGenerator;
        // Update idGenerator if it's updated dynamically to prevent clashes
        this.idGenerator.update(this);
    }

}