information-system-model-documentation-generator/src/main/java/org/gcube/informationsystem/utils/documentation/knowledge/FacetKnowledge.java

package org.gcube.informationsystem.utils.documentation.knowledge;

import java.util.Collection;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Set;
import java.util.TreeSet;

import org.gcube.informationsystem.types.reference.properties.LinkedEntity;

public class FacetKnowledge {
	
	private static FacetKnowledge singleton;
	
	public static FacetKnowledge getInstace() {
		if(FacetKnowledge.singleton==null) {
			FacetKnowledge.singleton = new FacetKnowledge();
		}
		return FacetKnowledge.singleton;
	}
	
	protected Map<String, Set<LinkedEntity>> map;
	
	private FacetKnowledge(){
		this.map = new LinkedHashMap<>();
	}
	
	protected void add(String typeName, LinkedEntity linkedEntity) {
		Set<LinkedEntity> list = map.get(typeName);
		if(list==null) {
			list = new TreeSet<>();
			map.put(typeName, list);
		}
		list.add(linkedEntity);
	}
	
	public void add(LinkedEntity linkedEntity) {
		String source = linkedEntity.getSource();
		add(source, linkedEntity);
		String relation = linkedEntity.getRelation();
		add(relation, linkedEntity);
		String target = linkedEntity.getTarget();
		add(target, linkedEntity);
	}
	
	public void addAll(Collection<LinkedEntity> linkedEntities) {
		for(LinkedEntity le : linkedEntities) {
			add(le);
		}
	}
	
	public Set<LinkedEntity> getUsage(String typeName){
		Set<LinkedEntity> list = map.get(typeName);
		return list;
	}
	
}