Open Source Repository

Home /hibernate/hibernate-3.2.6.ga | Repository Home



org/hibernate/sql/SimpleSelect.java
//$Id: SimpleSelect.java 7627 2005-07-24 06:53:06Z oneovthafew $
package org.hibernate.sql;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.hibernate.LockMode;
import org.hibernate.dialect.Dialect;

/**
 * An SQL <tt>SELECT</tt> statement with no table joins
 *
 @author Gavin King
 */
public class SimpleSelect {

  public SimpleSelect(Dialect dialect) {
    this.dialect = dialect;
  }

  //private static final Alias DEFAULT_ALIAS = new Alias(10, null);

  private String tableName;
  private String orderBy;
  private Dialect dialect;
  private LockMode lockMode = LockMode.READ;
  private String comment;

  private List columns = new ArrayList();
  private Map aliases = new HashMap();
  private List whereTokens = new ArrayList();

  public SimpleSelect addColumns(String[] columnNames, String[] columnAliases) {
    for int i=0; i<columnNames.length; i++ ) {
      if columnNames[i]!=null  ) {
        addColumncolumnNames[i], columnAliases[i] );
      }
    }
    return this;
  }

  public SimpleSelect addColumns(String[] columns, String[] aliases, boolean[] ignore) {
    for int i=0; i<ignore.length; i++ ) {
      if !ignore[i&& columns[i]!=null ) {
        addColumncolumns[i], aliases[i] );
      }
    }
    return this;
  }

  public SimpleSelect addColumns(String[] columnNames) {
    for int i=0; i<columnNames.length; i++ ) {
      if columnNames[i]!=null addColumncolumnNames[i] );
    }
    return this;
  }
  public SimpleSelect addColumn(String columnName) {
    columns.add(columnName);
    //aliases.put( columnName, DEFAULT_ALIAS.toAliasString(columnName) );
    return this;
  }

  public SimpleSelect addColumn(String columnName, String alias) {
    columns.add(columnName);
    aliases.put(columnName, alias);
    return this;
  }

  public SimpleSelect setTableName(String tableName) {
    this.tableName = tableName;
    return this;
  }

  public SimpleSelect setLockMode(LockMode lockMode) {
    this.lockMode = lockMode;
    return this;
  }

  public SimpleSelect addWhereToken(String token) {
    whereTokens.add(token);
    return this;
  }
  
  private void and() {
    if whereTokens.size()>) {
      whereTokens.add("and");
    }
  }

  public SimpleSelect addCondition(String lhs, String op, String rhs) {
    and();
    whereTokens.addlhs + ' ' + op + ' ' + rhs );
    return this;
  }

  public SimpleSelect addCondition(String lhs, String condition) {
    and();
    whereTokens.addlhs + ' ' + condition );
    return this;
  }

  public SimpleSelect addCondition(String[] lhs, String op, String[] rhs) {
    for int i=0; i<lhs.length; i++ ) {
      addConditionlhs[i], op, rhs[i] );
    }
    return this;
  }

  public SimpleSelect addCondition(String[] lhs, String condition) {
    for int i=0; i<lhs.length; i++ ) {
      if lhs[i]!=null addConditionlhs[i], condition );
    }
    return this;
  }

  public String toStatementString() {
    StringBuffer buf = new StringBuffer
        columns.size()*10 
        tableName.length() 
        whereTokens.size() 10 
        10 
      );
    
    if comment!=null ) {
      buf.append("/* ").append(comment).append(" */ ");
    }
    
    buf.append("select ");
    Set uniqueColumns = new HashSet();
    Iterator iter = columns.iterator();
    boolean appendComma = false;
    while iter.hasNext() ) {
      String col = (Stringiter.next();
      String alias = (Stringaliases.get(col);
      if uniqueColumns.add(alias==null ? col : alias) ) {
        if (appendCommabuf.append(", ");
        buf.append(col);
        if alias!=null && !alias.equals(col) ) {
          buf.append(" as ")
            .append(alias);
        }
        appendComma = true;
      }
    }
    
    buf.append(" from ")
      .appenddialect.appendLockHint(lockMode, tableName) );
    
    if whereTokens.size() ) {
      buf.append(" where ")
        .appendtoWhereClause() );
    }
    
    if (orderBy!=nullbuf.append(orderBy);
    
    if (lockMode!=null) {
      buf.appenddialect.getForUpdateString(lockMode) );
    }

    return dialect.transformSelectStringbuf.toString() );
  }

  public String toWhereClause() {
    StringBuffer buf = new StringBufferwhereTokens.size() );
    Iterator iter = whereTokens.iterator();
    while iter.hasNext() ) {
      buf.appenditer.next() );
      if iter.hasNext() ) buf.append(' ');
    }
    return buf.toString();
  }

  public SimpleSelect setOrderBy(String orderBy) {
    this.orderBy = orderBy;
    return this;
  }

  public SimpleSelect setComment(String comment) {
    this.comment = comment;
    return this;
  }

}