Skip to content

HHH-19598 - support gaussdb procedure #10482

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 5 commits into from
Jul 8, 2025
Merged

HHH-19598 - support gaussdb procedure #10482

Show file tree
Hide file tree
Changes from 4 commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
f9036f2
hhh-19598 - support gaussdb callable
SweetWuXiaoMei Jul 8, 2025
4485624
hhh-19598 - support gaussdb callable
SweetWuXiaoMei Jul 8, 2025
0446117
hhh-19598 - support gaussdb callable
SweetWuXiaoMei Jul 8, 2025
35c3a0e
hhh-19598 - support gaussdb callable
SweetWuXiaoMei Jul 8, 2025
36cb58e
hhh-19598 - support gaussdb callable
SweetWuXiaoMei Jul 8, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
165 changes: 11 additions & 154 deletions ...alects/src/main/java/org/hibernate/community/dialect/GaussDBCallableStatementSupport.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -4,174 +4,31 @@
*/
package org.hibernate.community.dialect;

import java.util.List;

import org.hibernate.HibernateException;
import org.hibernate.dialect.type.AbstractPostgreSQLStructJdbcType;
import org.hibernate.procedure.internal.AbstractStandardCallableStatementSupport;
import org.hibernate.procedure.spi.FunctionReturnImplementor;
import org.hibernate.procedure.spi.ProcedureCallImplementor;
import org.hibernate.procedure.internal.StandardCallableStatementSupport;
import org.hibernate.procedure.spi.ProcedureParameterImplementor;
import org.hibernate.type.OutputableType;
import org.hibernate.query.spi.ProcedureParameterMetadataImplementor;
import org.hibernate.sql.exec.internal.JdbcCallImpl;
import org.hibernate.sql.exec.spi.JdbcCallParameterRegistration;
import org.hibernate.sql.exec.spi.JdbcOperationQueryCall;
import org.hibernate.type.SqlTypes;

import jakarta.persistence.ParameterMode;

