blob: 158a8ec79abd0cc4bc8ae2d7f234c3729f2649f6 [file] [log] [blame]
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you under the Apache License, Version 2.0 (the
* "License"); you may not use this file except in compliance
* with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
* KIND, either express or implied. See the License for the
* specific language governing permissions and limitations
* under the License.
*/
package org.apache.empire.db;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Set;
import java.util.Vector;
import org.apache.empire.commons.StringUtils;
import org.apache.empire.data.DataType;
import org.apache.empire.db.exceptions.DatabaseMismatchException;
import org.apache.empire.db.expr.compare.DBCompareColExpr;
import org.apache.empire.db.expr.compare.DBCompareExpr;
import org.apache.empire.db.expr.join.DBColumnJoinExpr;
import org.apache.empire.db.expr.join.DBCompareJoinExpr;
import org.apache.empire.db.expr.join.DBCrossJoinExpr;
import org.apache.empire.db.expr.join.DBJoinExpr;
import org.apache.empire.db.expr.set.DBSetExpr;
import org.apache.empire.exceptions.InternalException;
import org.apache.empire.exceptions.InvalidArgumentException;
import org.apache.empire.exceptions.MiscellaneousErrorException;
import org.apache.empire.exceptions.ObjectNotValidException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
* This abstract class handles the creation of the SQL-Commands.
* There are methods to create SQL-Commands, like update, insert,
* delete and select.
*/
public abstract class DBCommand extends DBCommandExpr
implements Cloneable
{
private final static long serialVersionUID = 1L;
// Logger
protected static final Logger log = LoggerFactory.getLogger(DBCommand.class);
// Distinct Select
protected boolean selectDistinct = false;
// Lists
protected List<DBColumnExpr> select = null;
protected List<DBSetExpr> set = null;
protected List<DBJoinExpr> joins = null; // Join Info
protected List<DBCompareExpr> where = null;
protected List<DBCompareExpr> having = null;
protected List<DBColumnExpr> groupBy = null;
// Parameters for prepared Statements
protected Vector<DBCmdParam> cmdParams = null;
private int paramUsageCount= 0;
// Database
private transient DBDatabase db;
/**
* Constructs a new DBCommand object and set the specified DBDatabase object.
*
* @param db the current database object
*/
protected DBCommand(DBDatabase db)
{
this.db = db;
}
/**
* Custom serialization for transient database.
*/
private void writeObject(ObjectOutputStream strm) throws IOException
{
if (db==null)
{ // No database
strm.writeObject("");
strm.defaultWriteObject();
return;
}
String dbid = db.getId();
strm.writeObject(dbid);
if (log.isDebugEnabled())
log.debug("Serialization: writing DBCommand "+dbid);
// write the rest
strm.defaultWriteObject();
}
/**
* Custom deserialization for transient database.
*/
private void readObject(ObjectInputStream strm) throws IOException, ClassNotFoundException,
SecurityException, NoSuchFieldException, IllegalArgumentException, IllegalAccessException
{
String dbid = String.valueOf(strm.readObject());
if (StringUtils.isNotEmpty(dbid))
{ // Find database
if (log.isDebugEnabled())
log.debug("Serialization: reading DBCommand "+dbid);
// find database
DBDatabase sdb = DBDatabase.findById(dbid);
if (sdb==null)
throw new ClassNotFoundException(dbid);
// set final field
Field f = DBCommand.class.getDeclaredField("db");
f.setAccessible(true);
f.set(this, sdb);
f.setAccessible(false);
}
// read the rest
strm.defaultReadObject();
}
/**
* internally used to reset the command param usage count.
* Note: Only one thread my generate an SQL statement
*/
protected void resetParamUsage()
{
paramUsageCount = 0;
}
/**
* internally used to reorder the command params to match their order of occurance
*/
protected synchronized void notifyParamUsage(DBCmdParam param)
{
int index = cmdParams.indexOf(param);
if (index < paramUsageCount)
{ // Error: parameter probably used twice in statement!
throw new MiscellaneousErrorException("A parameter may only be used once in a command.");
}
if (index > paramUsageCount)
{ // Correct parameter order
cmdParams.remove(index);
cmdParams.insertElementAt(param, paramUsageCount);
}
paramUsageCount++;
}
/**
* internally used to remove the command param used in a constraint
*/
private void removeCommandParam(DBCompareColExpr cmp)
{
if (cmdParams!=null && (cmp.getValue() instanceof DBCmdParam))
cmdParams.remove(cmp.getValue());
}
/**
* internally used to remove all command params used in a list of constraints
*/
private void removeAllCommandParams(List<DBCompareExpr> list)
{
if (cmdParams == null)
return;
for(DBCompareExpr cmp : list)
{ // Check whether it is a compare column expr.
if (!(cmp instanceof DBCompareColExpr))
continue;
// Check the value is a DBCommandParam
removeCommandParam((DBCompareColExpr)cmp);
}
}
/**
* Creates a clone of this class.
*/
@Override
public DBCommand clone()
{
try
{
DBCommand clone = (DBCommand)super.clone();
clone.db = db;
// Clone lists
if (select!=null)
clone.select = new ArrayList<DBColumnExpr>(select);
if (set!=null)
clone.set = new ArrayList<DBSetExpr>(set);
if (joins!=null)
clone.joins = new ArrayList<DBJoinExpr>(joins);
if (where!=null)
clone.where = new ArrayList<DBCompareExpr>(where);
if (groupBy!=null)
clone.groupBy = new ArrayList<DBColumnExpr>(groupBy);
if (having!=null)
clone.having = new ArrayList<DBCompareExpr>(having);
if (cmdParams!=null)
{ // clone params
clone.paramUsageCount = 0;
clone.cmdParams = new Vector<DBCmdParam>();
for (DBCmdParam p : cmdParams)
{
DBCmdParam param = new DBCmdParam(this, p.getDataType(), p.getValue());
clone.cmdParams.add(param);
}
}
// done
return clone;
} catch (CloneNotSupportedException e) {
log.error("Cloning DBCommand object failed!", e);
throw new InternalException(e);
}
}
/**
* Returns the current DBDatabase object.
*
* @return the current DBDatabase object
*/
@Override
public DBDatabase getDatabase()
{
return db;
}
@Override
public boolean isValid()
{
return isValidQuery() || isValidUpdate();
}
/**
* Returns whether the command object can produce a select sql-statement.
*
* @return true if at least one select expression has been set
*/
public boolean isValidQuery()
{
return (select != null);
}
/**
* Returns whether the command object can produce a update sql-statement.
*
* @return true if a set expression has been set.
*/
public boolean isValidUpdate()
{
return (set != null);
}
/**
* Sets whether or not the select statement should contain
* the distinct directive .
*/
public void selectDistinct()
{
this.selectDistinct = true;
}
/**
* Returns whether or not the select statement will be distinct or not.
*
* @return true if the select will contain the distinct directive or false otherwise.
*/
public boolean isSelectDistinct()
{
return selectDistinct;
}
/**
* Adds a DBColumnExpr object to the Vector: 'select'.
*
* @param expr the DBColumnExpr object
*/
public void select(DBColumnExpr expr)
{ // Select this column
checkDatabase(expr);
if (select == null)
select = new ArrayList<DBColumnExpr>();
if (expr != null && select.contains(expr) == false)
select.add(expr);
}
/**
* Adds a list of columns to the select phrase of an sql statement.
*
* @param exprs an vararg of DBColumnExpr's to select
*/
public final void select(DBColumnExpr... exprs)
{
for (DBColumnExpr expr : exprs)
{
select(expr);
}
}
/**
* Adds a collection of columns to the select phrase of an sql statement.
*
* @param columns the column expressions to add
*/
public final void select(Collection<? extends DBColumnExpr> columns)
{
for (DBColumnExpr expr : columns)
{
select(expr);
}
}
/**
* Adds a list of columns with their qualified name to the select phrase of an sql statement.
*
* @param exprs one or more columns to select
*/
public void selectQualified(DBColumn... columns)
{
for (DBColumn col : columns)
{
select(col.qualified());
}
}
/**
* Adds a collection of columns to the select phrase of an sql statement.
*
* @param columns the column expressions to add
*/
public final void selectQualified(Collection<? extends DBColumn> columns)
{
for (DBColumn col : columns)
{
select(col.qualified());
}
}
/**
* returns true if prepared statements are enabled for this database
*/
protected boolean isPreparedStatementsEnabled()
{
return db.isPreparedStatementsEnabled();
}
/**
* returns true if a cmdParam should be used for the given column or false otherwise
*/
protected boolean useCmdParam(DBColumn col, Object value)
{
// Cannot wrap DBExpr or DBSystemDate
if (value instanceof DBExpr || value instanceof DBDatabase.DBSystemDate)
return false;
// Check if prepared statements are enabled
if (isPreparedStatementsEnabled())
return true;
// Only use a command param if column is of type BLOB or CLOB
DataType dt = col.getDataType();
return ( dt==DataType.BLOB || dt==DataType.CLOB );
}
/**
* Adds a single set expressions to this command
* Use column.to(...) to create a set expression
* @param expr the DBSetExpr object(s)
*/
public void set(DBSetExpr expr)
{
checkDatabase(expr);
if (set == null)
set = new ArrayList<DBSetExpr>();
for (int i = 0; i < set.size(); i++)
{
DBSetExpr chk = set.get(i);
if (chk.column.equals(expr.column))
{ // Overwrite existing value
if (useCmdParam(expr.column, expr.value))
{ // replace parameter value
// int index = ((DBCommandParam) chk.value).index;
// this.setCmdParam(index, getCmdParamValue(expr.column, expr.value));
if (chk.value instanceof DBCmdParam)
((DBCmdParam)chk.value).setValue(expr.value);
else
chk.value = addParam(expr.column.getDataType(), expr.value);
}
else
{ // remove from parameter list (if necessary)
if (cmdParams!=null && chk.value instanceof DBCmdParam)
cmdParams.remove(chk.value);
// replace value
chk.value = expr.value;
}
return;
}
}
// Replace with parameter
if (useCmdParam(expr.column, expr.value))
expr.value = addParam(expr.column.getDataType(), expr.value);
// new Value!
set.add(expr);
}
/**
* Adds a list of set expressions to this command
* Use column.to(...) to create a set expression
* @param expr the DBSetExpr object(s)
*/
public final void set(DBSetExpr... exprs)
{
for (int i=0; i<exprs.length; i++)
set(exprs[i]);
}
/**
* Checks whether a column is in the list of set expressions
* @param column
* @return <code>true</code> if there is a set expression
*/
protected boolean hasSetExprOn(DBColumn column)
{
if (set==null)
return false;
Iterator<DBSetExpr> i = set.iterator();
while (i.hasNext())
{
DBSetExpr chk = i.next();
if (chk.column.equals(column))
return true;
}
return false;
}
/**
* Adds an command parameter which will be used in a prepared statement.
* The command parameter returned may be used to alter the value.
*
* @param type the data type of the parameter
* @param value the initial parameter value
*
* @return the command parameter object
*/
public DBCmdParam addParam(DataType type, Object value)
{
if (cmdParams==null)
cmdParams= new Vector<DBCmdParam>();
// Adds the parameter
DBCmdParam param = new DBCmdParam(this, type, value);
if (cmdParams.add(param)==false)
return null; // unknown error
// Creates a Parameter expression
return param;
}
/**
* Adds an command parameter which will be used in a prepared statement.
* The initial value of the command parameter is null but can be modified using the setValue method.
*
* @param colExpr the column expression for which to create the parameter
* @param value the initial parameter value
*
* @return the command parameter object
*/
public final DBCmdParam addParam(DBColumnExpr colExpr, Object value)
{
return addParam(colExpr.getDataType(), value);
}
/**
* Adds an command parameter which will be used in a prepared statement.
* The initial value of the command parameter is null but can be modified using the setValue method.
*
* @return the command parameter object
*/
public final DBCmdParam addParam(Object value)
{
return addParam(DataType.UNKNOWN, value);
}
/**
* Adds an command parameter which will be used in a prepared statement.
* The initial value of the command parameter is null but can be modified using the setValue method.
*
* @return the command parameter object
*/
public final DBCmdParam addParam()
{
return addParam(DataType.UNKNOWN, null);
}
/**
* Adds a join to the list of join expressions.
*
* @param join the join expression
*/
public void join(DBJoinExpr join)
{
checkDatabase(join);
// check tables
if (join.getLeftTable().equals(join.getRightTable()))
throw new InvalidArgumentException("left|right", join.getLeftTable());
// create list
if (joins == null)
joins = new ArrayList<DBJoinExpr>();
// Create a new join
for (int i = 0; i < joins.size(); i++)
{ // Check whether join exists
DBJoinExpr item = joins.get(i);
if (item.equals(join))
return;
}
joins.add(join);
}
/**
* Adds a cross join for two tables or views
* @param left the left RowSet
* @param right the right RowSet
* @return the join expression
*/
public final DBCrossJoinExpr join(DBRowSet left, DBRowSet right)
{
DBCrossJoinExpr join = new DBCrossJoinExpr(left, right);
join(join);
return join;
}
/**
* Adds a join based on two columns to the list of join expressions.
*
* @param left the left join value
* @param right the right join
* @param joinType type of join ({@link DBJoinType#INNER}, {@link DBJoinType#LEFT}, {@link DBJoinType#RIGHT})
*
* @return the join expression
*/
public final DBColumnJoinExpr join(DBColumnExpr left, DBColumnExpr right, DBJoinType joinType)
{
DBColumnJoinExpr join = new DBColumnJoinExpr(left, right, joinType);
join(join);
return join;
}
/**
* Adds an inner join based on two columns to the list of join expressions.
*
* @param left the left join value
* @param right the right join
*
* @return the join expresion
*/
public final DBColumnJoinExpr join(DBColumnExpr left, DBColumn right)
{
return join(left, right, DBJoinType.INNER);
}
/**
* Adds a join based on a compare expression to the command.
*
* @param rowset table or view to join
* @param cmp the compare expression with wich to join the table
* @param joinType type of join ({@link DBJoinType#INNER}, {@link DBJoinType#LEFT}, {@link DBJoinType#RIGHT})
*
* @return the join expresion
*/
public final DBCompareJoinExpr join(DBRowSet rowset, DBCompareExpr cmp, DBJoinType joinType)
{
DBCompareJoinExpr join = new DBCompareJoinExpr(rowset, cmp, joinType);
join(join);
return join;
}
/**
* Adds an inner join based on a compare expression to the command.
*
* @param rowset table of view which to join
* @param cmp the compare expression with wich to join the table
*
* @return the join expresion
*/
public final DBCompareJoinExpr join(DBRowSet rowset, DBCompareExpr cmp)
{
return join(rowset, cmp, DBJoinType.INNER);
}
/**
* Adds a list of join expressions to the command.
*
* @param joinExprList list of join expressions
*/
public void addJoins(List<DBJoinExpr> joinExprList)
{
if (joins == null)
{
joins = new ArrayList<DBJoinExpr>();
}
this.joins.addAll(joinExprList);
}
/**
* Returns true if the command has a join on the given table or false otherwise.
*
* @param rowset rowset table or view to join
*
* @return true if the command has a join on the given table or false otherwise
*/
public boolean hasJoinOn(DBRowSet rowset)
{
if (joins==null)
return false;
// Examine all joins
for (DBJoinExpr join : joins)
{
if (join.isJoinOn(rowset))
return true;
}
// not found
return false;
}
/**
* Returns true if the command has a constraint on the given table or false otherwise.
*
* @param rowset rowset table or view to join
*
* @return true if the command has a join on the given table or false otherwise
*/
public boolean hasConstraintOn(DBRowSet rowset)
{
if (where==null && having==null)
return false;
// Examine all constraints
int i = 0;
Set<DBColumn> columns = new HashSet<DBColumn>();
for (i = 0; where != null && i < where.size(); i++)
((DBExpr) where.get(i)).addReferencedColumns(columns);
/*
for (i = 0; groupBy != null && i < groupBy.size(); i++)
((DBExpr) groupBy.get(i)).addReferencedColumns(columns);
*/
for (i = 0; having != null && i < having.size(); i++)
((DBExpr) having.get(i)).addReferencedColumns(columns);
// now we have all columns
Iterator<DBColumn> iterator = columns.iterator();
while (iterator.hasNext())
{ // get the table
DBColumn col = iterator.next();
DBRowSet table = col.getRowSet();
if (table.equals(rowset))
return true;
}
// not found
return false;
}
/**
* Returns true if the command has a join on the given column or false otherwise.
*
* @param column the column to test
*
* @return true if the command has a join on the given column or false otherwise
*/
public boolean hasJoinOn(DBColumn column)
{
if (joins==null)
return false;
// Examine all joins
for (DBJoinExpr join : joins)
{
if (join.isJoinOn(column))
return true;
}
// not found
return false;
}
/**
* removes all joins to a given table or view
*
* @param rowset the table or view for which to remove all joins
*
* @return true if any joins have been removed or false otherwise
*/
public boolean removeJoinsOn(DBRowSet rowset)
{
if (joins==null)
return false;
// Examine all joins
int size = joins.size();
for (int i=size-1; i>=0; i--)
{
if (joins.get(i).isJoinOn(rowset))
joins.remove(i);
}
return (size!=joins.size());
}
/**
* removes all joins to a given column
*
* @param column the column for which to remove all joins
*
* @return true if any joins have been removed or false otherwise
*/
public boolean removeJoinsOn(DBColumn column)
{
if (joins==null)
return false;
// Examine all joins
int size = joins.size();
for (int i=size-1; i>=0; i--)
{
if (joins.get(i).isJoinOn(column))
joins.remove(i);
}
return (size!=joins.size());
}
/**
* Adds a constraint to the where phrase of the sql statement
* If another restriction already exists for the same column it will be replaced.
*
* @param expr the DBCompareExpr object
*/
public void where(DBCompareExpr expr)
{
checkDatabase(expr);
if (where == null)
where = new ArrayList<DBCompareExpr>();
setConstraint(where, expr);
}
/**
* Adds a list of constraints to the where phrase of the sql statement
* If another restriction already exists for the same column it will be replaced.
*
* @param expr the DBCompareExpr object
*/
public final void where(DBCompareExpr... exprs)
{
for (int i=0; i<exprs.length; i++)
where(exprs[i]);
}
/**
* Returns true if the command has constraints or false if not.
*
* @return true if constraints have been set on the command
*/
public boolean hasWhereConstraints()
{
return (where!=null && where.size()>0);
}
/**
* Returns a copy of the defined where clauses.
*
* @return vector of where clauses
*/
public List<DBCompareExpr> getWhereConstraints()
{
return (this.where!=null ? Collections.unmodifiableList(this.where) : null);
}
/**
* removes a constraint on a particular column from the where clause
* @param col the column expression for which to remove the constraint
*/
public void removeWhereConstraintOn(DBColumnExpr col)
{
if (where == null)
return;
removeConstraintOn(where, col);
}
/**
* Returns a copy of the defined joins.
*
* @return the list of joins
*/
public List<DBJoinExpr> getJoins()
{
return (this.joins!=null ? Collections.unmodifiableList(this.joins) : null);
}
/**
* Adds a list of constraints to the command.
* @param constraints list of constraints
*/
public void addWhereConstraints(List<DBCompareExpr> constraints)
{
// allocate
if (where == null)
where = new ArrayList<DBCompareExpr>();
// add
this.where.addAll(constraints);
}
/**
* adds a constraint to the having clause.
* @param expr the DBCompareExpr object
*/
public void having(DBCompareExpr expr)
{
checkDatabase(expr);
if (having == null)
having = new ArrayList<DBCompareExpr>();
setConstraint(having, expr);
}
/**
* Returns true if the command has having-constraints or false if not.
*
* @return true if constraints have been set on the command
*/
public boolean hasHavingConstraints()
{
return (having!=null && having.size()>0);
}
/**
* Returns a copy of the defined having clauses.
*
* @return list of having constraints
*/
public List<DBCompareExpr> getHavingConstraints()
{
return (this.having!=null ? Collections.unmodifiableList(this.having) : null);
}
/**
* removes a constraint on a particular column from the having clause
* @param col the column expression for which to remove the constraint
*/
public void removeHavingConstraintOn(DBColumnExpr col)
{
if (having == null)
return;
removeConstraintOn(having, col);
}
/**
* Adds a list of columns to the group by phrase of an sql statement.
*
* @param exprs vararg of columns by which to group the rows
*/
public void groupBy(DBColumnExpr...exprs)
{
if (groupBy == null)
groupBy = new ArrayList<DBColumnExpr>();
// Add all
for(DBColumnExpr expr : exprs)
{
checkDatabase(expr);
if (expr.isAggregate()==false && groupBy.contains(expr)==false)
groupBy.add(expr);
}
}
/**
* Adds a collection of columns to the group by phrase of an sql statement.
*
* @param columns the column expressions to add
*/
public final void groupBy(Collection<? extends DBColumnExpr> columns)
{
for (DBColumnExpr expr : columns)
{
groupBy(expr);
}
}
public boolean hasSelectExpr()
{
return (select!=null && select.size()>0);
}
@Override
public synchronized void getSelect(StringBuilder buf)
{
resetParamUsage();
if (select == null)
throw new ObjectNotValidException(this); // invalid!
// Prepares statement
addSelect(buf);
// From clause
addFrom(buf);
// Add Where
addWhere(buf);
// Add Grouping
addGrouping(buf);
// Add Order
addOrder(buf);
}
/**
* Returns a array of all select DBColumnExpr for this command
*
* @return a array of all DBColumnExpr objects or <code>null</code> if there are no selects
*/
@Override
public DBColumnExpr[] getSelectExprList()
{
int count = (select != null) ? select.size() : 0;
if (count < 1)
return null;
// The List
DBColumnExpr[] exprList = new DBColumnExpr[count];
for (int i = 0; i < count; i++)
exprList[i] = select.get(i);
// The expression List
return exprList;
}
/**
* Clears the select distinct option.
*/
public void clearSelectDistinct()
{
this.selectDistinct = false;
}
/**
* Clears the list of selected columns.
*/
public void clearSelect()
{
select = null;
}
/**
* Clears the list of set expressions.
*/
public void clearSet()
{
set = null;
cmdParams = null;
}
/**
* Clears the list of join expressions.
*/
public void clearJoin()
{
joins = null;
}
/**
* Clears the list of where constraints.
*/
public void clearWhere()
{
removeAllCommandParams(where);
where = null;
}
/**
* Clears the list of having constraints.
*/
public void clearHaving()
{
removeAllCommandParams(having);
having = null;
}
/**
* Clears the list of group by constraints.
*/
public void clearGroupBy()
{
groupBy = null;
}
/**
* Clears the entire command object.
*/
public void clear()
{
cmdParams = null;
clearSelectDistinct();
clearSelect();
clearSet();
clearJoin();
clearWhere();
clearHaving();
clearGroupBy();
clearOrderBy();
clearLimit();
resetParamUsage();
}
/**
* adds a constraint to the 'where' or 'having' collections
* @param list the 'where' or 'having' list
* @param expr the DBCompareExpr object
*/
protected void setConstraint(List<DBCompareExpr> list, DBCompareExpr expr)
{ // adds a comparison to the where or having list
for (int i = 0; i < list.size(); i++)
{ // check expression
DBCompareExpr other = list.get(i);
if (expr.isMutuallyExclusive(other)==false)
continue;
// Check if we replace a DBCommandParam
if (other instanceof DBCompareColExpr)
removeCommandParam((DBCompareColExpr)other);
// columns match
list.set(i, expr);
return;
}
// add expression
list.add(expr);
}
/**
* removes a constraint on a particular column to the 'where' or 'having' collections
* @param list the 'where' or 'having' list
* @param col the column expression for which to remove the constraint
*/
protected void removeConstraintOn(List<DBCompareExpr> list, DBColumnExpr col)
{
if (list == null)
return;
for(DBCompareExpr cmp : list)
{ // Check whether it is a compare column expr.
if (!(cmp instanceof DBCompareColExpr))
continue;
// Compare columns
DBColumnExpr c = ((DBCompareColExpr)cmp).getColumnExpr();
DBColumn udc = c.getUpdateColumn();
if (c.equals(col) || (udc!=null && udc.equals(col.getUpdateColumn())))
{ // Check if we replace a DBCommandParam
removeCommandParam((DBCompareColExpr)cmp);
// remove the constraint
list.remove(cmp);
return;
}
}
}
/**
* Gets a list of all tables referenced by the query.
*
* @return list of all rowsets (tables or views) used by the query
*/
protected List<DBRowSet> getRowSetList()
{
// Check all tables
int i = 0;
Set<DBColumn> columns = new HashSet<DBColumn>();
for (i = 0; select != null && i < select.size(); i++)
((DBExpr) select.get(i)).addReferencedColumns(columns);
for (i = 0; joins != null && i < joins.size(); i++)
((DBExpr) joins.get(i)).addReferencedColumns(columns);
for (i = 0; where != null && i < where.size(); i++)
((DBExpr) where.get(i)).addReferencedColumns(columns);
for (i = 0; groupBy != null && i < groupBy.size(); i++)
((DBExpr) groupBy.get(i)).addReferencedColumns(columns);
for (i = 0; having != null && i < having.size(); i++)
((DBExpr) having.get(i)).addReferencedColumns(columns);
for (i = 0; orderBy != null && i < orderBy.size(); i++)
((DBExpr) orderBy.get(i)).addReferencedColumns(columns);
// now we have all columns
List<DBRowSet> tables = new ArrayList<DBRowSet>();
Iterator<DBColumn> iterator = columns.iterator();
while (iterator.hasNext())
{ // get the table
DBColumn col = iterator.next();
DBRowSet table = col.getRowSet();
if (table == cmdQuery)
{ // Recursion
log.error("Recursive Column Selection in Command!");
continue;
}
if (tables.contains(table) == false && table != null)
{ // Add table
tables.add(table);
}
}
return tables;
}
/**
* Adds Columns
*/
@Override
public void addReferencedColumns(Set<DBColumn> list)
{
// nothing to do!
return;
}
/**
* Returns an array of parameter values for a prepared statement.
* To ensure that all values are in the order of their occurrence, getSelect() should be called first.
* @return an array of parameter values for a prepared statement
*/
@Override
public Object[] getParamValues()
{
if (cmdParams==null || cmdParams.size()==0)
return null;
// Check whether all parameters have been used
if (paramUsageCount>0 && paramUsageCount!=cmdParams.size())
log.warn("DBCommand parameter count ("+String.valueOf(cmdParams.size())
+ ") does not match parameter use count ("+String.valueOf(paramUsageCount)+")");
// Create result array
Object[] values = new Object[cmdParams.size()];
for (int i=0; i<values.length; i++)
values[i]=cmdParams.get(i).getValue();
// values
return values;
}
/**
* Creates the update SQL-Command.
*
* @return the update SQL-Command
*/
public synchronized String getUpdate()
{
resetParamUsage();
if (set == null)
return null;
StringBuilder buf = new StringBuilder("UPDATE ");
DBRowSet table = set.get(0).getTable();
if (joins!=null && !joins.isEmpty())
{ // Join Update
buf.append( table.getAlias() );
long context = CTX_DEFAULT;
// Set Expressions
buf.append("\r\nSET ");
addListExpr(buf, set, context, ", ");
// From clause
addFrom(buf);
// Add Where
addWhere(buf, context);
}
else
{ // Simple Statement
table.addSQL(buf, CTX_FULLNAME);
long context = CTX_NAME | CTX_VALUE;
// Set Expressions
buf.append("\r\nSET ");
addListExpr(buf, set, context, ", ");
// Add Where
addWhere(buf, context);
}
// done
return buf.toString();
}
/**
* Creates the insert SQL-Command.
*
* @return the insert SQL-Command
*/
// get Insert
public synchronized String getInsert()
{
resetParamUsage();
if (set==null || set.get(0)==null)
return null;
StringBuilder buf = new StringBuilder("INSERT INTO ");
// addTableExpr(buf, CTX_NAME);
DBRowSet table = set.get(0).getTable();
table.addSQL(buf, CTX_FULLNAME);
// Set Expressions
buf.append("( ");
// Set Expressions
ArrayList<DBCompareColExpr> compexpr = null;
if (where!=null && !where.isEmpty())
{ // Convert ColumnExpression List to Column List
compexpr = new ArrayList<DBCompareColExpr>(where.size());
for (DBCompareExpr expr : where)
{ if (expr instanceof DBCompareColExpr)
{ DBColumn column = ((DBCompareColExpr)expr).getColumnExpr().getUpdateColumn();
if (column!=null && hasSetExprOn(column)==false)
compexpr.add((DBCompareColExpr)expr);
}
}
// Add Column Names from where clause
if (compexpr.size()>0)
{
// add List
addListExpr(buf, compexpr, CTX_NAME, ", ");
// add separator
if (set != null)
buf.append(", ");
}
else
{ // No columns to set
compexpr = null;
}
}
if (set != null)
addListExpr(buf, set, CTX_NAME, ", ");
// Values
buf.append(") VALUES ( ");
if (compexpr != null)
addListExpr(buf, compexpr, CTX_VALUE, ", ");
if (compexpr != null && set != null)
buf.append(", ");
if (set != null)
addListExpr(buf, set, CTX_VALUE, ", ");
// End
buf.append(")");
return buf.toString();
}
/**
* Creates the delete SQL-Command.
*
* @param table the table object
*
* @return the delete SQL-Command
*/
public synchronized String getDelete(DBTable table)
{
resetParamUsage();
StringBuilder buf = new StringBuilder("DELETE FROM ");
table.addSQL(buf, CTX_FULLNAME);
// Set Expressions
if (where!=null && !where.isEmpty())
{ // add where condition
buf.append("\r\nWHERE ");
addListExpr(buf, where, CTX_NAME|CTX_VALUE, " AND ");
}
return buf.toString();
}
// ------- Select Statement Parts -------
protected void addSelect(StringBuilder buf)
{
// Prepares statement
buf.append("SELECT ");
if (selectDistinct)
buf.append("DISTINCT ");
// Add Select Expressions
addListExpr(buf, select, CTX_ALL, ", ");
}
protected void addFrom(StringBuilder buf)
{
int originalLength = buf.length();
buf.append("\r\nFROM ");
// Join
boolean sep = false;
List<DBRowSet> tables = getRowSetList();
if (joins!=null && joins.size()>0)
{ // Join
List<DBRowSet> joinTables = new ArrayList<DBRowSet>();
for (int i=0; i<joins.size(); i++)
{ // append join
long context;
DBJoinExpr join = joins.get(i);
if (i<1)
{ // Add Join Tables
joinTables.add(join.getLeftTable());
joinTables.add(join.getRightTable());
// Remove from List
tables.remove(join.getLeftTable());
tables.remove(join.getRightTable());
// Context
context = CTX_NAME|CTX_VALUE;
}
else
{ // Extend the join
if ( joinTables.contains(join.getRightTable()))
join.reverse();
// Add Right Table
joinTables.add(join.getRightTable());
tables .remove(join.getRightTable());
// Context
context = CTX_VALUE;
buf.append( "\t" );
}
join.addSQL(buf, context);
// add CRLF
if( i!=joins.size()-1 )
buf.append("\r\n");
}
sep = true;
}
for (int i=0; i<tables.size(); i++)
{
if (sep) buf.append(", ");
DBRowSet t = tables.get(i);
t.addSQL(buf, CTX_DEFAULT|CTX_ALIAS);
sep = true;
}
if (sep==false)
{ // add pseudo table or omitt from
String pseudoTable = db.getDriver().getSQLPhrase(DBDatabaseDriver.SQL_PSEUDO_TABLE);
if (StringUtils.isNotEmpty(pseudoTable))
{ // add pseudo table
buf.append(pseudoTable);
}
else
{ // remove from
buf.setLength(originalLength);
}
}
}
protected void addWhere(StringBuilder buf, long context)
{
if (where!=null && !where.isEmpty())
{
buf.append("\r\nWHERE ");
// add where expression
addListExpr(buf, where, context, " AND ");
}
}
protected final void addWhere(StringBuilder buf)
{
addWhere(buf, CTX_DEFAULT);
}
protected void addGrouping(StringBuilder buf)
{
if (groupBy!=null && !groupBy.isEmpty())
{ // Having
buf.append("\r\nGROUP BY ");
addListExpr(buf, groupBy, CTX_DEFAULT, ", ");
}
if (having!=null && !having.isEmpty())
{ // Having
buf.append("\r\nHAVING ");
addListExpr(buf, having, CTX_DEFAULT, " AND ");
}
}
protected void addOrder(StringBuilder buf)
{
if (orderBy!=null && !orderBy.isEmpty())
{ // order By
buf.append("\r\nORDER BY ");
addListExpr(buf, orderBy, CTX_DEFAULT, ", ");
}
}
protected void checkDatabase(DBExpr expr)
{
if (getDatabase().equals(expr.getDatabase()))
return;
// not the same database
throw new DatabaseMismatchException(this, expr);
}
}