/*
 * Hibernate, Relational Persistence for Idiomatic Java
 *
 * Copyright (c) 2008, Red Hat Middleware LLC or third-party contributors as
 * indicated by the @author tags or express copyright attribution
 * statements applied by the authors.  All third-party contributions are
 * distributed under license by Red Hat Middleware LLC.
 *
 * This copyrighted material is made available to anyone wishing to use, modify,
 * copy, or redistribute it subject to the terms and conditions of the GNU
 * Lesser General Public License, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
 * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
 * for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this distribution; if not, write to:
 * Free Software Foundation, Inc.
 * 51 Franklin Street, Fifth Floor
 * Boston, MA  02110-1301  USA
 *
 */
package org.hibernate.id;
import java.io.Serializable;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

import org.hibernate.AssertionFailure;
import org.hibernate.HibernateException;
import org.hibernate.dialect.Dialect;
import org.hibernate.engine.spi.SessionImplementor;
import org.hibernate.id.insert.AbstractReturningDelegate;
import org.hibernate.id.insert.AbstractSelectingDelegate;
import org.hibernate.id.insert.IdentifierGeneratingInsert;
import org.hibernate.id.insert.InsertGeneratedIdentifierDelegate;
import org.hibernate.id.insert.InsertSelectIdentityInsert;

A generator for use with ANSI-SQL IDENTITY columns used as the primary key. The IdentityGenerator for autoincrement/identity key generation.

Indicates to the Session that identity (ie. identity/autoincrement column) key generation should be used.
Author:Christoph Sturm
/** * A generator for use with ANSI-SQL IDENTITY columns used as the primary key. * The IdentityGenerator for autoincrement/identity key generation. * <br><br> * Indicates to the <tt>Session</tt> that identity (ie. identity/autoincrement * column) key generation should be used. * * @author Christoph Sturm */
public class IdentityGenerator extends AbstractPostInsertGenerator { public InsertGeneratedIdentifierDelegate getInsertGeneratedIdentifierDelegate( PostInsertIdentityPersister persister, Dialect dialect, boolean isGetGeneratedKeysEnabled) throws HibernateException { if ( isGetGeneratedKeysEnabled ) { return new GetGeneratedKeysDelegate( persister, dialect ); } else if ( dialect.supportsInsertSelectIdentity() ) { return new InsertSelectDelegate( persister, dialect ); } else { return new BasicDelegate( persister, dialect ); } }
Delegate for dealing with IDENTITY columns using JDBC3 getGeneratedKeys
/** * Delegate for dealing with IDENTITY columns using JDBC3 getGeneratedKeys */
public static class GetGeneratedKeysDelegate extends AbstractReturningDelegate implements InsertGeneratedIdentifierDelegate { private final PostInsertIdentityPersister persister; private final Dialect dialect; public GetGeneratedKeysDelegate(PostInsertIdentityPersister persister, Dialect dialect) { super( persister ); this.persister = persister; this.dialect = dialect; } public IdentifierGeneratingInsert prepareIdentifierGeneratingInsert() { IdentifierGeneratingInsert insert = new IdentifierGeneratingInsert( dialect ); insert.addIdentityColumn( persister.getRootTableKeyColumnNames()[0] ); return insert; } protected PreparedStatement prepare(String insertSQL, SessionImplementor session) throws SQLException { return session.getTransactionCoordinator() .getJdbcCoordinator() .getStatementPreparer() .prepareStatement( insertSQL, PreparedStatement.RETURN_GENERATED_KEYS ); } public Serializable executeAndExtract(PreparedStatement insert, SessionImplementor session) throws SQLException { session.getTransactionCoordinator().getJdbcCoordinator().getResultSetReturn().executeUpdate( insert ); ResultSet rs = null; try { rs = insert.getGeneratedKeys(); return IdentifierGeneratorHelper.getGeneratedIdentity( rs, persister.getRootTableKeyColumnNames()[0], persister.getIdentifierType() ); } finally { if ( rs != null ) { session.getTransactionCoordinator().getJdbcCoordinator().release( rs, insert ); } } } }
Delegate for dealing with IDENTITY columns where the dialect supports returning the generated IDENTITY value directly from the insert statement.
/** * Delegate for dealing with IDENTITY columns where the dialect supports returning * the generated IDENTITY value directly from the insert statement. */
public static class InsertSelectDelegate extends AbstractReturningDelegate implements InsertGeneratedIdentifierDelegate { private final PostInsertIdentityPersister persister; private final Dialect dialect; public InsertSelectDelegate(PostInsertIdentityPersister persister, Dialect dialect) { super( persister ); this.persister = persister; this.dialect = dialect; } public IdentifierGeneratingInsert prepareIdentifierGeneratingInsert() { InsertSelectIdentityInsert insert = new InsertSelectIdentityInsert( dialect ); insert.addIdentityColumn( persister.getRootTableKeyColumnNames()[0] ); return insert; } protected PreparedStatement prepare(String insertSQL, SessionImplementor session) throws SQLException { return session.getTransactionCoordinator() .getJdbcCoordinator() .getStatementPreparer() .prepareStatement( insertSQL, PreparedStatement.NO_GENERATED_KEYS ); } public Serializable executeAndExtract(PreparedStatement insert, SessionImplementor session) throws SQLException { ResultSet rs = session.getTransactionCoordinator().getJdbcCoordinator().getResultSetReturn().execute( insert ); try { return IdentifierGeneratorHelper.getGeneratedIdentity( rs, persister.getRootTableKeyColumnNames()[0], persister.getIdentifierType() ); } finally { session.getTransactionCoordinator().getJdbcCoordinator().release( rs, insert ); } } public Serializable determineGeneratedIdentifier(SessionImplementor session, Object entity) { throw new AssertionFailure( "insert statement returns generated value" ); } }
Delegate for dealing with IDENTITY columns where the dialect requires an additional command execution to retrieve the generated IDENTITY value
/** * Delegate for dealing with IDENTITY columns where the dialect requires an * additional command execution to retrieve the generated IDENTITY value */
public static class BasicDelegate extends AbstractSelectingDelegate implements InsertGeneratedIdentifierDelegate { private final PostInsertIdentityPersister persister; private final Dialect dialect; public BasicDelegate(PostInsertIdentityPersister persister, Dialect dialect) { super( persister ); this.persister = persister; this.dialect = dialect; } public IdentifierGeneratingInsert prepareIdentifierGeneratingInsert() { IdentifierGeneratingInsert insert = new IdentifierGeneratingInsert( dialect ); insert.addIdentityColumn( persister.getRootTableKeyColumnNames()[0] ); return insert; } protected String getSelectSQL() { return persister.getIdentitySelectString(); } protected Serializable getResult( SessionImplementor session, ResultSet rs, Object object) throws SQLException { return IdentifierGeneratorHelper.getGeneratedIdentity( rs, persister.getRootTableKeyColumnNames()[0], persister.getIdentifierType() ); } } }