You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
227 lines
5.8 KiB
227 lines
5.8 KiB
/* |
|
* Hibernate, Relational Persistence for Idiomatic Java |
|
* |
|
* License: GNU Lesser General Public License (LGPL), version 2.1 or later. |
|
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>. |
|
*/ |
|
package com.fr.third.org.hibernate.sql; |
|
import java.util.Iterator; |
|
import java.util.LinkedHashMap; |
|
import java.util.Map; |
|
|
|
import com.fr.third.org.hibernate.dialect.Dialect; |
|
import com.fr.third.org.hibernate.type.LiteralType; |
|
|
|
/** |
|
* An SQL <tt>UPDATE</tt> statement |
|
* |
|
* @author Gavin King |
|
*/ |
|
public class Update { |
|
|
|
private String tableName; |
|
private String versionColumnName; |
|
private String where; |
|
private String assignments; |
|
private String comment; |
|
|
|
private Map primaryKeyColumns = new LinkedHashMap(); |
|
private Map columns = new LinkedHashMap(); |
|
private Map whereColumns = new LinkedHashMap(); |
|
|
|
private Dialect dialect; |
|
|
|
public Update(Dialect dialect) { |
|
this.dialect = dialect; |
|
} |
|
|
|
public String getTableName() { |
|
return tableName; |
|
} |
|
|
|
public Update appendAssignmentFragment(String fragment) { |
|
if ( assignments == null ) { |
|
assignments = fragment; |
|
} |
|
else { |
|
assignments += ", " + fragment; |
|
} |
|
return this; |
|
} |
|
|
|
public Update setTableName(String tableName) { |
|
this.tableName = tableName; |
|
return this; |
|
} |
|
|
|
public Update setPrimaryKeyColumnNames(String[] columnNames) { |
|
this.primaryKeyColumns.clear(); |
|
addPrimaryKeyColumns(columnNames); |
|
return this; |
|
} |
|
|
|
public Update addPrimaryKeyColumns(String[] columnNames) { |
|
for ( String columnName : columnNames ) { |
|
addPrimaryKeyColumn( columnName, "?" ); |
|
} |
|
return this; |
|
} |
|
|
|
public Update addPrimaryKeyColumns(String[] columnNames, boolean[] includeColumns, String[] valueExpressions) { |
|
for ( int i=0; i<columnNames.length; i++ ) { |
|
if( includeColumns[i] ) { |
|
addPrimaryKeyColumn( columnNames[i], valueExpressions[i] ); |
|
} |
|
} |
|
return this; |
|
} |
|
|
|
public Update addPrimaryKeyColumns(String[] columnNames, String[] valueExpressions) { |
|
for ( int i=0; i<columnNames.length; i++ ) { |
|
addPrimaryKeyColumn( columnNames[i], valueExpressions[i] ); |
|
} |
|
return this; |
|
} |
|
|
|
public Update addPrimaryKeyColumn(String columnName, String valueExpression) { |
|
this.primaryKeyColumns.put(columnName, valueExpression); |
|
return this; |
|
} |
|
|
|
public Update setVersionColumnName(String versionColumnName) { |
|
this.versionColumnName = versionColumnName; |
|
return this; |
|
} |
|
|
|
|
|
public Update setComment(String comment) { |
|
this.comment = comment; |
|
return this; |
|
} |
|
|
|
public Update addColumns(String[] columnNames) { |
|
for ( String columnName : columnNames ) { |
|
addColumn( columnName ); |
|
} |
|
return this; |
|
} |
|
|
|
public Update addColumns(String[] columnNames, boolean[] updateable, String[] valueExpressions) { |
|
for ( int i=0; i<columnNames.length; i++ ) { |
|
if ( updateable[i] ) { |
|
addColumn( columnNames[i], valueExpressions[i] ); |
|
} |
|
} |
|
return this; |
|
} |
|
|
|
public Update addColumns(String[] columnNames, String valueExpression) { |
|
for ( String columnName : columnNames ) { |
|
addColumn( columnName, valueExpression ); |
|
} |
|
return this; |
|
} |
|
|
|
public Update addColumn(String columnName) { |
|
return addColumn(columnName, "?"); |
|
} |
|
|
|
public Update addColumn(String columnName, String valueExpression) { |
|
columns.put(columnName, valueExpression); |
|
return this; |
|
} |
|
|
|
public Update addColumn(String columnName, Object value, LiteralType type) throws Exception { |
|
return addColumn( columnName, type.objectToSQLString(value, dialect) ); |
|
} |
|
|
|
public Update addWhereColumns(String[] columnNames) { |
|
for ( String columnName : columnNames ) { |
|
addWhereColumn( columnName ); |
|
} |
|
return this; |
|
} |
|
|
|
public Update addWhereColumns(String[] columnNames, String valueExpression) { |
|
for ( String columnName : columnNames ) { |
|
addWhereColumn( columnName, valueExpression ); |
|
} |
|
return this; |
|
} |
|
|
|
public Update addWhereColumn(String columnName) { |
|
return addWhereColumn(columnName, "=?"); |
|
} |
|
|
|
public Update addWhereColumn(String columnName, String valueExpression) { |
|
whereColumns.put(columnName, valueExpression); |
|
return this; |
|
} |
|
|
|
public Update setWhere(String where) { |
|
this.where=where; |
|
return this; |
|
} |
|
|
|
public String toStatementString() { |
|
StringBuilder buf = new StringBuilder( (columns.size() * 15) + tableName.length() + 10 ); |
|
if ( comment!=null ) { |
|
buf.append( "/* " ).append( Dialect.escapeComment(comment) ).append( " */ " ); |
|
} |
|
buf.append( "update " ).append( tableName ).append( " set " ); |
|
boolean assignmentsAppended = false; |
|
Iterator iter = columns.entrySet().iterator(); |
|
while ( iter.hasNext() ) { |
|
Map.Entry e = (Map.Entry) iter.next(); |
|
buf.append( e.getKey() ).append( '=' ).append( e.getValue() ); |
|
if ( iter.hasNext() ) { |
|
buf.append( ", " ); |
|
} |
|
assignmentsAppended = true; |
|
} |
|
if ( assignments != null ) { |
|
if ( assignmentsAppended ) { |
|
buf.append( ", " ); |
|
} |
|
buf.append( assignments ); |
|
} |
|
|
|
boolean conditionsAppended = false; |
|
if ( !primaryKeyColumns.isEmpty() || where != null || !whereColumns.isEmpty() || versionColumnName != null ) { |
|
buf.append( " where " ); |
|
} |
|
iter = primaryKeyColumns.entrySet().iterator(); |
|
while ( iter.hasNext() ) { |
|
Map.Entry e = (Map.Entry) iter.next(); |
|
buf.append( e.getKey() ).append( '=' ).append( e.getValue() ); |
|
if ( iter.hasNext() ) { |
|
buf.append( " and " ); |
|
} |
|
conditionsAppended = true; |
|
} |
|
if ( where != null ) { |
|
if ( conditionsAppended ) { |
|
buf.append( " and " ); |
|
} |
|
buf.append( where ); |
|
conditionsAppended = true; |
|
} |
|
iter = whereColumns.entrySet().iterator(); |
|
while ( iter.hasNext() ) { |
|
final Map.Entry e = (Map.Entry) iter.next(); |
|
if ( conditionsAppended ) { |
|
buf.append( " and " ); |
|
} |
|
buf.append( e.getKey() ).append( e.getValue() ); |
|
conditionsAppended = true; |
|
} |
|
if ( versionColumnName != null ) { |
|
if ( conditionsAppended ) { |
|
buf.append( " and " ); |
|
} |
|
buf.append( versionColumnName ).append( "=?" ); |
|
} |
|
|
|
return buf.toString(); |
|
} |
|
}
|
|
|