prerna.reactor.frame.rdbms.ExtractLettersReactor 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.frame.rdbms;

import java.util.List;
import java.util.Vector;

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

import prerna.ds.OwlTemporalEngineMeta;
import prerna.ds.rdbms.AbstractRdbmsFrame;
import prerna.reactor.frame.AbstractFrameReactor;
import prerna.sablecc2.om.GenRowStruct;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.PixelOperationType;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.util.Constants;

public class ExtractLettersReactor extends AbstractFrameReactor {
	
	private static final Logger classLogger = LogManager.getLogger(ExtractLettersReactor.class);
	
	public static final String COLUMNS = "columns";
	public static final String OVERRIDE = "override";
	public static final String ALPHA_COLUMN_NAME = "_ALPHA";

	@Override
	public NounMetadata execute() {
		AbstractRdbmsFrame frame = (AbstractRdbmsFrame) getFrame();
		// get table name
		String table = frame.getName();
		// get columns to extract alphabet characters
		List columns = getColumns();
		// check if user want to override the column or create new columns
		boolean overrideColumn = getOverride();
		// update existing columns
		if (overrideColumn) {
			String update = "";
			for (int i = 0; i < columns.size(); i++) {
				String column = columns.get(i);
				update += "UPDATE " + table + " SET " + column + "= REGEXP_REPLACE(" + column + ", '[^a-zA-Z\\_]', ''); ";
			}
			try {
				frame.getBuilder().runQuery(update);
			} catch (Exception e) {
				classLogger.error(Constants.STACKTRACE, e);
			}
		}
		// create new columns
		else {
			for (int i = 0; i < columns.size(); i++) {
				String column = columns.get(i);
				String newColumn = getCleanNewColName(frame, column + ALPHA_COLUMN_NAME);
				// add new column
				String update = "ALTER TABLE " + table + " ADD " + newColumn + " varchar(800);";
				// update extract alpha characters and underscores
				update += "UPDATE " + table + " SET " + newColumn + " = REGEXP_REPLACE(" + column + ", '[^a-zA-Z\\_]', '');";
				try {
					frame.getBuilder().runQuery(update);
				} catch (Exception e) {
					classLogger.error(Constants.STACKTRACE, e);
				}
				// if query runs successfully add new column metadata
				OwlTemporalEngineMeta metaData = frame.getMetaData();
				metaData.addProperty(table, table + "__" + newColumn);
				metaData.setAliasToProperty(table + "__" + newColumn, newColumn);
				metaData.setDataTypeToProperty(table + "__" + newColumn, "String");
			}
		}
		return new NounMetadata(frame, PixelDataType.FRAME, PixelOperationType.FRAME_DATA_CHANGE);
	}
	
	private List getColumns() {
		GenRowStruct grs = this.store.getNoun(COLUMNS);
		Vector columns = new Vector();
		NounMetadata noun;
		if (grs != null) {
			for (int i = 0; i < grs.size(); i++) {
				noun = grs.getNoun(i);
				if (noun != null) {
					String column = noun.getValue() + "";
					if (column.length() > 0) {
						columns.add(column);
					}
				}
			}
		}
		return columns;
	}
	
	private boolean getOverride() {
		GenRowStruct grs = this.store.getNoun(OVERRIDE);
		boolean override = false;
		NounMetadata noun;
		if (grs != null) {
			noun = grs.getNoun(0);
			override = (Boolean) noun.getValue();
		}
		return override;
	}

}