org.relique.jdbc.csv.SimpleCaseExpression Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of csvjdbc Show documentation
Show all versions of csvjdbc Show documentation
a Java JDBC driver for reading comma-separated-value files
/*
* CsvJdbc - a JDBC driver for CSV files
* Copyright (C) 2015 Simon Chenery
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
*/
package org.relique.jdbc.csv;
import java.sql.SQLException;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
class SimpleCaseExpression extends Expression
{
private Expression caseExpression;
private List switches;
private List values;
private Expression elseExpression;
public SimpleCaseExpression(Expression caseExpression,
List switches, List values, Expression elseExpression)
{
this.caseExpression = caseExpression;
this.switches = switches;
this.values = values;
this.elseExpression = elseExpression;
}
@Override
public Object eval(Map env) throws SQLException
{
Object o1 = caseExpression.eval(env);
for (int i = 0; i < switches.size(); i++)
{
Expression expr = switches.get(i);
Object o2 = expr.eval(env);
Integer comparison = RelopExpression.compare((Comparable)o1, (Comparable)o2, env);
if (comparison.intValue() == 0)
return values.get(i).eval(env);
}
if (elseExpression != null)
return elseExpression.eval(env);
return null;
}
@Override
public String toString()
{
StringBuilder sb = new StringBuilder("CASE ");
sb.append(caseExpression.toString());
for (int i = 0; i < switches.size(); i++)
{
Expression expr = switches.get(i);
sb.append(" WHEN ").append(expr.toString()).append(" THEN ").append(values.get(i));
}
if (elseExpression != null)
sb.append(" ELSE ").append(elseExpression);
sb.append(" END");
return sb.toString();
}
@Override
public List usedColumns(Set availableColumns)
{
List result = new LinkedList<>();
result.addAll(caseExpression.usedColumns(availableColumns));
Iterator it = switches.iterator();
while (it.hasNext())
{
result.addAll(it.next().usedColumns(availableColumns));
}
it = values.iterator();
while (it.hasNext())
{
result.addAll(it.next().usedColumns(availableColumns));
}
if (elseExpression != null)
result.addAll(elseExpression.usedColumns(availableColumns));
return result;
}
@Override
public List aggregateFunctions()
{
List result = new LinkedList<>();
Iterator it = switches.iterator();
while (it.hasNext())
{
result.addAll(it.next().aggregateFunctions());
}
it = values.iterator();
while (it.hasNext())
{
result.addAll(it.next().aggregateFunctions());
}
if (elseExpression != null)
result.addAll(elseExpression.aggregateFunctions());
return result;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy