prerna.om.InsightStore Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
There is a newer version: 4.2.2
package prerna.om;

import java.util.HashSet;
import java.util.Hashtable;
import java.util.Map;
import java.util.Set;

import org.apache.log4j.LogManager;
import org.apache.log4j.Logger;

import prerna.util.Utility;

public class InsightStore extends Hashtable {

	private static final Logger LOGGER = LogManager.getLogger(InsightStore.class.getName());
	private Map> sessionIdHash = new Hashtable>();
	
	// required for thick client
	public static Insight activeInsight = null;
	public static int idCount = 0;
	
	/**
	 * Singleton for the class
	 */
	private static InsightStore store;
	
	/**
	 * Constructor for class
	 */
	private InsightStore() {
		// do nothing
	}

	/**
	 * Returns the single insight store instance in the application
	 * @return
	 */
	public static InsightStore getInstance() {
		if(store == null) {
			store = new InsightStore();
		}
		return store; 
	}
	
	/**
	 * Adds an insight to be kept in memory while returning a unique key to retrieve the insight
	 * @param data					The insight to kept in storage
	 * @return						The unique id for the insight
	 */
	public String put(Insight data) {
		String uniqueID = data.getInsightId();
		if(uniqueID == null || uniqueID.isEmpty()) {
			uniqueID = (++idCount) + "_" + Utility.getRandomString(16);
		} else {
			uniqueID = (++idCount) + "_"  + uniqueID;
		}
		super.put(uniqueID, data);
		// update the new id inside the insight
		data.setInsightId(uniqueID);
		
		return uniqueID;
	}
	
	/**
	 * Returns a boolean true/false if insight was successfully remove using the key
	 * @param key					The unique id for the data-frame
	 * @return						boolean true if the key was successful at removing data, false otherwise
	 */
	public boolean remove(String key) {
		Insight data = super.remove(key);
		if(activeInsight != null && activeInsight.getInsightId().equalsIgnoreCase(key)) {
			activeInsight = null;
		}
		
		if(data != null) {
			return true;
		} else {
			return false;
		}
	}
	
	public void addToSessionHash(String sessionID, String insightID) {
		Set insightIDs = null;
		
		if(sessionIdHash.containsKey(sessionID)) {
			insightIDs = sessionIdHash.get(sessionID);
			if(insightIDs == null) {
				insightIDs = new HashSet();
			}
			insightIDs.add(insightID);
		} else {
			insightIDs = new HashSet();
			insightIDs.add(insightID);
		}
		
		sessionIdHash.put(sessionID, insightIDs);
	}
	
	public boolean removeFromSessionHash(String sessionID, String insightID) {
		if(!sessionIdHash.containsKey(sessionID)) {
			return false;
		}
		Set insightIDs = sessionIdHash.get(sessionID);
		if(insightIDs.contains(insightID)) {
			insightIDs.remove(insightID);
			return true;
		} 

		return false;
	}
	
	public Set getInsightIDsForSession(String sessionID) {
		return sessionIdHash.get(sessionID);
	}
	
	public Insight findInsightInStore(String engineName, String rdbmsId) {
		Insight retIn = null;
		INSIGHT_LOOP : for(String insightKey : this.keySet()) {
			Insight in = this.get(insightKey);
			String inEngineName = in.getEngineId();
			String inRdbmsId = in.getRdbmsId();
			if(engineName.equals(inEngineName) && rdbmsId.equals(inRdbmsId)) {
				retIn = in;
				break INSIGHT_LOOP;
			}
		}
		return retIn;
	}
	
	
	////////////////CODE FOR THICK CLIENT///////////////////////////
	public void setActiveInsight(Insight insight) {
		activeInsight = insight;
	}
	
	public void setActiveInsight(String insightID) {
		activeInsight = this.get(insightID);
	}

	public Insight getActiveInsight() {
		return activeInsight;
	}
	
	public Set getAllInsights() {
		return this.keySet();
	}
	
	public static int getIdCount(){
		return idCount;
	}
	
}