prerna.util.gson.QueryConstantSelectorAdapter 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 com.google.gson.stream.JsonReader;
import com.google.gson.stream.JsonToken;
import com.google.gson.stream.JsonWriter;

import prerna.query.querystruct.selectors.IQuerySelector;
import prerna.query.querystruct.selectors.QueryConstantSelector;
import prerna.reactor.qs.SubQueryExpression;

public class QueryConstantSelectorAdapter extends AbstractSemossTypeAdapter implements IQuerySelectorAdapterHelper {

	@Override 
	public QueryConstantSelector read(JsonReader in) throws IOException {
		if (in.peek() == JsonToken.NULL) {
			in.nextNull();
			return null;
		}
		
		// remove the beginning objects
		in.beginObject();
		in.nextName();
		in.nextString();
		in.nextName();
		
		// now we read the actual content
		QueryConstantSelector value = readContent(in);
		in.endObject();
		return value;
	}

	@Override
	public QueryConstantSelector readContent(JsonReader in) throws IOException {
		QueryConstantSelector value = new QueryConstantSelector();
		in.beginObject();
		while(in.hasNext()) {
			String key = in.nextName();
			if(key.equals("alias")) {
				value.setAlias(in.nextString());
			} else if(key.equals("constant")) {
				if(in.peek() == JsonToken.NULL) {
					value.setConstant(null);
				} else if(in.peek() == JsonToken.NUMBER) {
					value.setConstant(in.nextDouble());
				} else if(in.peek() == JsonToken.BOOLEAN) {
					value.setConstant(in.nextBoolean());
				} else if(in.peek() == JsonToken.STRING) {
					value.setConstant(in.nextString());
				} else if(in.peek() == JsonToken.BEGIN_OBJECT){
					SubQueryExpressionAdapter adapter = new SubQueryExpressionAdapter();
					SubQueryExpression obj = adapter.read(in);
					obj.setInsight(this.insight);
					value.setConstant(obj);
				}
			}
		}
		in.endObject();
		return value;
	}
	
	@Override 
	public void write(JsonWriter out, QueryConstantSelector value) throws IOException {
		if (value == null) {
			out.nullValue();
			return;
		}
		
		// always start with the type of the query selector
		out.beginObject();
		out.name("type").value(IQuerySelector.SELECTOR_TYPE.CONSTANT.toString());
		out.name("content");
		// content object
		out.beginObject();
		out.name("alias").value(value.getAlias());
		Object val = value.getConstant();
		if(val == null) {
			out.name("constant").nullValue();
		} else if(val instanceof Number) {
			out.name("constant").value((Number) val);
		} else if(val instanceof Boolean) {
			out.name("constant").value((Boolean) val);
		} else if(val instanceof SubQueryExpression) {
			out.name("constant");
			SubQueryExpressionAdapter adapter = new SubQueryExpressionAdapter();
			adapter.write(out, (SubQueryExpression) val);
		} else {
			out.name("constant").value(val.toString());
		}
		out.endObject();
		out.endObject();
	}
}