package org.hibernate.sql;
import org.hibernate.dialect.Dialect;
import org.hibernate.internal.util.StringHelper;
public class QueryJoinFragment extends JoinFragment {
private StringBuilder afterFrom = new StringBuilder();
private StringBuilder afterWhere = new StringBuilder();
private Dialect dialect;
private boolean useThetaStyleInnerJoins;
public QueryJoinFragment(Dialect dialect, boolean useThetaStyleInnerJoins) {
this.dialect = dialect;
this.useThetaStyleInnerJoins = useThetaStyleInnerJoins;
}
public void addJoin(String tableName, String alias, String[] fkColumns, String[] pkColumns, JoinType joinType) {
addJoin( tableName, alias, alias, fkColumns, pkColumns, joinType, null );
}
public void addJoin(String tableName, String alias, String[] fkColumns, String[] pkColumns, JoinType joinType, String on) {
addJoin( tableName, alias, alias, fkColumns, pkColumns, joinType, on );
}
public void addJoin(String tableName, String alias, String[][] fkColumns, String[] pkColumns, JoinType joinType) {
addJoin( tableName, alias, alias, fkColumns, pkColumns, joinType, null );
}
public void addJoin(String tableName, String alias, String[][] fkColumns, String[] pkColumns, JoinType joinType, String on) {
addJoin( tableName, alias, alias, fkColumns, pkColumns, joinType, on );
}
private void addJoin(String tableName, String alias, String concreteAlias, String[] fkColumns, String[] pkColumns, JoinType joinType, String on) {
if ( !useThetaStyleInnerJoins || joinType != JoinType.INNER_JOIN ) {
JoinFragment jf = dialect.createOuterJoinFragment();
jf.addJoin( tableName, alias, fkColumns, pkColumns, joinType, on );
addFragment( jf );
}
else {
addCrossJoin( tableName, alias );
addCondition( concreteAlias, fkColumns, pkColumns );
addCondition( on );
}
}
private void addJoin(String tableName, String alias, String concreteAlias, String[][] fkColumns, String[] pkColumns, JoinType joinType, String on) {
if ( !useThetaStyleInnerJoins || joinType != JoinType.INNER_JOIN ) {
JoinFragment jf = dialect.createOuterJoinFragment();
jf.addJoin( tableName, alias, fkColumns, pkColumns, joinType, on );
addFragment( jf );
}
else {
addCrossJoin( tableName, alias );
addCondition( concreteAlias, fkColumns, pkColumns );
addCondition( on );
}
}
public String toFromFragmentString() {
return afterFrom.toString();
}
public String toWhereFragmentString() {
return afterWhere.toString();
}
public void addJoins(String fromFragment, String whereFragment) {
afterFrom.append( fromFragment );
afterWhere.append( whereFragment );
}
public JoinFragment copy() {
QueryJoinFragment copy = new QueryJoinFragment( dialect, useThetaStyleInnerJoins );
copy.afterFrom = new StringBuilder( afterFrom.toString() );
copy.afterWhere = new StringBuilder( afterWhere.toString() );
return copy;
}
public void addCondition(String alias, String[] columns, String condition) {
for ( int i = 0; i < columns.length; i++ ) {
afterWhere.append( " and " )
.append( alias )
.append( '.' )
.append( columns[i] )
.append( condition );
}
}
public void addCrossJoin(String tableName, String alias) {
afterFrom.append( ", " )
.append( tableName )
.append( ' ' )
.append( alias );
}
public void addCondition(String alias, String[] fkColumns, String[] pkColumns) {
for ( int j = 0; j < fkColumns.length; j++ ) {
afterWhere.append( " and " )
.append( fkColumns[j] )
.append( '=' )
.append( alias )
.append( '.' )
.append( pkColumns[j] );
}
}
public void addCondition(String alias, String[][] fkColumns, String[] pkColumns) {
afterWhere.append( " and " );
if ( fkColumns.length > 1 ) {
afterWhere.append( "(" );
}
for ( int i = 0; i < fkColumns.length; i++ ) {
for ( int j = 0; j < fkColumns[i].length; j++ ) {
afterWhere.append( fkColumns[i][j] )
.append( '=' )
.append( alias )
.append( '.' )
.append( pkColumns[j] );
if ( j < fkColumns[i].length - 1 ) {
afterWhere.append( " and " );
}
}
if ( i < fkColumns.length - 1 ) {
afterWhere.append( " or " );
}
}
if ( fkColumns.length > 1 ) {
afterWhere.append( ")" );
}
}
public boolean addCondition(String condition) {
if (
!StringHelper.isEmpty( condition ) &&
afterFrom.toString().indexOf( condition.trim() ) < 0 &&
afterWhere.toString().indexOf( condition.trim() ) < 0
) {
if ( !condition.startsWith( " and " ) ) {
afterWhere.append( " and " );
}
afterWhere.append( condition );
return true;
}
else {
return false;
}
}
public void addFromFragmentString(String fromFragmentString) {
afterFrom.append( fromFragmentString );
}
public void clearWherePart() {
afterWhere.setLength( 0 );
}
}