HHH-18332 Test stored procedures on DB2
This commit is contained in:
Marco Belladelli
parent
a86b35e664
commit
4c64ddc153
|
|
@ -0,0 +1,436 @@
|
||||||
|
/*
|
||||||
|
* 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.orm.test.procedure;
|
||||||
|
|
||||||
|
import java.sql.PreparedStatement;
|
||||||
|
import java.sql.ResultSet;
|
||||||
|
import java.sql.SQLException;
|
||||||
|
import java.sql.Statement;
|
||||||
|
import java.sql.Timestamp;
|
||||||
|
import java.time.LocalDateTime;
|
||||||
|
import java.time.ZoneOffset;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
import org.hibernate.Session;
|
||||||
|
import org.hibernate.dialect.DB2Dialect;
|
||||||
|
import org.hibernate.procedure.ProcedureCall;
|
||||||
|
import org.hibernate.query.procedure.ProcedureParameter;
|
||||||
|
import org.hibernate.result.Output;
|
||||||
|
import org.hibernate.result.ResultSetOutput;
|
||||||
|
import org.hibernate.type.NumericBooleanConverter;
|
||||||
|
import org.hibernate.type.YesNoConverter;
|
||||||
|
|
||||||
|
import org.hibernate.testing.orm.junit.EntityManagerFactoryScope;
|
||||||
|
import org.hibernate.testing.orm.junit.Jira;
|
||||||
|
import org.hibernate.testing.orm.junit.Jpa;
|
||||||
|
import org.hibernate.testing.orm.junit.RequiresDialect;
|
||||||
|
import org.junit.jupiter.api.AfterAll;
|
||||||
|
import org.junit.jupiter.api.BeforeAll;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
import jakarta.persistence.Entity;
|
||||||
|
import jakarta.persistence.Id;
|
||||||
|
import jakarta.persistence.NamedStoredProcedureQueries;
|
||||||
|
import jakarta.persistence.NamedStoredProcedureQuery;
|
||||||
|
import jakarta.persistence.ParameterMode;
|
||||||
|
import jakarta.persistence.StoredProcedureParameter;
|
||||||
|
import jakarta.persistence.StoredProcedureQuery;
|
||||||
|
|
||||||
|
import static org.assertj.core.api.Assertions.assertThat;
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertFalse;
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
||||||
|
import static org.junit.jupiter.api.Assertions.fail;
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @author Marco Belladelli
|
||||||
|
*/
|
||||||
|
@Jpa( annotatedClasses = {
|
||||||
|
Person.class,
|
||||||
|
Phone.class,
|
||||||
|
Vote.class,
|
||||||
|
DB2StoredProcedureTest.IdHolder.class,
|
||||||
|
} )
|
||||||
|
@RequiresDialect( value = DB2Dialect.class )
|
||||||
|
@Jira( "https://hibernate.atlassian.net/browse/HHH-18332" )
|
||||||
|
public class DB2StoredProcedureTest {
|
||||||
|
@Test
|
||||||
|
public void testStoredProcedureOutParameter(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final StoredProcedureQuery query = entityManager.createStoredProcedureQuery( "sp_count_phones" );
|
||||||
|
query.registerStoredProcedureParameter( 1, Long.class, ParameterMode.IN );
|
||||||
|
query.registerStoredProcedureParameter( 2, Long.class, ParameterMode.OUT );
|
||||||
|
|
||||||
|
query.setParameter( 1, 1L );
|
||||||
|
|
||||||
|
query.execute();
|
||||||
|
final Long phoneCount = (Long) query.getOutputParameterValue( 2 );
|
||||||
|
assertEquals( Long.valueOf( 2 ), phoneCount );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
@Jira( "https://hibernate.atlassian.net/browse/HHH-18302" )
|
||||||
|
public void testStoredProcedureNamedParameters(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final StoredProcedureQuery query = entityManager.createStoredProcedureQuery( "sp_count_phones" );
|
||||||
|
query.registerStoredProcedureParameter( "personId", Long.class, ParameterMode.IN );
|
||||||
|
query.registerStoredProcedureParameter( "phoneCount", Long.class, ParameterMode.OUT );
|
||||||
|
|
||||||
|
query.setParameter( "personId", 1L );
|
||||||
|
|
||||||
|
query.execute();
|
||||||
|
final Long phoneCount = (Long) query.getOutputParameterValue( "phoneCount" );
|
||||||
|
assertEquals( Long.valueOf( 2 ), phoneCount );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testStoredProcedureRefCursor(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final StoredProcedureQuery query = entityManager.createStoredProcedureQuery( "sp_person_phones" );
|
||||||
|
query.registerStoredProcedureParameter( 1, Long.class, ParameterMode.IN );
|
||||||
|
query.registerStoredProcedureParameter( 2, Class.class, ParameterMode.REF_CURSOR );
|
||||||
|
query.setParameter( 1, 1L );
|
||||||
|
|
||||||
|
query.execute();
|
||||||
|
final List<Object[]> postComments = query.getResultList();
|
||||||
|
assertNotNull( postComments );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testHibernateProcedureCallRefCursor(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final Session session = entityManager.unwrap( Session.class );
|
||||||
|
|
||||||
|
final ProcedureCall call = session.createStoredProcedureCall( "sp_person_phones" );
|
||||||
|
final ProcedureParameter<Long> inParam = call.registerParameter(
|
||||||
|
1,
|
||||||
|
Long.class,
|
||||||
|
ParameterMode.IN
|
||||||
|
);
|
||||||
|
call.setParameter( inParam, 1L );
|
||||||
|
call.registerParameter( 2, Class.class, ParameterMode.REF_CURSOR );
|
||||||
|
|
||||||
|
final Output output = call.getOutputs().getCurrent();
|
||||||
|
final List<Object[]> postComments = ( (ResultSetOutput) output ).getResultList();
|
||||||
|
assertEquals( 2, postComments.size() );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testStoredProcedureReturnValue(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final Integer phoneCount = (Integer) entityManager
|
||||||
|
.createNativeQuery( "SELECT fn_count_phones(:personId) FROM SYSIBM.DUAL" )
|
||||||
|
.setParameter( "personId", 1L )
|
||||||
|
.getSingleResult();
|
||||||
|
assertEquals( 2, phoneCount );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testFunctionCallWithJdbc(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final Session session = entityManager.unwrap( Session.class );
|
||||||
|
session.doWork( connection -> {
|
||||||
|
try (final PreparedStatement function = connection.prepareStatement(
|
||||||
|
"select * from table(fn_person_and_phones( ? ))" )) {
|
||||||
|
function.setInt( 1, 1 );
|
||||||
|
function.execute();
|
||||||
|
try (final ResultSet resultSet = function.getResultSet()) {
|
||||||
|
while ( resultSet.next() ) {
|
||||||
|
assertThat( resultSet.getLong( 1 ) ).isInstanceOf( Long.class );
|
||||||
|
assertThat( resultSet.getString( 2 ) ).isInstanceOf( String.class );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSysRefCursorAsOutParameter(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final StoredProcedureQuery function = entityManager.createNamedStoredProcedureQuery( "singleRefCursor" );
|
||||||
|
|
||||||
|
function.execute();
|
||||||
|
|
||||||
|
assertFalse( function.hasMoreResults() );
|
||||||
|
Long value = null;
|
||||||
|
|
||||||
|
try (final ResultSet resultSet = (ResultSet) function.getOutputParameterValue( 1 )) {
|
||||||
|
while ( resultSet.next() ) {
|
||||||
|
value = resultSet.getLong( 1 );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (final SQLException e) {
|
||||||
|
fail( e.getMessage() );
|
||||||
|
}
|
||||||
|
|
||||||
|
assertEquals( Long.valueOf( 1 ), value );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testOutAndSysRefCursorAsOutParameter(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final StoredProcedureQuery function = entityManager.createNamedStoredProcedureQuery( "outAndRefCursor" );
|
||||||
|
|
||||||
|
function.execute();
|
||||||
|
|
||||||
|
assertFalse( function.hasMoreResults() );
|
||||||
|
Long value = null;
|
||||||
|
|
||||||
|
try (final ResultSet resultSet = (ResultSet) function.getOutputParameterValue( 1 )) {
|
||||||
|
while ( resultSet.next() ) {
|
||||||
|
value = resultSet.getLong( 1 );
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (final SQLException e) {
|
||||||
|
fail( e.getMessage() );
|
||||||
|
}
|
||||||
|
|
||||||
|
assertEquals( value, function.getOutputParameterValue( 2 ) );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testBindParameterAsHibernateType(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final StoredProcedureQuery query = entityManager.createStoredProcedureQuery( "sp_phone_validity" )
|
||||||
|
.registerStoredProcedureParameter( 1, NumericBooleanConverter.class, ParameterMode.IN )
|
||||||
|
.registerStoredProcedureParameter( 2, Class.class, ParameterMode.REF_CURSOR )
|
||||||
|
.setParameter( 1, true );
|
||||||
|
|
||||||
|
query.execute();
|
||||||
|
final List phones = query.getResultList();
|
||||||
|
assertEquals( 1, phones.size() );
|
||||||
|
assertEquals( "123-456-7890", phones.get( 0 ) );
|
||||||
|
} );
|
||||||
|
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final Vote vote1 = new Vote();
|
||||||
|
vote1.setId( 1L );
|
||||||
|
vote1.setVoteChoice( true );
|
||||||
|
|
||||||
|
entityManager.persist( vote1 );
|
||||||
|
|
||||||
|
final Vote vote2 = new Vote();
|
||||||
|
vote2.setId( 2L );
|
||||||
|
vote2.setVoteChoice( false );
|
||||||
|
|
||||||
|
entityManager.persist( vote2 );
|
||||||
|
} );
|
||||||
|
|
||||||
|
scope.inTransaction( entityManager -> {
|
||||||
|
final StoredProcedureQuery query = entityManager.createStoredProcedureQuery( "sp_votes" )
|
||||||
|
.registerStoredProcedureParameter( 1, YesNoConverter.class, ParameterMode.IN )
|
||||||
|
.registerStoredProcedureParameter( 2, Class.class, ParameterMode.REF_CURSOR )
|
||||||
|
.setParameter( 1, true );
|
||||||
|
|
||||||
|
query.execute();
|
||||||
|
final List votes = query.getResultList();
|
||||||
|
assertEquals( 1, votes.size() );
|
||||||
|
assertEquals( 1, ( (Number) votes.get( 0 ) ).intValue() );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@BeforeAll
|
||||||
|
public void prepareSchema(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inTransaction( (entityManager) -> entityManager.unwrap( Session.class ).doWork( (connection) -> {
|
||||||
|
try (final Statement statement = connection.createStatement()) {
|
||||||
|
statement.executeUpdate(
|
||||||
|
"CREATE OR REPLACE PROCEDURE sp_count_phones( " +
|
||||||
|
" IN personId INT, " +
|
||||||
|
" OUT phoneCount INT ) " +
|
||||||
|
"BEGIN " +
|
||||||
|
" SELECT COUNT(*) INTO phoneCount " +
|
||||||
|
" FROM phone " +
|
||||||
|
" WHERE person_id = personId; " +
|
||||||
|
"END;"
|
||||||
|
);
|
||||||
|
statement.executeUpdate(
|
||||||
|
"CREATE OR REPLACE PROCEDURE sp_person_phones( " +
|
||||||
|
" IN personId INT," +
|
||||||
|
" OUT personPhones CURSOR) " +
|
||||||
|
"BEGIN " +
|
||||||
|
" SET personPhones = CURSOR FOR " +
|
||||||
|
" SELECT *" +
|
||||||
|
" FROM phone " +
|
||||||
|
" WHERE person_id = personId; " +
|
||||||
|
" OPEN personPhones; " +
|
||||||
|
"END;"
|
||||||
|
);
|
||||||
|
statement.executeUpdate(
|
||||||
|
"CREATE OR REPLACE FUNCTION fn_count_phones( " +
|
||||||
|
" IN personId INT ) " +
|
||||||
|
" RETURNS INT " +
|
||||||
|
"BEGIN " +
|
||||||
|
" DECLARE phoneCount INT; " +
|
||||||
|
" SELECT COUNT(*) INTO phoneCount " +
|
||||||
|
" FROM phone " +
|
||||||
|
" WHERE person_id = personId; " +
|
||||||
|
" RETURN phoneCount; " +
|
||||||
|
"END;"
|
||||||
|
);
|
||||||
|
statement.executeUpdate(
|
||||||
|
"CREATE OR REPLACE FUNCTION fn_person_and_phones(personId INTEGER) " +
|
||||||
|
" RETURNS TABLE " +
|
||||||
|
" ( " +
|
||||||
|
" \"pr.id\" BIGINT, " +
|
||||||
|
" \"pr.name\" VARCHAR(255), " +
|
||||||
|
" \"pr.nickName\" VARCHAR(255), " +
|
||||||
|
" \"pr.address\" VARCHAR(255), " +
|
||||||
|
" \"pr.createdOn\" TIMESTAMP, " +
|
||||||
|
" \"pr.version\" INTEGER, " +
|
||||||
|
" \"ph.id\" BIGINT, " +
|
||||||
|
" \"ph.person_id\" BIGINT, " +
|
||||||
|
" \"ph.phone_number\" VARCHAR(255), " +
|
||||||
|
" \"ph.valid\" BOOLEAN " +
|
||||||
|
" ) " +
|
||||||
|
" BEGIN ATOMIC " +
|
||||||
|
" RETURN SELECT pr.id AS \"pr.id\", " +
|
||||||
|
" pr.name AS \"pr.name\", " +
|
||||||
|
" pr.nickName AS \"pr.nickName\", " +
|
||||||
|
" pr.address AS \"pr.address\", " +
|
||||||
|
" pr.createdOn AS \"pr.createdOn\", " +
|
||||||
|
" pr.version AS \"pr.version\", " +
|
||||||
|
" ph.id AS \"ph.id\", " +
|
||||||
|
" ph.person_id AS \"ph.person_id\", " +
|
||||||
|
" ph.phone_number AS \"ph.phone_number\", " +
|
||||||
|
" ph.valid AS \"ph.valid\" " +
|
||||||
|
" FROM person pr " +
|
||||||
|
" JOIN phone ph ON pr.id = ph.person_id " +
|
||||||
|
" WHERE pr.id = personId; " +
|
||||||
|
" END; " );
|
||||||
|
statement.executeUpdate(
|
||||||
|
"CREATE OR REPLACE " +
|
||||||
|
"PROCEDURE singleRefCursor(OUT p_recordset CURSOR) " +
|
||||||
|
" BEGIN " +
|
||||||
|
" SET p_recordset = CURSOR FOR " +
|
||||||
|
" SELECT 1 as id " +
|
||||||
|
" FROM sysibm.dual; " +
|
||||||
|
" OPEN p_recordset; " +
|
||||||
|
" END; "
|
||||||
|
);
|
||||||
|
statement.executeUpdate(
|
||||||
|
"CREATE OR REPLACE " +
|
||||||
|
"PROCEDURE outAndRefCursor(OUT p_recordset CURSOR, OUT p_value INT) " +
|
||||||
|
" BEGIN " +
|
||||||
|
" SET p_recordset = CURSOR FOR " +
|
||||||
|
" SELECT 1 as id " +
|
||||||
|
" FROM sysibm.dual; " +
|
||||||
|
" SELECT 1 INTO p_value FROM sysibm.dual; " +
|
||||||
|
" OPEN p_recordset; " +
|
||||||
|
" END; "
|
||||||
|
);
|
||||||
|
statement.executeUpdate(
|
||||||
|
"CREATE OR REPLACE PROCEDURE sp_phone_validity ( " +
|
||||||
|
" IN validity BOOLEAN, " +
|
||||||
|
" OUT personPhones CURSOR ) " +
|
||||||
|
"BEGIN " +
|
||||||
|
" SET personPhones = CURSOR FOR " +
|
||||||
|
" SELECT phone_number " +
|
||||||
|
" FROM phone " +
|
||||||
|
" WHERE valid = validity; " +
|
||||||
|
" OPEN personPhones; " +
|
||||||
|
"END;"
|
||||||
|
);
|
||||||
|
statement.executeUpdate(
|
||||||
|
"CREATE OR REPLACE PROCEDURE sp_votes ( " +
|
||||||
|
" IN validity BOOLEAN, " +
|
||||||
|
" OUT votes CURSOR ) " +
|
||||||
|
"BEGIN " +
|
||||||
|
" SET votes = CURSOR FOR " +
|
||||||
|
" SELECT id " +
|
||||||
|
" FROM vote " +
|
||||||
|
" WHERE vote_choice = validity; " +
|
||||||
|
" OPEN votes; " +
|
||||||
|
"END;"
|
||||||
|
);
|
||||||
|
}
|
||||||
|
catch (final SQLException e) {
|
||||||
|
System.err.println( "Error exporting procedure and function definitions to DB2 database : " + e.getMessage() );
|
||||||
|
e.printStackTrace( System.err );
|
||||||
|
}
|
||||||
|
} ) );
|
||||||
|
|
||||||
|
scope.inTransaction( (entityManager) -> {
|
||||||
|
final Person person1 = new Person( 1L, "John Doe" );
|
||||||
|
person1.setNickName( "JD" );
|
||||||
|
person1.setAddress( "Earth" );
|
||||||
|
person1.setCreatedOn( Timestamp.from( LocalDateTime.of( 2000, 1, 1, 0, 0, 0 ).toInstant( ZoneOffset.UTC ) ) );
|
||||||
|
|
||||||
|
entityManager.persist( person1 );
|
||||||
|
|
||||||
|
final Phone phone1 = new Phone( "123-456-7890" );
|
||||||
|
phone1.setId( 1L );
|
||||||
|
phone1.setValid( true );
|
||||||
|
|
||||||
|
person1.addPhone( phone1 );
|
||||||
|
|
||||||
|
final Phone phone2 = new Phone( "098_765-4321" );
|
||||||
|
phone2.setId( 2L );
|
||||||
|
phone2.setValid( false );
|
||||||
|
|
||||||
|
person1.addPhone( phone2 );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@AfterAll
|
||||||
|
public void cleanUpSchema(EntityManagerFactoryScope scope) {
|
||||||
|
scope.inEntityManager( (em) -> {
|
||||||
|
final Session session = em.unwrap( Session.class );
|
||||||
|
session.doWork( connection -> {
|
||||||
|
try (final Statement statement = connection.createStatement()) {
|
||||||
|
statement.executeUpdate( "DROP PROCEDURE sp_count_phones" );
|
||||||
|
statement.executeUpdate( "DROP PROCEDURE sp_person_phones" );
|
||||||
|
statement.executeUpdate( "DROP FUNCTION fn_count_phones" );
|
||||||
|
statement.executeUpdate( "DROP FUNCTION fn_person_and_phones" );
|
||||||
|
statement.executeUpdate( "DROP PROCEDURE singleRefCursor" );
|
||||||
|
statement.executeUpdate( "DROP PROCEDURE outAndRefCursor" );
|
||||||
|
statement.executeUpdate( "DROP PROCEDURE sp_phone_validity" );
|
||||||
|
statement.executeUpdate( "DROP PROCEDURE sp_votes" );
|
||||||
|
}
|
||||||
|
catch (final SQLException ignore) {
|
||||||
|
}
|
||||||
|
} );
|
||||||
|
|
||||||
|
scope.inTransaction( em, (em2) -> {
|
||||||
|
final List<Person> people = em.createQuery( "from Person", Person.class ).getResultList();
|
||||||
|
people.forEach( em::remove );
|
||||||
|
} );
|
||||||
|
} );
|
||||||
|
}
|
||||||
|
|
||||||
|
@NamedStoredProcedureQueries( {
|
||||||
|
@NamedStoredProcedureQuery(
|
||||||
|
name = "singleRefCursor",
|
||||||
|
procedureName = "singleRefCursor",
|
||||||
|
parameters = {
|
||||||
|
@StoredProcedureParameter( mode = ParameterMode.REF_CURSOR, type = void.class )
|
||||||
|
}
|
||||||
|
),
|
||||||
|
@NamedStoredProcedureQuery(
|
||||||
|
name = "outAndRefCursor",
|
||||||
|
procedureName = "outAndRefCursor",
|
||||||
|
parameters = {
|
||||||
|
@StoredProcedureParameter( mode = ParameterMode.REF_CURSOR, type = void.class ),
|
||||||
|
@StoredProcedureParameter( mode = ParameterMode.OUT, type = Long.class ),
|
||||||
|
}
|
||||||
|
)
|
||||||
|
} )
|
||||||
|
@Entity( name = "IdHolder" )
|
||||||
|
public static class IdHolder {
|
||||||
|
@Id
|
||||||
|
Long id;
|
||||||
|
String name;
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue