HHH-4443 : generic handling of any Hibernate type for post-insert generated identifiers
git-svn-id: https://svn.jboss.org/repos/hibernate/core/trunk@17822 1b8cb986-b30d-0410-93ca-fae66ebed9b2
This commit is contained in:
parent
cb34e79283
commit
9e76e1b184
|
@ -33,11 +33,13 @@ import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
import org.hibernate.HibernateException;
|
import org.hibernate.HibernateException;
|
||||||
import org.hibernate.type.Type;
|
import org.hibernate.type.Type;
|
||||||
|
import org.hibernate.type.CustomType;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Factory and helper methods for <tt>IdentifierGenerator</tt> framework.
|
* Factory and helper methods for {@link IdentifierGenerator} framework.
|
||||||
*
|
*
|
||||||
* @author Gavin King
|
* @author Gavin King
|
||||||
|
* @author Steve Ebersole
|
||||||
*/
|
*/
|
||||||
public final class IdentifierGeneratorHelper {
|
public final class IdentifierGeneratorHelper {
|
||||||
private static final Logger log = LoggerFactory.getLogger( IdentifierGeneratorHelper.class );
|
private static final Logger log = LoggerFactory.getLogger( IdentifierGeneratorHelper.class );
|
||||||
|
@ -94,6 +96,16 @@ public final class IdentifierGeneratorHelper {
|
||||||
* @throws IdentifierGenerationException Indicates an unknown type.
|
* @throws IdentifierGenerationException Indicates an unknown type.
|
||||||
*/
|
*/
|
||||||
public static Serializable get(ResultSet rs, Type type) throws SQLException, IdentifierGenerationException {
|
public static Serializable get(ResultSet rs, Type type) throws SQLException, IdentifierGenerationException {
|
||||||
|
if ( ResultSetIdentifierConsumer.class.isInstance( type ) ) {
|
||||||
|
return ( ( ResultSetIdentifierConsumer ) type ).consumeIdentifier( rs );
|
||||||
|
}
|
||||||
|
if ( CustomType.class.isInstance( type ) ) {
|
||||||
|
final CustomType customType = (CustomType) type;
|
||||||
|
if ( ResultSetIdentifierConsumer.class.isInstance( customType.getUserType() ) ) {
|
||||||
|
return ( (ResultSetIdentifierConsumer) customType.getUserType() ).consumeIdentifier( rs );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
Class clazz = type.getReturnedClass();
|
Class clazz = type.getReturnedClass();
|
||||||
if ( clazz == Long.class ) {
|
if ( clazz == Long.class ) {
|
||||||
return new Long( rs.getLong( 1 ) );
|
return new Long( rs.getLong( 1 ) );
|
||||||
|
|
|
@ -0,0 +1,47 @@
|
||||||
|
/*
|
||||||
|
* Hibernate, Relational Persistence for Idiomatic Java
|
||||||
|
*
|
||||||
|
* Copyright (c) 2009 by Red Hat Inc and/or its affiliates or by
|
||||||
|
* third-party contributors as indicated by either @author tags or express
|
||||||
|
* copyright attribution statements applied by the authors. All
|
||||||
|
* third-party contributions are distributed under license by Red Hat Inc.
|
||||||
|
*
|
||||||
|
* 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.sql.ResultSet;
|
||||||
|
import java.io.Serializable;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* An optional contract for {@link org.hibernate.type.Type} or
|
||||||
|
* {@link org.hibernate.usertype.UserType} implementations to handle generated
|
||||||
|
* id values any way they see fit as opposed to being limited to the discrete set of
|
||||||
|
* numeric types handled by {@link IdentifierGeneratorHelper}
|
||||||
|
*
|
||||||
|
* @author Steve Ebersole
|
||||||
|
*/
|
||||||
|
public interface ResultSetIdentifierConsumer {
|
||||||
|
/**
|
||||||
|
* Given a result set, consume/extract the necessary values and construct an
|
||||||
|
* appropriate identifier value.
|
||||||
|
*
|
||||||
|
* @param resultSet The result set containing the value(s) to be used in building
|
||||||
|
* the identifier value.
|
||||||
|
* @return The identifier value.
|
||||||
|
*/
|
||||||
|
public Serializable consumeIdentifier(ResultSet resultSet);
|
||||||
|
}
|
|
@ -93,6 +93,10 @@ public class CustomType extends AbstractType implements IdentifierType, Discrimi
|
||||||
customLogging = LoggableUserType.class.isAssignableFrom( userTypeClass );
|
customLogging = LoggableUserType.class.isAssignableFrom( userTypeClass );
|
||||||
}
|
}
|
||||||
|
|
||||||
|
public UserType getUserType() {
|
||||||
|
return userType;
|
||||||
|
}
|
||||||
|
|
||||||
public int[] sqlTypes(Mapping pi) {
|
public int[] sqlTypes(Mapping pi) {
|
||||||
return types;
|
return types;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue