prerna.reactor.utils.CheckRecommendOptimizationReactor 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.utils;

import java.io.IOException;
import java.util.List;

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

import prerna.engine.api.IDatabaseEngine;
import prerna.engine.api.IRawSelectWrapper;
import prerna.masterdatabase.utility.MasterDatabaseUtility;
import prerna.reactor.AbstractReactor;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.ReactorKeysEnum;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.util.Constants;
import prerna.util.Utility;

public class CheckRecommendOptimizationReactor extends AbstractReactor {

	private static final Logger classLogger = LogManager.getLogger(CheckRecommendOptimizationReactor.class);

	public CheckRecommendOptimizationReactor() {
		this.keysToGet = new String[] { ReactorKeysEnum.DATABASE.getKey() };
	}

	@Override
	public NounMetadata execute() {
		organizeKeys();
		String databaseId = this.keyValue.get(this.keysToGet[0]);
		databaseId = MasterDatabaseUtility.testDatabaseIdIfAlias(databaseId);

		IDatabaseEngine database = Utility.getDatabase(databaseId);
		List allTableCols = MasterDatabaseUtility.getAllTablesAndColumns(databaseId);
		for (Object[] tableCol : allTableCols) {
			if (tableCol.length == 4) {
				String table = tableCol[0] + "";
				String col = tableCol[1] + "";
				String dataType = tableCol[2] + "";
				// only care about strings
				if (dataType != null && dataType.equals("STRING")) {
					String queryCol = col;
					String uniqueValQuery = "SELECT DISTINCT ?concept ?unique WHERE "
							+ "{ BIND( AS ?concept)"
							+ "{?concept  ?unique}}";
					IRawSelectWrapper it = null;
					try {
						it = database.getOWLEngineFactory().getReadOWL().query(uniqueValQuery);
						if (it.hasNext()) {
							// it had a unique value so we assume they're all there
							return new NounMetadata(true, PixelDataType.BOOLEAN);
						} else {
							// the first string value didn't have a unique count so
							// we assume none do
							return new NounMetadata(false, PixelDataType.BOOLEAN);
						}
					} catch (Exception e) {
						classLogger.error(Constants.STACKTRACE, e);
					} finally {
						if(it != null) {
							try {
								it.close();
							} catch (IOException e) {
								classLogger.error(Constants.STACKTRACE, e);
							}
						}
					}
					return new NounMetadata(false, PixelDataType.BOOLEAN);
				}
			}
		}
		// none were strings so no need to get uniques
		return new NounMetadata(true, PixelDataType.BOOLEAN);
	}
}