prerna.reactor.database.metaeditor.SaveOwlPositionsReactor 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.database.metaeditor;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.List;
import java.util.Map;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;

import prerna.cluster.util.ClusterUtil;
import prerna.engine.api.IDatabaseEngine;
import prerna.masterdatabase.utility.MasterDatabaseUtility;
import prerna.reactor.AbstractReactor;
import prerna.sablecc2.om.GenRowStruct;
import prerna.sablecc2.om.PixelDataType;
import prerna.sablecc2.om.ReactorKeysEnum;
import prerna.sablecc2.om.nounmeta.NounMetadata;
import prerna.util.EngineSyncUtility;
import prerna.util.UploadInputUtility;
import prerna.util.Utility;
import prerna.util.Constants;

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

public class SaveOwlPositionsReactor extends AbstractReactor {

	private static final Gson GSON = new GsonBuilder().setPrettyPrinting().create();
	protected static final Logger classLogger = LogManager.getLogger(SaveOwlPositionsReactor.class);

	public SaveOwlPositionsReactor() {
		this.keysToGet = new String[] {ReactorKeysEnum.DATABASE.getKey(), ReactorKeysEnum.POSITION_MAP.getKey()};
	}
	
	@Override
	public NounMetadata execute() {
		String databaseId = UploadInputUtility.getDatabaseNameOrId(this.store);
		if(databaseId == null) {
			throw new IllegalArgumentException("Must pass in the database id");
		}
		// run security tests + alias replacement
		databaseId = testDatabaseId(databaseId, true);
		Map positions = getPositionMap();
		if(positions == null || positions.isEmpty()) {
			throw new IllegalArgumentException("Must pass in the valid position map");
		}
		
		//TODO: below does not even work/is wrong
		//TODO: need to make a method to push/pull the positions file
		
		// write the json file in the database folder
		// just put it in the same location as the OWL
		IDatabaseEngine database = Utility.getDatabase(databaseId);
		ClusterUtil.pullOwl(databaseId);
		File positionFile = database.getOwlPositionFile();
		
		FileWriter writer = null;
		try {
			writer = new FileWriter(positionFile);
			GSON.toJson(positions, writer);
		} catch (IOException e) {
			classLogger.error(Constants.STACKTRACE, e);
		} finally {
			if(writer != null) {
				try {
					writer.close();
				} catch (IOException e) {
					classLogger.error(Constants.STACKTRACE, e);
				}
			}
		}
		ClusterUtil.pushOwl(databaseId);
		// update the positions cache
		EngineSyncUtility.setMetamodelPositions(databaseId, positions);
		MasterDatabaseUtility.saveMetamodelPositions(databaseId, positions);

		return new NounMetadata(true, PixelDataType.BOOLEAN);
	}
	
	private Map getPositionMap() {
		GenRowStruct grs = this.store.getNoun(this.keysToGet[1]);
		if(grs != null && !grs.isEmpty()) {
			List maps = grs.getNounsOfType(PixelDataType.MAP);
			if(maps != null && !maps.isEmpty()) {
				return (Map) maps.get(0).getValue();
			}
		}
		
		// check is passed as direct input
		List maps = this.curRow.getNounsOfType(PixelDataType.MAP);
		if(maps != null && !maps.isEmpty()) {
			return (Map) maps.get(0).getValue();
		}
		
		return null;
	}

}