package org.hibernate.id.insert;
import java.io.Serializable;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import org.hibernate.engine.spi.SharedSessionContractImplementor;
import org.hibernate.id.PostInsertIdentityPersister;
import org.hibernate.pretty.MessageHelper;
public abstract class AbstractSelectingDelegate implements InsertGeneratedIdentifierDelegate {
private final PostInsertIdentityPersister persister;
protected AbstractSelectingDelegate(PostInsertIdentityPersister persister) {
this.persister = persister;
}
@Override
public final Serializable performInsert(
String insertSQL,
SharedSessionContractImplementor session,
Binder binder) {
try {
PreparedStatement insert = session
.getJdbcCoordinator()
.getStatementPreparer()
.prepareStatement( insertSQL, PreparedStatement.NO_GENERATED_KEYS );
try {
binder.bindValues( insert );
session.getJdbcCoordinator().getResultSetReturn().executeUpdate( insert );
}
finally {
session.getJdbcCoordinator().getLogicalConnection().getResourceRegistry().release( insert );
session.getJdbcCoordinator().afterStatementExecution();
}
}
catch (SQLException sqle) {
throw session.getJdbcServices().getSqlExceptionHelper().convert(
sqle,
"could not insert: " + MessageHelper.infoString( persister ),
insertSQL
);
}
final String selectSQL = getSelectSQL();
try {
PreparedStatement idSelect = session
.getJdbcCoordinator()
.getStatementPreparer()
.prepareStatement( selectSQL, false );
try {
bindParameters( session, idSelect, binder.getEntity() );
ResultSet rs = session.getJdbcCoordinator().getResultSetReturn().extract( idSelect );
try {
return getResult( session, rs, binder.getEntity() );
}
finally {
session.getJdbcCoordinator().getLogicalConnection().getResourceRegistry().release( rs, idSelect );
}
}
finally {
session.getJdbcCoordinator().getLogicalConnection().getResourceRegistry().release( idSelect );
session.getJdbcCoordinator().afterStatementExecution();
}
}
catch (SQLException sqle) {
throw session.getJdbcServices().getSqlExceptionHelper().convert(
sqle,
"could not retrieve generated id after insert: " + MessageHelper.infoString( persister ),
insertSQL
);
}
}
protected abstract String getSelectSQL();
protected void bindParameters(
SharedSessionContractImplementor session,
PreparedStatement ps,
Object entity) throws SQLException {
}
protected abstract Serializable getResult(
SharedSessionContractImplementor session,
ResultSet rs,
Object entity) throws SQLException;
}