prerna.reactor.function.ExecuteStreamingFunctionEngineReactor 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.function;

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

import prerna.auth.utils.SecurityEngineUtils;
import prerna.engine.api.IFunctionEngine;
import prerna.engine.impl.function.StreamRESTFunctionEngine;
import prerna.reactor.AbstractReactor;
import prerna.reactor.job.JobReactor;
import prerna.sablecc2.om.GenRowStruct;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.ReactorKeysEnum;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.util.Utility;

public class ExecuteStreamingFunctionEngineReactor extends AbstractReactor {

	public ExecuteStreamingFunctionEngineReactor() {
		this.keysToGet = new String[] {ReactorKeysEnum.ENGINE.getKey(), ReactorKeysEnum.MAP.getKey()};
	}
	
	@Override
	public NounMetadata execute() {
		organizeKeys();
		String engineId = this.keyValue.get(this.keysToGet[0]);
		if(!SecurityEngineUtils.userCanViewEngine(this.insight.getUser(), engineId)) {
			throw new IllegalArgumentException("Fucntion Engine " + engineId + " does not exist or user does not have access to this function");
		}
		
		IFunctionEngine engine = Utility.getFunctionEngine(engineId);
		if(!(engine instanceof StreamRESTFunctionEngine)) {
			throw new IllegalArgumentException("This engine is not a streaming function engine");
		}
		
		Map parameterValues = getMap();
		parameterValues.put(JobReactor.JOB_KEY, this.getJobId());
		
		Object execValue = engine.execute(parameterValues);
		return new NounMetadata(execValue, PixelDataType.CUSTOM_DATA_STRUCTURE);
	}
	
	/**
	 * 
	 * @return
	 */
	private Map getMap() {
		Map parameterValues = new HashMap<>();

		GenRowStruct mapGrs = this.store.getNoun(this.keysToGet[1]);
		if(mapGrs != null && !mapGrs.isEmpty()) {
			for(int i = 0; i < mapGrs.size(); i++) {
				NounMetadata noun = mapGrs.getNoun(i);
				parameterValues.putAll( (Map) noun.getValue() );
			}
		} else {
			List