HHH-12072 - BasicFormatterImpl throws a NPE if native SQL begins with a parentheses

This commit is contained in:
ageery 2017-10-31 16:08:08 -04:00 committed by Vlad Mihalcea
parent f20fe860c9
commit 310eb69b6e
4 changed files with 91 additions and 4 deletions

View File

@ -356,6 +356,10 @@ public class BasicFormatterImpl implements Formatter {
} }
private static boolean isFunctionName(String tok) { private static boolean isFunctionName(String tok) {
if ( tok == null || tok.length() == 0 ) {
return false;
}
final char begin = tok.charAt( 0 ); final char begin = tok.charAt( 0 );
final boolean isIdentifier = Character.isJavaIdentifierStart( begin ) || '"' == begin; final boolean isIdentifier = Character.isJavaIdentifierStart( begin ) || '"' == begin;
return isIdentifier && return isIdentifier &&

View File

@ -13,6 +13,8 @@ import org.junit.Test;
import org.hibernate.engine.jdbc.internal.FormatStyle; import org.hibernate.engine.jdbc.internal.FormatStyle;
import org.hibernate.testing.junit4.BaseUnitTestCase; import org.hibernate.testing.junit4.BaseUnitTestCase;
import org.jboss.logging.Logger;
import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertFalse;
@ -22,6 +24,9 @@ import static org.junit.Assert.assertFalse;
* @author Steve Ebersole * @author Steve Ebersole
*/ */
public class BasicFormatterTest extends BaseUnitTestCase { public class BasicFormatterTest extends BaseUnitTestCase {
private static final Logger log = Logger.getLogger( BasicFormatterTest.class );
@Test @Test
public void testNoLoss() { public void testNoLoss() {
assertNoLoss( "insert into Address (city, state, zip, \"from\") values (?, ?, ?, 'insert value')" ); assertNoLoss( "insert into Address (city, state, zip, \"from\") values (?, ?, ?, 'insert value')" );
@ -43,6 +48,9 @@ public class BasicFormatterTest extends BaseUnitTestCase {
assertNoLoss( assertNoLoss(
"/* Here we' go! */ select case when p.age > 50 then 'old' when p.age > 18 then 'adult' else 'child' end from Person p where ( case when p.age > 50 then 'old' when p.age > 18 then 'adult' else 'child' end ) like ?" "/* Here we' go! */ select case when p.age > 50 then 'old' when p.age > 18 then 'adult' else 'child' end from Person p where ( case when p.age > 50 then 'old' when p.age > 18 then 'adult' else 'child' end ) like ?"
); );
assertNoLoss(
"(select p.pid from Address where city = 'Boston') union (select p.pid from Address where city = 'Taipei')"
);
} }
private void assertNoLoss(String query) { private void assertNoLoss(String query) {
@ -50,8 +58,9 @@ public class BasicFormatterTest extends BaseUnitTestCase {
StringTokenizer formatted = new StringTokenizer( formattedQuery, " \t\n\r\f()" ); StringTokenizer formatted = new StringTokenizer( formattedQuery, " \t\n\r\f()" );
StringTokenizer plain = new StringTokenizer( query, " \t\n\r\f()" ); StringTokenizer plain = new StringTokenizer( query, " \t\n\r\f()" );
System.out.println( "Original: " + query ); log.debugf( "Original: {}", query );
System.out.println( "Formatted: " + formattedQuery ); log.debugf( "Formatted: {}", formattedQuery );
while ( formatted.hasMoreTokens() && plain.hasMoreTokens() ) { while ( formatted.hasMoreTokens() && plain.hasMoreTokens() ) {
String plainToken = plain.nextToken(); String plainToken = plain.nextToken();
String formattedToken = formatted.nextToken(); String formattedToken = formatted.nextToken();

View File

@ -13,6 +13,8 @@ import org.hibernate.engine.jdbc.internal.FormatStyle;
import org.hibernate.testing.junit4.BaseUnitTestCase; import org.hibernate.testing.junit4.BaseUnitTestCase;
import org.junit.Test; import org.junit.Test;
import org.jboss.logging.Logger;
import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse; import static org.junit.Assert.assertFalse;
@ -20,6 +22,9 @@ import static org.junit.Assert.assertFalse;
* @author Vlad Mihalcea * @author Vlad Mihalcea
*/ */
public class DdlFormatterTest extends BaseUnitTestCase { public class DdlFormatterTest extends BaseUnitTestCase {
private static final Logger log = Logger.getLogger( DdlFormatterTest.class );
@Test @Test
public void testNoLoss() { public void testNoLoss() {
@ -44,8 +49,9 @@ public class DdlFormatterTest extends BaseUnitTestCase {
StringTokenizer formatted = new StringTokenizer( formattedQuery, " \t\n\r\f()" ); StringTokenizer formatted = new StringTokenizer( formattedQuery, " \t\n\r\f()" );
StringTokenizer plain = new StringTokenizer( query, " \t\n\r\f()" ); StringTokenizer plain = new StringTokenizer( query, " \t\n\r\f()" );
System.out.println( "Original: " + query ); log.debugf( "Original: {}", query );
System.out.println( "Formatted: " + formattedQuery ); log.debugf( "Formatted: {}", formattedQuery );
while ( formatted.hasMoreTokens() && plain.hasMoreTokens() ) { while ( formatted.hasMoreTokens() && plain.hasMoreTokens() ) {
String plainToken = plain.nextToken(); String plainToken = plain.nextToken();
String formattedToken = formatted.nextToken(); String formattedToken = formatted.nextToken();

View File

@ -0,0 +1,68 @@
/*
* 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>.
*/
package org.hibernate.query;
import javax.persistence.Entity;
import javax.persistence.Id;
import javax.persistence.Table;
import org.hibernate.jpa.test.BaseEntityManagerFunctionalTestCase;
import org.junit.Test;
import static org.hibernate.testing.transaction.TransactionUtil.doInJPA;
/**
* @author Vlad Mihalcea
*/
public class NativeQueryWithParenthesesTest extends BaseEntityManagerFunctionalTestCase {
@Override
public Class[] getAnnotatedClasses() {
return new Class[] {
Person.class
};
}
@Test
public void testParseParentheses() {
doInJPA( this::entityManagerFactory, entityManager -> {
entityManager.createNativeQuery(
"(SELECT p.id, p.name FROM Person p WHERE p.name LIKE 'A%') " +
"UNION " +
"(SELECT p.id, p.name FROM Person p WHERE p.name LIKE 'B%')", Person.class)
.getResultList();
} );
}
@Entity
@Table(name = "Person")
public static class Person {
@Id
private Integer id;
private String name;
public Integer getId() {
return id;
}
public void setId(Integer id) {
this.id = id;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
}
}