prerna.util.gson.TemporalEngineHardSelectQueryStructAdapter 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.util.gson;

import java.io.IOException;
import java.util.Map;

import com.google.gson.Gson;
import com.google.gson.TypeAdapter;
import com.google.gson.stream.JsonReader;
import com.google.gson.stream.JsonToken;
import com.google.gson.stream.JsonWriter;

import prerna.algorithm.api.ITableDataFrame;
import prerna.query.querystruct.AbstractQueryStruct.QUERY_STRUCT_TYPE;
import prerna.query.querystruct.TemporalEngineHardQueryStruct;

public class TemporalEngineHardSelectQueryStructAdapter extends AbstractSemossTypeAdapter {

	private static final Gson gson = new Gson();

	@Override
	public TemporalEngineHardQueryStruct read(JsonReader in) throws IOException {
		if (in.peek() == JsonToken.NULL) {
			in.nextNull();
			return null;
		}

		TemporalEngineHardQueryStruct qs = new TemporalEngineHardQueryStruct();

		in.beginObject();
		while(in.hasNext()) {
			String name = in.nextName();
			if(name.equals("qsType")) {
				qs.setQsType(QUERY_STRUCT_TYPE.valueOf(in.nextString()));
			} else if(name.equals("engineName")) {
				qs.setEngineId(in.nextString());
			} else if(name.equals("frameName")) {
				qs.setFrameName(in.nextString());
			} else if(name.equals("frameType")) {
				qs.setFrameType(in.nextString());
			} else if(name.equals("query")) {
				qs.setQuery(in.nextString());
			} else if(name.equals("config")) {
				TypeAdapter configReader = gson.getAdapter(Map.class);
				Map config = configReader.read(in);
				qs.setConfig(config);
			}
		}
		in.endObject();
		
		// the frame is not cached
		// but we store the frame name
		// set it in the QS if a 
		// context insight is defined
		if(this.insight != null) {
			if(qs.getFrameName() != null) {
				qs.setFrame((ITableDataFrame) this.insight.getVar(qs.getFrameName()));
			}
		}

		return qs;
	}
	
	@Override
	public void write(JsonWriter out, TemporalEngineHardQueryStruct value) throws IOException {
		if (value == null) {
			out.nullValue();
			return;
		}

		// this will be fun...
		// will try to go ahead and write everything

		out.beginObject();

		// lets do the easy ones first
		// qs type
		out.name("qsType").value(value.getQsType().toString());
		if(value.getEngineId() != null) {
			out.name("engineName").value(value.getEngineId());
		}
		if(value.getFrameName() != null) {
			out.name("frameName").value(value.getFrameName());
			out.name("frameType").value(value.getFrameType());
		}
		out.name("query").value(value.getQuery());
		
		if(value.getConfig() != null) {
			TypeAdapter configWriter = gson.getAdapter(Map.class);
			out.name("config");
			configWriter.write(out, value.getConfig());
		}
		out.endObject();
	}
}