Open Source Repository

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



org/hibernate/id/insert/AbstractSelectingDelegate.java
package org.hibernate.id.insert;

import org.hibernate.exception.JDBCExceptionHelper;
import org.hibernate.pretty.MessageHelper;
import org.hibernate.engine.SessionImplementor;
import org.hibernate.id.PostInsertIdentityPersister;

import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.sql.ResultSet;
import java.io.Serializable;

/**
 * Abstract InsertGeneratedIdentifierDelegate implementation where the
 * underlying strategy requires an subsequent select after the insert
 * to determine the generated identifier.
 *
 @author Steve Ebersole
 */
public abstract class AbstractSelectingDelegate implements InsertGeneratedIdentifierDelegate {
  private final PostInsertIdentityPersister persister;

  protected AbstractSelectingDelegate(PostInsertIdentityPersister persister) {
    this.persister = persister;
  }

  public final Serializable performInsert(String insertSQL, SessionImplementor session, Binder binder) {
    try {
      // prepare and execute the insert
      PreparedStatement insert = session.getBatcher().prepareStatementinsertSQL, false );
      try {
        binder.bindValuesinsert );
        insert.executeUpdate();
      }
      finally {
        session.getBatcher().closeStatementinsert );
      }
    }
    catch SQLException sqle ) {
      throw JDBCExceptionHelper.convert(
          session.getFactory().getSQLExceptionConverter(),
              sqle,
              "could not insert: " + MessageHelper.infoStringpersister ),
              insertSQL
        );
    }

    final String selectSQL = getSelectSQL();

    try {
      //fetch the generated id in a separate query
      PreparedStatement idSelect = session.getBatcher().prepareStatementselectSQL );
      try {
        bindParameterssession, idSelect, binder.getEntity() );
        ResultSet rs = idSelect.executeQuery();
        try {
          return getResultsession, rs, binder.getEntity() );
        }
        finally {
          rs.close();
        }
      }
      finally {
        session.getBatcher().closeStatementidSelect );
      }

    }
    catch SQLException sqle ) {
      throw JDBCExceptionHelper.convert(
          session.getFactory().getSQLExceptionConverter(),
              sqle,
              "could not retrieve generated id after insert: " + MessageHelper.infoStringpersister ),
              insertSQL
      );
    }
  }

  /**
   * Get the SQL statement to be used to retrieve generated key values.
   *
   @return The SQL command string
   */
  protected abstract String getSelectSQL();

  /**
   * Bind any required parameter values into the SQL command {@link #getSelectSQL}.
   *
   @param session The session
   @param ps The prepared {@link #getSelectSQL SQL} command
   @param entity The entity being saved.
   @throws SQLException
   */
  protected void bindParameters(
      SessionImplementor session,
          PreparedStatement ps,
          Object entitythrows SQLException {
  }

  /**
   * Extract the generated key value from the given result set.
   *
   @param session The session
   @param rs The result set containing the generated primay key values.
   @param entity The entity being saved.
   @return The generated identifier
   @throws SQLException
   */
  protected abstract Serializable getResult(
      SessionImplementor session,
          ResultSet rs,
          Object entitythrows SQLException;

}