prerna.reactor.masterdatabase.GetFrameDatabaseJoinsReactor Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of semoss Show documentation
SEMOSS
The newest version!
package prerna.reactor.masterdatabase;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Vector;

import prerna.algorithm.api.ITableDataFrame;
import prerna.auth.utils.SecurityEngineUtils;
import prerna.ds.OwlTemporalEngineMeta;
import prerna.masterdatabase.utility.MasterDatabaseUtility;
import prerna.reactor.frame.AbstractFrameReactor;
import prerna.sablecc2.om.GenRowStruct;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.PixelOperationType;
import prerna.sablecc2.om.ReactorKeysEnum;
import prerna.sablecc2.om.nounmeta.NounMetadata;

public class GetFrameDatabaseJoinsReactor extends AbstractFrameReactor {
	
	public GetFrameDatabaseJoinsReactor() {
		this.keysToGet = new String[]{ReactorKeysEnum.FRAME.getKey(), ReactorKeysEnum.DATABASE.getKey()};
	}

	@Override
	public NounMetadata execute() {
		Map>> connections = new HashMap>>();
		
		List dbFilters = null;
		String specificDbFilter = getDatabase();
		if(specificDbFilter != null) {
			if(!SecurityEngineUtils.userCanViewEngine(this.insight.getUser(), specificDbFilter)) {
				throw new IllegalArgumentException("Database " + specificDbFilter + " does not exist or user does not have access to database");
			}
			dbFilters = new Vector();
			dbFilters.add(specificDbFilter);
		} else {
			dbFilters = SecurityEngineUtils.getVisibleUserDatabaseIds(this.insight.getUser());
		}
		
		ITableDataFrame frame = getFrame();
		OwlTemporalEngineMeta meta = frame.getMetaData();
		Map> dbInfo = meta.getDatabaseInformation();
		
		for(String uniqueFrameHeader : dbInfo.keySet()) {
			// keep a list of the physical name ids
			List physicalNameIds = new Vector();
			List unknowns = new Vector();
			
			// grab a list of the physical names
			List qsData = dbInfo.get(uniqueFrameHeader);
			
			for(String[] info : qsData) {
				if(info.length == 2) {
					String physicalNameId = MasterDatabaseUtility.getPhysicalConceptIdFromPixelName(info[0], info[1]);
					physicalNameIds.add(physicalNameId);
				} else {
					unknowns.add(info[0]);
				}
			}

			// now query to find all related things to this frame header
			List conceptualNames = MasterDatabaseUtility.getConceptualNamesFromPhysicalIds(physicalNameIds);
			List headerConnections = MasterDatabaseUtility.getConceptualConnections(conceptualNames, dbFilters);
			
			// in the connections map
			// i want to put
			// {appId -> {app_pixel_selector -> [frame_unique_id1, frame_unique_id2] } }
			for(String[] hConn : headerConnections) {
				String appId = hConn[0];
				String appSelector = hConn[1];
				
				Map> appMap = null;
				if(connections.containsKey(appId)) {
					appMap = connections.get(appId);
				} else {
					appMap = new HashMap>();
					connections.put(appId, appMap);
				}
				
				List frameHeaderList = null;
				if(appMap.containsKey(appSelector)) {
					frameHeaderList = appMap.get(appSelector);
				} else {
					frameHeaderList = new Vector();
					appMap.put(appSelector, frameHeaderList);
				}
				
				frameHeaderList.add(uniqueFrameHeader);
			}
		}
		
		return new NounMetadata(connections, PixelDataType.CUSTOM_DATA_STRUCTURE, PixelOperationType.DATABASE_TRAVERSE_OPTIONS);
	}
	
	private String getDatabase() {
		GenRowStruct grs = this.store.getNoun(this.keysToGet[1]);
		if(grs != null && !grs.isEmpty()) {
			return grs.get(0).toString();
		}
		return null;
	}
}