Implement on the fly callable named native query to stored procedure translation

This commit is contained in:
Christian Beikov 2022-02-16 14:21:47 +01:00
parent 439788198f
commit c02ce35aa0
4 changed files with 316 additions and 8 deletions

View File

@ -7,8 +7,10 @@
package org.hibernate.boot.model.source.internal.hbm;
import java.util.Locale;
import jakarta.xml.bind.JAXBElement;
import org.hibernate.boot.MappingException;
import org.hibernate.boot.jaxb.hbm.spi.JaxbHbmNamedNativeQueryType;
import org.hibernate.boot.jaxb.hbm.spi.JaxbHbmNamedQueryType;
import org.hibernate.boot.jaxb.hbm.spi.JaxbHbmNativeQueryCollectionLoadReturnType;
@ -20,6 +22,8 @@ import org.hibernate.boot.jaxb.hbm.spi.JaxbHbmSynchronizeType;
import org.hibernate.boot.query.ImplicitHbmResultSetMappingDescriptorBuilder;
import org.hibernate.boot.query.NamedHqlQueryDefinition;
import org.hibernate.boot.query.NamedNativeQueryDefinitionBuilder;
import org.hibernate.boot.query.NamedProcedureCallDefinition;
import org.hibernate.cfg.annotations.QueryBinder;
import org.hibernate.internal.log.DeprecationLogger;
import org.hibernate.internal.util.StringHelper;
@ -160,7 +164,33 @@ public class NamedQueryBinder {
builder.setResultSetMappingName( implicitResultSetMappingBuilder.getRegistrationName() );
}
context.getMetadataCollector().addNamedNativeQuery( builder.build() );
if ( namedQueryBinding.isCallable() ) {
final NamedProcedureCallDefinition definition = QueryBinder.createStoredProcedure(
builder, context,
() -> illegalCallSyntax( context, namedQueryBinding, builder.getSqlString() )
);
context.getMetadataCollector().addNamedProcedureCallDefinition( definition );
DeprecationLogger.DEPRECATION_LOGGER.warn(
"Marking named native queries as callable is deprecated; use `<named-stored-procedure-query/>` instead."
);
}
else {
context.getMetadataCollector().addNamedNativeQuery( builder.build() );
}
}
private static MappingException illegalCallSyntax(
HbmLocalMetadataBuildingContext context,
JaxbHbmNamedNativeQueryType namedQueryBinding,
String sqlString) {
return new MappingException(
String.format(
"Callable named native query [%s] doesn't use the JDBC call syntax: %s",
namedQueryBinding.getName(),
sqlString
),
context.getOrigin()
);
}
private static boolean processNamedQueryContentItem(

View File

@ -6,21 +6,33 @@
*/
package org.hibernate.cfg.annotations;
import java.util.ArrayList;
import java.util.List;
import java.util.function.Supplier;
import org.hibernate.AnnotationException;
import org.hibernate.AssertionFailure;
import org.hibernate.CacheMode;
import org.hibernate.FlushMode;
import org.hibernate.annotations.CacheModeType;
import org.hibernate.annotations.FlushModeType;
import org.hibernate.annotations.common.annotationfactory.AnnotationDescriptor;
import org.hibernate.annotations.common.annotationfactory.AnnotationFactory;
import org.hibernate.boot.internal.NamedHqlQueryDefinitionImpl;
import org.hibernate.boot.internal.NamedProcedureCallDefinitionImpl;
import org.hibernate.boot.query.NamedHqlQueryDefinition;
import org.hibernate.boot.query.NamedNativeQueryDefinition;
import org.hibernate.boot.query.NamedNativeQueryDefinitionBuilder;
import org.hibernate.boot.query.NamedProcedureCallDefinition;
import org.hibernate.boot.spi.MetadataBuildingContext;
import org.hibernate.cfg.BinderHelper;
import org.hibernate.internal.CoreMessageLogger;
import org.hibernate.internal.log.DeprecationLogger;
import org.hibernate.internal.util.collections.CollectionHelper;
import org.hibernate.jpa.HibernateHints;
import org.hibernate.query.sql.internal.ParameterParser;
import org.hibernate.query.sql.spi.ParameterRecognizer;
import org.hibernate.type.BasicType;
import org.jboss.logging.Logger;
@ -29,8 +41,11 @@ import jakarta.persistence.NamedNativeQuery;
import jakarta.persistence.NamedQueries;
import jakarta.persistence.NamedQuery;
import jakarta.persistence.NamedStoredProcedureQuery;
import jakarta.persistence.ParameterMode;
import jakarta.persistence.QueryHint;
import jakarta.persistence.SqlResultSetMapping;
import jakarta.persistence.SqlResultSetMappings;
import jakarta.persistence.StoredProcedureParameter;
/**
* Query binder
@ -166,16 +181,116 @@ public abstract class QueryBinder {
.setFetchSize( queryAnn.fetchSize() < 0 ? null : queryAnn.fetchSize() )
.setFlushMode( getFlushMode( queryAnn.flushMode() ) )
.setReadOnly( queryAnn.readOnly() )
.setQuerySpaces( CollectionHelper.setOf( queryAnn.querySpaces() ) )
.setComment( BinderHelper.getAnnotationValueStringOrNull( queryAnn.comment() ) );
final NamedNativeQueryDefinition queryDefinition = builder.build();
if ( queryAnn.callable() ) {
final NamedProcedureCallDefinition definition = createStoredProcedure(
builder, context,
() -> illegalCallSyntax(
queryAnn,
queryAnn.query()
)
);
context.getMetadataCollector().addNamedProcedureCallDefinition( definition );
DeprecationLogger.DEPRECATION_LOGGER.warn(
"Marking named native queries as callable is no longer supported; use `@jakarta.persistence.NamedStoredProcedureQuery` instead. Ignoring."
);
}
else {
final NamedNativeQueryDefinition queryDefinition = builder.build();
if ( LOG.isDebugEnabled() ) {
LOG.debugf( "Binding named native query: %s => %s", queryDefinition.getRegistrationName(), queryDefinition.getSqlQueryString() );
if ( LOG.isDebugEnabled() ) {
LOG.debugf(
"Binding named native query: %s => %s",
queryDefinition.getRegistrationName(),
queryDefinition.getSqlQueryString()
);
}
context.getMetadataCollector().addNamedNativeQuery( queryDefinition );
}
context.getMetadataCollector().addNamedNativeQuery( queryDefinition );
}
public static NamedProcedureCallDefinition createStoredProcedure(
NamedNativeQueryDefinitionBuilder builder,
MetadataBuildingContext context,
Supplier<RuntimeException> exceptionProducer) {
List<StoredProcedureParameter> storedProcedureParameters = new ArrayList<>();
List<QueryHint> queryHints = new ArrayList<>();
List<String> parameterNames = new ArrayList<>();
final String sqlString = builder.getSqlString().trim();
if ( !sqlString.startsWith( "{" ) || !sqlString.endsWith( "}" ) ) {
throw exceptionProducer.get();
}
final String procedureName = QueryBinder.parseJdbcCall(
sqlString,
parameterNames,
exceptionProducer
);
AnnotationDescriptor ann = new AnnotationDescriptor( NamedStoredProcedureQuery.class );
ann.setValue( "name", builder.getName() );
ann.setValue( "procedureName", procedureName );
for ( String parameterName : parameterNames ) {
AnnotationDescriptor parameterDescriptor = new AnnotationDescriptor( StoredProcedureParameter.class );
parameterDescriptor.setValue( "name", parameterName );
parameterDescriptor.setValue( "mode", ParameterMode.IN );
final String typeName = builder.getParameterTypes().get( parameterName );
if ( typeName == null ) {
parameterDescriptor.setValue( "type", Object.class );
}
else {
final BasicType<Object> registeredType = context.getBootstrapContext()
.getTypeConfiguration()
.getBasicTypeRegistry()
.getRegisteredType( typeName );
parameterDescriptor.setValue( "type", registeredType.getJavaType() );
}
storedProcedureParameters.add( AnnotationFactory.create( parameterDescriptor ) );
}
ann.setValue(
"parameters",
storedProcedureParameters.toArray( new StoredProcedureParameter[storedProcedureParameters.size()] )
);
if ( builder.getResultSetMappingName() != null ) {
ann.setValue( "resultSetMappings", new String[]{ builder.getResultSetMappingName() } );
}
else {
ann.setValue( "resultSetMappings", new String[0] );
}
if ( builder.getResultSetMappingClassName() != null ) {
ann.setValue(
"resultClasses",
new Class[] {
context.getBootstrapContext()
.getClassLoaderAccess().classForName( builder.getResultSetMappingClassName() )
}
);
}
else {
ann.setValue( "resultClasses", new Class[0] );
}
if ( builder.getQuerySpaces() != null ) {
AnnotationDescriptor hintDescriptor = new AnnotationDescriptor( QueryHint.class );
hintDescriptor.setValue( "name", HibernateHints.HINT_NATIVE_SPACES );
hintDescriptor.setValue( "value", String.join( " ", builder.getQuerySpaces() ) );
queryHints.add( AnnotationFactory.create( hintDescriptor ) );
}
AnnotationDescriptor hintDescriptor2 = new AnnotationDescriptor( QueryHint.class );
hintDescriptor2.setValue( "name", HibernateHints.HINT_CALLABLE_FUNCTION );
hintDescriptor2.setValue( "value", "true" );
queryHints.add( AnnotationFactory.create( hintDescriptor2 ) );
ann.setValue( "hints", queryHints.toArray( new QueryHint[queryHints.size()] ) );
return new NamedProcedureCallDefinitionImpl( AnnotationFactory.create( ann ) );
}
public static void bindQueries(NamedQueries queriesAnn, MetadataBuildingContext context, boolean isDefault) {
@ -349,5 +464,84 @@ public abstract class QueryBinder {
context.getMetadataCollector().addSecondPass( new ResultsetMappingSecondPass( ann, context, isDefault ) );
}
public static String parseJdbcCall(
String sqlString,
List<String> parameterNames,
Supplier<RuntimeException> exceptionProducer) {
String procedureName = null;
int index = skipWhitespace( sqlString, 1 );
// Parse the out param `?=` part
if ( sqlString.charAt( index ) == '?' ) {
index++;
index = skipWhitespace( sqlString, index );
if ( sqlString.charAt( index ) != '=' ) {
throw exceptionProducer.get();
}
index++;
index = skipWhitespace( sqlString, index );
}
// Parse the call keyword
if ( !sqlString.regionMatches( true, index, "call", 0, 4 ) ) {
throw exceptionProducer.get();
}
index += 4;
index = skipWhitespace( sqlString, index );
// Parse the procedure name
final int procedureStart = index;
for ( ; index < sqlString.length(); index++ ) {
final char c = sqlString.charAt( index );
if ( c == '(' || Character.isWhitespace( c ) ) {
procedureName = sqlString.substring( procedureStart, index );
break;
}
}
index = skipWhitespace( sqlString, index );
ParameterParser.parse(
sqlString.substring( index, sqlString.length() - 1 ),
new ParameterRecognizer() {
@Override
public void ordinalParameter(int sourcePosition) {
parameterNames.add( "" );
}
@Override
public void namedParameter(String name, int sourcePosition) {
parameterNames.add( name );
}
@Override
public void jpaPositionalParameter(int label, int sourcePosition) {
parameterNames.add( "" );
}
@Override
public void other(char character) {
}
}
);
return procedureName;
}
private static int skipWhitespace(String sqlString, int i) {
while ( i < sqlString.length() ) {
if ( !Character.isWhitespace( sqlString.charAt( i ) ) ) {
break;
}
i++;
}
return i;
}
private static AnnotationException illegalCallSyntax(
org.hibernate.annotations.NamedNativeQuery queryAnn,
String sqlString) {
return new AnnotationException(
String.format(
"Callable named native query [%s] doesn't use the JDBC call syntax: %s",
queryAnn.name(),
sqlString
)
);
}
}

View File

@ -0,0 +1,48 @@
<!--
~ Hibernate, Relational Persistence for Idiomatic Java
~
~ License: GNU Lesser General Public License (LGPL), version 2.1 or later.
~ See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
-->
<entity-mappings xmlns="http://xmlns.jcp.org/xml/ns/persistence/orm" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://xmlns.jcp.org/xml/ns/persistence/orm http://xmlns.jcp.org/xml/ns/persistence/orm_2_1.xsd"
version="2.1">
<named-stored-procedure-query name="simpleScalar" procedure-name="simpleScalar">
<parameter class="java.lang.Integer" mode="IN" name="number"/>
<result-set-mapping>simpleScalar</result-set-mapping>
<hint name="org.hibernate.callableFunction" value="true"/>
</named-stored-procedure-query>
<named-stored-procedure-query name="paramhandling" procedure-name="testParamHandling">
<parameter class="java.lang.Integer" mode="IN"/>
<parameter class="java.lang.Integer" mode="IN"/>
<result-set-mapping>paramhandling</result-set-mapping>
<hint name="org.hibernate.callableFunction" value="true"/>
</named-stored-procedure-query>
<named-stored-procedure-query name="selectAllEmployments" procedure-name="allEmployments">
<result-set-mapping>selectAllEmployments</result-set-mapping>
<hint name="org.hibernate.callableFunction" value="true"/>
</named-stored-procedure-query>
<sql-result-set-mapping name="simpleScalar">
<column-result name="name" class="java.lang.String"/>
<column-result name="value" class="java.lang.Long"/>
</sql-result-set-mapping>
<sql-result-set-mapping name="paramhandling">
<column-result name="value" class="java.lang.Long"/>
<column-result name="value2" class="java.lang.Long"/>
</sql-result-set-mapping>
<sql-result-set-mapping name="selectAllEmployments">
<entity-result entity-class="org.hibernate.orm.test.sql.hand.Employment">
<field-result name="employee" column="EMPLOYEE"/>
<field-result name="employee" column="EMPLOYEE"/>
<field-result name="employer" column="EMPLOYER"/>
<field-result name="startDate" column="STARTDATE"/>
<field-result name="endDate" column="ENDDATE"/>
<field-result name="regionCode" column="REGIONCODE"/>
<field-result name="employmentId" column="EMPID"/>
<field-result name="salary.value" column="VALUE"/>
<field-result name="salary.currency" column="CURRENCY"/>
</entity-result>
</sql-result-set-mapping>
</entity-mappings>

View File

@ -474,26 +474,62 @@ should be changed to use `@NamedStoredProcedureQuery` instead -
@NamedStoredProcedureQuery(
name = "personAndPhones",
procedureName = "fn_person_and_phones",
resultSetMapping = "personWithPhonesResultMapping",
resultSetMappings = "personWithPhonesResultMapping",
hints = @QueryHint(name = "org.hibernate.callableFunction", value = "true"),
parameters = @StoredProcedureParameter(type = Long.class)
)
```
Callable named native queries in hbm.xml files should be migrated to the orm.xml version.
E.g., the following `<sql-query callable="true">` -
```
<sql-query name="simpleScalar" callable="true">
<return-scalar column="name" type="string"/>
<return-scalar column="`value`" type="long"/>
{ ? = call simpleScalar(:number) }
</sql-query>
...
final List<Object[]> results = entityManager
.createNamedQuery("simpleScalar" )
.setParameter( 1, 1L )
.getResultList();
```
should be changed to use `<named-stored-procedure-query/>` instead -
```xml
<named-stored-procedure-query name="simpleScalar" procedure-name="simpleScalar">
<parameter class="java.lang.Integer" mode="IN" name="number"/>
<result-set-mapping>simpleScalar</result-set-mapping>
<hint name="org.hibernate.callableFunction" value="true"/>
</named-stored-procedure-query>
<sql-result-set-mapping name="simpleScalar">
<column-result name="name" class="java.lang.String"/>
<column-result name="value" class="java.lang.Long"/>
</sql-result-set-mapping>
```
TIP: To ease the migration, `<sql-query callable="true"/>` and `@NamedNativeQuery(callable = true)` queries
will be translated and registered as named stored procedure in 6.0, but future versions will drop this automatic translation.
Either `org.hibernate.procedure.ProcedureCall` or `jakarta.persistence.StoredProcedureQuery`
can be used to execute the named query -
```
// Use StoredProcedureQuery
final List<Object[]> personAndPhones = entityManager
.createNamedStoredProcedureQuery( "personAndPhones" )
.createNamedStoredProcedureQuery( "simpleScalar" )
.setParameter( 1, 1L )
.getResultList();
// Use ProcedureCall
final List<Object[]> personAndPhones = entityManager
.unwrap( Session.class )
.getNamedProcedureCall( "personAndPhones" )
.getNamedProcedureCall( "simpleScalar" )
.setParameter( 1, 1L )
.getResultList();
```