prerna.engine.impl.json.JsonWrapper2 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.engine.impl.json;

import java.util.Hashtable;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ObjectNode;

import prerna.algorithm.api.SemossDataType;
import prerna.engine.api.IHeadersDataRow;
import prerna.om.HeadersDataRow;

public class JsonWrapper2 extends JsonWrapper {
	
	private ArrayNode data = null;
	
	@Override
	public void execute() throws Exception {
		// sorry for the bad way to transport data
		Hashtable output = (Hashtable)engine.execQuery(query);
		this.data = (ArrayNode) output.get("DATA");

		this.headers = (String[]) output.get("HEADERS");
		this.numColumns = this.headers.length;
		this.numRows = (Integer) output.get("COUNT");

		String[] strTypes = (String[]) output.get("TYPES");
		this.types = new SemossDataType[this.numColumns];
		for(int i = 0; i < this.numColumns; i++) {
			this.types[i] = SemossDataType.convertStringToDataType(strTypes[i]);
		}
	}

	@Override
	public IHeadersDataRow next() {
		ArrayNode thisRow = (ArrayNode) data.get(curRow);

		Object[] values = new Object[this.numColumns];
		for(int colIndex = 0; colIndex < this.numColumns; colIndex++) {
			Object retValue = null;
			JsonNode value = thisRow.get(colIndex);
			if(types[colIndex] == SemossDataType.STRING) {
				// check if value is an object if so stringify
				if(value.isArray()) {
					ArrayNode arrayValue = (ArrayNode)value;
					retValue = arrayValue.toString();
				}
				else if (value.isObject()) {
					ObjectNode objectValue = (ObjectNode) value;
					retValue = objectValue.toString();
				} else {
					retValue = value.asText("");
				}
			} else if(types[colIndex] == SemossDataType.DOUBLE) {
				retValue = value.asDouble();
			} else if(types[colIndex] == SemossDataType.INT) {
				retValue = value.asInt();
			}
			
			values[colIndex] = retValue;
		}
		this.curRow++;

		IHeadersDataRow retRow = new HeadersDataRow(this.headers, values);
		return retRow;
	}

}