prerna.reactor.tax.TaxRetrieveValue2 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.tax;

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

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

import prerna.reactor.AbstractReactor;
import prerna.reactor.BaseJavaRuntime;
import prerna.reactor.PixelPlanner;
import prerna.sablecc2.om.GenRowStruct;
import prerna.sablecc2.om.InMemStore;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.VarStore;
import prerna.sablecc2.om.nounmeta.NounMetadata;

public class TaxRetrieveValue2 extends AbstractReactor {

	private static final Logger logger = LogManager.getLogger(TaxRetrieveValue2.class);

	// TODO: find a common place to put these
//	private static final String STORE_NOUN = "store";
	private static final String KEY_NOUN = "key";

	/**
	 * This reactor takes in 2 nouns store -> this points to the store name this
	 * will automatically be replaced with the NounMetadata that is in the
	 * pixel planner
	 * 
	 * key -> the key that the value is stored under
	 */
	@Override
	public NounMetadata execute() {
		PixelPlanner planner = getPlanner();
		Class javaClass = (Class) planner.getProperty("RUN_CLASS", "RUN_CLASS");
		BaseJavaRuntime javaRunClass = null;
		try {
			javaRunClass = javaClass.newInstance();
		} catch (InstantiationException | IllegalAccessException e) {
			logger.error("StackTrace: ", e);
		}

		if (javaRunClass == null) {
			throw new NullPointerException("javaRunClass cannot be null here.");
		}

		javaRunClass.execute();
		// for each scenario to hold the subportion
		// of data to send back
		InMemStore retStoreVar = new VarStore();

		GenRowStruct grs = this.store.getNoun(KEY_NOUN);
		int numGrs = grs.size();

		// go through and get all the aliases the FE want
		List aliases = new Vector<>();
		for (int i = 0; i < numGrs; i++) {
			String key = grs.get(i).toString();
			aliases.add(key);
		}
		
		// convert the alias to the hashcode which was used in execution
		Map aliasHashMap = TaxUtility.mapAliasToHash(aliases);
		
		for(String alias : aliasHashMap.keySet()) {
			String hashcode = aliasHashMap.get(alias);
			Object value =  javaRunClass.getVariables().get(hashcode);
			// return alias to the value associated with the hash
			retStoreVar.put(alias, new NounMetadata(value, PixelDataType.CONST_STRING));
		}
		
		return new NounMetadata(retStoreVar, PixelDataType.IN_MEM_STORE);
	}

	private PixelPlanner getPlanner() {
		GenRowStruct allNouns = getNounStore().getNoun(PixelDataType.PLANNER.getKey());
		PixelPlanner planner = null;
		if (allNouns != null && !allNouns.isEmpty()) {
			planner = (PixelPlanner) allNouns.get(0);
		}
		return planner;
	}

	@Override
	public List getOutputs() {
		// output is the signature
		List outputs = new Vector<>();
		NounMetadata output = new NounMetadata(this.signature, PixelDataType.LAMBDA);
		outputs.add(output);
		return outputs;
	}
}