/**
* GaussDB implementation of CallableStatementSupport.
*
* @author liubao
*
* Notes: Original code of this class is based on PostgreSQLCallableStatementSupport.
* Notes: Original code of this class is based on OracleCallableStatementSupport.
*/
public class GaussDBCallableStatementSupport extends AbstractStandardCallableStatementSupport {
/**
* Singleton access
*/
public static final GaussDBCallableStatementSupport INSTANCE = new GaussDBCallableStatementSupport( true );
public class GaussDBCallableStatementSupport extends StandardCallableStatementSupport {

private final boolean supportsProcedures;
public static final StandardCallableStatementSupport INSTANCE = new GaussDBCallableStatementSupport(
true );

private GaussDBCallableStatementSupport(boolean supportsProcedures) {
this.supportsProcedures = supportsProcedures;
super( supportsProcedures );
}

@Override
public JdbcOperationQueryCall interpretCall(ProcedureCallImplementor<?> procedureCall) {
final String procedureName = procedureCall.getProcedureName();
final FunctionReturnImplementor<?> functionReturn = procedureCall.getFunctionReturn();
final ProcedureParameterMetadataImplementor parameterMetadata = procedureCall.getParameterMetadata();
final boolean firstParamIsRefCursor = parameterMetadata.getParameterCount() != 0
&& isFirstParameterModeRefCursor( parameterMetadata );

final List<? extends ProcedureParameterImplementor<?>> registrations = parameterMetadata.getRegistrationsAsList();
final int paramStringSizeEstimate;
if ( functionReturn == null && parameterMetadata.hasNamedParameters() ) {
// That's just a rough estimate. I guess most params will have fewer than 8 chars on average
paramStringSizeEstimate = registrations.size() * 10;
}
else {
// For every param rendered as '?' we have a comma, hence the estimate
paramStringSizeEstimate = registrations.size() * 2;
}
final JdbcCallImpl.Builder builder = new JdbcCallImpl.Builder();

final int jdbcParameterOffset;
final int startIndex;
final CallMode callMode;
if ( functionReturn != null ) {
if ( functionReturn.getJdbcTypeCode() == SqlTypes.REF_CURSOR ) {
if ( firstParamIsRefCursor ) {
// validate that the parameter strategy is positional (cannot mix, and REF_CURSOR is inherently positional)
if ( parameterMetadata.hasNamedParameters() ) {
throw new HibernateException( "Cannot mix named parameters and REF_CURSOR parameter on GaussDB" );
}
callMode = CallMode.CALL_RETURN;
startIndex = 1;
jdbcParameterOffset = 1;
builder.addParameterRegistration( registrations.get( 0 ).toJdbcParameterRegistration( 1, procedureCall ) );
}
else {
callMode = CallMode.TABLE_FUNCTION;
startIndex = 0;
jdbcParameterOffset = 1;
}
}
else {
callMode = CallMode.FUNCTION;
startIndex = 0;
jdbcParameterOffset = 1;
}
}
else if ( supportsProcedures ) {
jdbcParameterOffset = 1;
startIndex = 0;
callMode = CallMode.NATIVE_CALL;
}
else if ( firstParamIsRefCursor ) {
// validate that the parameter strategy is positional (cannot mix, and REF_CURSOR is inherently positional)
if ( parameterMetadata.hasNamedParameters() ) {
throw new HibernateException( "Cannot mix named parameters and REF_CURSOR parameter on GaussDB" );
}
jdbcParameterOffset = 1;
startIndex = 1;
callMode = CallMode.CALL_RETURN;
builder.addParameterRegistration( registrations.get( 0 ).toJdbcParameterRegistration( 1, procedureCall ) );
}
else {
jdbcParameterOffset = 1;
startIndex = 0;
callMode = CallMode.CALL;
}

final StringBuilder buffer = new StringBuilder( callMode.start.length() + callMode.end.length() + procedureName.length() + paramStringSizeEstimate )
.append( callMode.start );
buffer.append( procedureName );

if ( startIndex == registrations.size() ) {
buffer.append( '(' );
}
else {
char sep = '(';
for ( int i = startIndex; i < registrations.size(); i++ ) {
final ProcedureParameterImplementor<?> parameter = registrations.get( i );
if ( !supportsProcedures && parameter.getMode() == ParameterMode.REF_CURSOR ) {
throw new HibernateException(
"GaussDB supports only one REF_CURSOR parameter, but multiple were registered" );
}
buffer.append( sep );
final JdbcCallParameterRegistration registration = parameter.toJdbcParameterRegistration(
i + jdbcParameterOffset,
procedureCall
);
final OutputableType<?> type = registration.getParameterType();
final String castType;
if ( parameter.getName() != null ) {
buffer.append( parameter.getName() ).append( " => " );
}
if ( type != null && type.getJdbcType() instanceof AbstractPostgreSQLStructJdbcType ) {
// We have to cast struct type parameters so that GaussDB understands nulls
castType = ( (AbstractPostgreSQLStructJdbcType) type.getJdbcType() ).getStructTypeName();
buffer.append( "cast(" );
}
else {
castType = null;
}
buffer.append( "?" );
if ( castType != null ) {
buffer.append( " as " ).append( castType ).append( ')' );
}
sep = ',';
builder.addParameterRegistration( registration );
}
}

buffer.append( callMode.end );
builder.setCallableName( buffer.toString() );
return builder.buildJdbcCall();
}

private static boolean isFirstParameterModeRefCursor(ProcedureParameterMetadataImplementor parameterMetadata) {
return parameterMetadata.getRegistrationsAsList().get( 0 ).getMode() == ParameterMode.REF_CURSOR;
}

enum CallMode {
TABLE_FUNCTION("select * from ", ")"),
FUNCTION("select ", ")"),
NATIVE_CALL("call ", ")"),
CALL_RETURN("{?=call ", ")}"),
CALL("{call ", ")}");

private final String start;
private final String end;

CallMode(String start, String end) {
this.start = start;
this.end = end;
}
protected void appendNameParameter(
StringBuilder buffer,
ProcedureParameterImplementor parameter,
JdbcCallParameterRegistration registration) {
buffer.append( parameter.getName() ).append( " => ?" );
}
}
Loading