Open Source Repository

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



org/hibernate/sql/InsertSelect.java
// $Id: InsertSelect.java 7057 2005-06-07 20:06:10Z steveebersole $
package org.hibernate.sql;

import org.hibernate.dialect.Dialect;
import org.hibernate.HibernateException;

import java.util.List;
import java.util.ArrayList;
import java.util.Iterator;

/**
 * Implementation of InsertSelect.
 *
 @author Steve Ebersole
 */
public class InsertSelect {

  private Dialect dialect;
  private String tableName;
  private String comment;
  private List columnNames = new ArrayList();
  private Select select;

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

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

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

  public InsertSelect addColumn(String columnName) {
    columnNames.addcolumnName );
    return this;
  }

  public InsertSelect addColumns(String[] columnNames) {
    for int i = 0; i < columnNames.length; i++ ) {
      this.columnNames.addcolumnNames[i] );
    }
    return this;
  }

  public InsertSelect setSelect(Select select) {
    this.select = select;
    return this;
  }

  public String toStatementString() {
    if tableName == null throw new HibernateException"no table name defined for insert-select" );
    if select == null throw new HibernateException"no select defined for insert-select" );

    StringBuffer buf = new StringBuffer( (columnNames.size() 15+ tableName.length() 10 );
    if comment!=null ) {
      buf.append"/* " ).appendcomment ).append" */ " );
    }
    buf.append"insert into " ).appendtableName );
    if !columnNames.isEmpty() ) {
      buf.append" (" );
      Iterator itr = columnNames.iterator();
      while itr.hasNext() ) {
        buf.appenditr.next() );
        if itr.hasNext() ) {
          buf.append", " );
        }
      }
      buf.append")" );
    }
    buf.append' ' ).appendselect.toStatementString() );
    return buf.toString();
  }
}