nl.topicus.jdbc.statement.AbstractSpannerExpressionVisitorAdapter Maven / Gradle / Ivy

package nl.topicus.jdbc.statement;

import javax.xml.bind.DatatypeConverter;

import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.DateValue;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.DoubleValue;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.Expression;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.ExpressionVisitorAdapter;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.HexValue;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.JdbcParameter;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.LongValue;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.NullValue;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.SignedExpression;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.StringValue;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.TimeValue;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.expression.TimestampValue;
import nl.topicus.jdbc.shaded.net.sf.jsqlparser.schema.Column;

abstract class AbstractSpannerExpressionVisitorAdapter extends ExpressionVisitorAdapter
{
	private ParameterStore parameterStore;

	private String column;

	AbstractSpannerExpressionVisitorAdapter(ParameterStore parameterStore)
	{
		this(parameterStore, null);
	}

	AbstractSpannerExpressionVisitorAdapter(ParameterStore parameterStore, String column)
	{
		this.parameterStore = parameterStore;
		this.column = column;
	}

	protected abstract void setValue(Object value);

	@Override
	public void visit(JdbcParameter parameter)
	{
		Object value = parameterStore.getParameter(parameter.getIndex());
		parameterStore.setColumn(parameter.getIndex(), column);
		setValue(value);
	}

	@Override
	public void visit(NullValue value)
	{
		setValue(null);
	}

	@Override
	public void visit(DoubleValue value)
	{
		setValue(value.getValue());
	}

	@Override
	public void visit(SignedExpression value)
	{
		Expression underlyingValue = value.getExpression();
		if (underlyingValue instanceof DoubleValue)
		{
			DoubleValue doubleValue = (DoubleValue) underlyingValue;
			doubleValue.setValue(value.getSign() == '-' ? -doubleValue.getValue() : doubleValue.getValue());
			visit(doubleValue);
		}
		else if (underlyingValue instanceof LongValue)
		{
			LongValue longValue = (LongValue) underlyingValue;
			longValue.setValue(value.getSign() == '-' ? -longValue.getValue() : longValue.getValue());
			visit(longValue);
		}
		else
		{
			super.visit(value);
		}
	}

	@Override
	public void visit(LongValue value)
	{
		setValue(value.getValue());
	}

	@Override
	public void visit(DateValue value)
	{
		setValue(value.getValue());
	}

	@Override
	public void visit(TimeValue value)
	{
		setValue(value.getValue());
	}

	@Override
	public void visit(TimestampValue value)
	{
		setValue(value.getValue());
	}

	@Override
	public void visit(StringValue value)
	{
		setValue(value.getValue());
	}

	@Override
	public void visit(HexValue value)
	{
		String stringValue = value.getValue().substring(2);
		byte[] byteValue = DatatypeConverter.parseHexBinary(stringValue);
		setValue(byteValue);
	}

	/**
	 * Booleans are not recognized by the parser, but are seen as column names.
	 * 
	 * @param column
	 */
	@Override
	public void visit(Column column)
	{
		String stringValue = column.getColumnName();
		if (stringValue.equalsIgnoreCase("true") || stringValue.equalsIgnoreCase("false"))
		{
			setValue(Boolean.valueOf(stringValue));
		}
	}

}