Skip to content

Commit 1856a2a

Browse files
beikovbeikov
committed
HHH-16780 Add array_agg and array constructor function
1 parent d59ecb6 commit 1856a2a

File tree

87 files changed

+2553
-324
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

87 files changed

+2553
-324
lines changed

‎hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/CockroachLegacyDialect.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -461,6 +461,8 @@ public void initializeFunctionRegistry(FunctionContributions functionContributio
461461
functionFactory.listagg_stringAgg( "string" );
462462
functionFactory.inverseDistributionOrderedSetAggregates();
463463
functionFactory.hypotheticalOrderedSetAggregates_windowEmulation();
464+
functionFactory.array_casting();
465+
functionFactory.arrayAggregate();
464466

465467
functionContributions.getFunctionRegistry().register(
466468
"trunc",

‎hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/DB2LegacyDialect.java

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@
4444
import org.hibernate.dialect.unique.AlterTableUniqueIndexDelegate;
4545
import org.hibernate.dialect.unique.SkipNullableUniqueDelegate;
4646
import org.hibernate.dialect.unique.UniqueDelegate;
47+
import org.hibernate.engine.jdbc.Size;
4748
import org.hibernate.engine.jdbc.dialect.spi.DialectResolutionInfo;
4849
import org.hibernate.engine.jdbc.env.spi.IdentifierHelper;
4950
import org.hibernate.engine.jdbc.env.spi.IdentifierHelperBuilder;
@@ -260,7 +261,8 @@ public boolean supportsDistinctFromPredicate() {
260261
public void initializeFunctionRegistry(FunctionContributions functionContributions) {
261262
super.initializeFunctionRegistry(functionContributions);
262263

263-
CommonFunctionFactory functionFactory = new CommonFunctionFactory(functionContributions);
264+
final DdlTypeRegistry ddlTypeRegistry = functionContributions.getTypeConfiguration().getDdlTypeRegistry();
265+
final CommonFunctionFactory functionFactory = new CommonFunctionFactory(functionContributions);
264266
// AVG by default uses the input type, so we possibly need to cast the argument type, hence a special function
265267
functionFactory.avg_castingNonDoubleArguments( this, SqlAstNodeRenderingMode.DEFAULT );
266268

@@ -362,14 +364,13 @@ public void initializeFunctionRegistry(FunctionContributions functionContributio
362364
functionContributions.getTypeConfiguration(),
363365
SqlAstNodeRenderingMode.DEFAULT,
364366
"||",
365-
functionContributions.getTypeConfiguration().getDdlTypeRegistry().getDescriptor( VARCHAR )
367+
ddlTypeRegistry.getDescriptor( VARCHAR )
366368
.getCastTypeName(
369+
Size.nil(),
367370
functionContributions.getTypeConfiguration()
368371
.getBasicTypeRegistry()
369372
.resolve( StandardBasicTypes.STRING ),
370-
null,
371-
null,
372-
null
373+
ddlTypeRegistry
373374
),
374375
true
375376
)

‎hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/DerbyLegacyDialect.java

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
import org.hibernate.dialect.sequence.SequenceSupport;
3535
import org.hibernate.dialect.temptable.TemporaryTable;
3636
import org.hibernate.dialect.temptable.TemporaryTableKind;
37+
import org.hibernate.engine.jdbc.Size;
3738
import org.hibernate.engine.jdbc.dialect.spi.DialectResolutionInfo;
3839
import org.hibernate.engine.jdbc.env.spi.IdentifierHelper;
3940
import org.hibernate.engine.jdbc.env.spi.IdentifierHelperBuilder;
@@ -271,8 +272,8 @@ public void initializeFunctionRegistry(FunctionContributions functionContributio
271272

272273
final BasicTypeRegistry basicTypeRegistry = functionContributions.getTypeConfiguration().getBasicTypeRegistry();
273274
final BasicType<String> stringType = basicTypeRegistry.resolve( StandardBasicTypes.STRING );
274-
275-
CommonFunctionFactory functionFactory = new CommonFunctionFactory(functionContributions);
275+
final DdlTypeRegistry ddlTypeRegistry = functionContributions.getTypeConfiguration().getDdlTypeRegistry();
276+
final CommonFunctionFactory functionFactory = new CommonFunctionFactory(functionContributions);
276277

277278
// Derby needs an actual argument type for aggregates like SUM, AVG, MIN, MAX to determine the result type
278279
functionFactory.aggregates( this, SqlAstNodeRenderingMode.NO_PLAIN_PARAMETER );
@@ -283,8 +284,8 @@ public void initializeFunctionRegistry(FunctionContributions functionContributio
283284
functionContributions.getTypeConfiguration(),
284285
SqlAstNodeRenderingMode.NO_PLAIN_PARAMETER,
285286
"||",
286-
functionContributions.getTypeConfiguration().getDdlTypeRegistry().getDescriptor( VARCHAR )
287-
.getCastTypeName( stringType, null, null, null ),
287+
ddlTypeRegistry.getDescriptor( VARCHAR )
288+
.getCastTypeName( Size.nil(), stringType, ddlTypeRegistry ),
288289
true
289290
)
290291
);

‎hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/H2LegacyDialect.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -366,10 +366,12 @@ public void initializeFunctionRegistry(FunctionContributions functionContributio
366366
functionFactory.rownum();
367367
if ( getVersion().isSameOrAfter( 1, 4, 200 ) ) {
368368
functionFactory.windowFunctions();
369+
functionFactory.inverseDistributionOrderedSetAggregates();
370+
functionFactory.hypotheticalOrderedSetAggregates();
369371
if ( getVersion().isSameOrAfter( 2 ) ) {
370372
functionFactory.listagg( null );
371-
functionFactory.inverseDistributionOrderedSetAggregates();
372-
functionFactory.hypotheticalOrderedSetAggregates();
373+
functionFactory.array();
374+
functionFactory.arrayAggregate();
373375
}
374376
else {
375377
// Use group_concat until 2.x as listagg was buggy

‎hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/HSQLLegacyDialect.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -247,6 +247,8 @@ public void initializeFunctionRegistry(FunctionContributions functionContributio
247247
functionFactory.rownum();
248248
}
249249
functionFactory.listagg_groupConcat();
250+
functionFactory.array();
251+
functionFactory.arrayAggregate();
250252
}
251253

252254
@Override

‎hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/OracleLegacyDialect.java

Lines changed: 39 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -103,12 +103,14 @@
103103
import org.hibernate.type.descriptor.jdbc.NullJdbcType;
104104
import org.hibernate.type.descriptor.jdbc.ObjectNullAsNullTypeJdbcType;
105105
import org.hibernate.type.descriptor.jdbc.spi.JdbcTypeRegistry;
106+
import org.hibernate.type.descriptor.sql.internal.ArrayDdlTypeImpl;
106107
import org.hibernate.type.descriptor.sql.internal.DdlTypeImpl;
107108
import org.hibernate.type.descriptor.sql.spi.DdlTypeRegistry;
108109
import org.hibernate.type.spi.TypeConfiguration;
109110

110111
import jakarta.persistence.TemporalType;
111112

113+
import static java.util.regex.Pattern.CASE_INSENSITIVE;
112114
import static org.hibernate.exception.spi.TemplatedViolatedConstraintNameExtractor.extractUsingTemplate;
113115
import static org.hibernate.query.sqm.TemporalUnit.DAY;
114116
import static org.hibernate.query.sqm.TemporalUnit.HOUR;
@@ -122,6 +124,8 @@
122124
import static org.hibernate.type.SqlTypes.BOOLEAN;
123125
import static org.hibernate.type.SqlTypes.DATE;
124126
import static org.hibernate.type.SqlTypes.DECIMAL;
127+
import static org.hibernate.type.SqlTypes.DOUBLE;
128+
import static org.hibernate.type.SqlTypes.FLOAT;
125129
import static org.hibernate.type.SqlTypes.GEOMETRY;
126130
import static org.hibernate.type.SqlTypes.INTEGER;
127131
import static org.hibernate.type.SqlTypes.JSON;
@@ -131,6 +135,7 @@
131135
import static org.hibernate.type.SqlTypes.SMALLINT;
132136
import static org.hibernate.type.SqlTypes.SQLXML;
133137
import static org.hibernate.type.SqlTypes.STRUCT;
138+
import static org.hibernate.type.SqlTypes.TABLE;
134139
import static org.hibernate.type.SqlTypes.TIME;
135140
import static org.hibernate.type.SqlTypes.TIMESTAMP;
136141
import static org.hibernate.type.SqlTypes.TIMESTAMP_WITH_TIMEZONE;
@@ -148,17 +153,20 @@
148153
*/
149154
public class OracleLegacyDialect extends Dialect {
150155

151-
private static final Pattern DISTINCT_KEYWORD_PATTERN = Pattern.compile( "\\bdistinct\\b" );
152-
private static final Pattern GROUP_BY_KEYWORD_PATTERN = Pattern.compile( "\\bgroup\\sby\\b" );
153-
private static final Pattern ORDER_BY_KEYWORD_PATTERN = Pattern.compile( "\\border\\sby\\b" );
154-
private static final Pattern UNION_KEYWORD_PATTERN = Pattern.compile( "\\bunion\\b" );
155-
private static final Pattern SQL_STATEMENT_TYPE_PATTERN = Pattern.compile("^(?:/\\*.*?\\*/)?\\s*(select|insert|update|delete)\\s+.*?", Pattern.CASE_INSENSITIVE);
156+
private static final Pattern DISTINCT_KEYWORD_PATTERN = Pattern.compile( "\\bdistinct\\b", CASE_INSENSITIVE );
157+
private static final Pattern GROUP_BY_KEYWORD_PATTERN = Pattern.compile( "\\bgroup\\s+by\\b", CASE_INSENSITIVE );
158+
private static final Pattern ORDER_BY_KEYWORD_PATTERN = Pattern.compile( "\\border\\s+by\\b", CASE_INSENSITIVE );
159+
private static final Pattern UNION_KEYWORD_PATTERN = Pattern.compile( "\\bunion\\b", CASE_INSENSITIVE );
160+
161+
private static final Pattern SQL_STATEMENT_TYPE_PATTERN =
162+
Pattern.compile( "^(?:/\\*.*?\\*/)?\\s*(select|insert|update|delete)\\s+.*?", CASE_INSENSITIVE );
163+
156164
private static final int PARAM_LIST_SIZE_LIMIT = 1000;
157165

158166
public static final String PREFER_LONG_RAW = "hibernate.dialect.oracle.prefer_long_raw";
159167

160168
private static final String yqmSelect =
161-
"( TRUNC(%2$s, 'MONTH') + NUMTOYMINTERVAL(%1$s, 'MONTH') + ( LEAST( EXTRACT( DAY FROM %2$s ), EXTRACT( DAY FROM LAST_DAY( TRUNC(%2$s, 'MONTH') + NUMTOYMINTERVAL(%1$s, 'MONTH') ) ) ) - 1 ) )";
169+
"(trunc(%2$s, 'MONTH') + numtoyminterval(%1$s, 'MONTH') + (least(extract(day from %2$s), extract(day from last_day(trunc(%2$s, 'MONTH') + numtoyminterval(%1$s, 'MONTH')))) - 1))";
162170

163171
private static final String ADD_YEAR_EXPRESSION = String.format( yqmSelect, "?2*12", "?3" );
164172
private static final String ADD_QUARTER_EXPRESSION = String.format( yqmSelect, "?2*3", "?3" );
@@ -218,6 +226,7 @@ public void initializeFunctionRegistry(FunctionContributions functionContributio
218226
functionFactory.addMonths();
219227
functionFactory.monthsBetween();
220228
functionFactory.everyAny_minMaxCase();
229+
functionFactory.repeat_rpad();
221230

222231
functionFactory.radians_acos();
223232
functionFactory.degrees_acos();
@@ -271,6 +280,9 @@ public void initializeFunctionRegistry(FunctionContributions functionContributio
271280
new OracleTruncFunction( functionContributions.getTypeConfiguration() )
272281
);
273282
functionContributions.getFunctionRegistry().registerAlternateKey( "truncate", "trunc" );
283+
284+
functionFactory.array_oracle();
285+
functionFactory.arrayAggregate_jsonArrayagg();
274286
}
275287

276288
@Override
@@ -623,6 +635,10 @@ protected String columnType(int sqlTypeCode) {
623635
case REAL:
624636
// Oracle's 'real' type is actually double precision
625637
return "float(24)";
638+
case DOUBLE:
639+
// Oracle's 'double precision' means float(126), and
640+
// we never need 126 bits (38 decimal digits)
641+
return "float(53)";
626642

627643
case NUMERIC:
628644
case DECIMAL:
@@ -668,6 +684,9 @@ else if ( getVersion().isSameOrAfter( 12 ) ) {
668684
ddlTypeRegistry.addDescriptor( new DdlTypeImpl( JSON, "blob", this ) );
669685
}
670686
}
687+
688+
ddlTypeRegistry.addDescriptor( new ArrayDdlTypeImpl( this, false ) );
689+
ddlTypeRegistry.addDescriptor( TABLE, new ArrayDdlTypeImpl( this, false ) );
671690
}
672691

673692
@Override
@@ -721,13 +740,20 @@ public JdbcType resolveSqlTypeDescriptor(
721740
}
722741
break;
723742
case Types.NUMERIC:
724-
if ( scale == -127 ) {
725-
// For some reason, the Oracle JDBC driver reports FLOAT
726-
// as NUMERIC with scale -127
727-
if ( precision <= getFloatPrecision() ) {
728-
return jdbcTypeRegistry.getDescriptor( Types.FLOAT );
743+
if ( precision > 8 // precision of 0 means something funny
744+
// For some reason, the Oracle JDBC driver reports
745+
// FLOAT or DOUBLE as NUMERIC with scale -127
746+
// (but note that expressions with unknown type
747+
// also get reported this way, so take care)
748+
&& scale == -127 ) {
749+
if ( precision <= 24 ) {
750+
// Can be represented as a Java float
751+
return jdbcTypeRegistry.getDescriptor( FLOAT );
752+
}
753+
else if ( precision <= 53 ) {
754+
// Can be represented as a Java double
755+
return jdbcTypeRegistry.getDescriptor( DOUBLE );
729756
}
730-
return jdbcTypeRegistry.getDescriptor( Types.DOUBLE );
731757
}
732758
//intentional fall-through:
733759
case Types.DECIMAL:
@@ -810,6 +836,7 @@ public void contributeTypes(TypeContributions typeContributions, ServiceRegistry
810836

811837
if ( OracleJdbcHelper.isUsable( serviceRegistry ) ) {
812838
typeContributions.contributeJdbcTypeConstructor( OracleJdbcHelper.getArrayJdbcTypeConstructor( serviceRegistry ) );
839+
typeContributions.contributeJdbcTypeConstructor( OracleJdbcHelper.getNestedTableJdbcTypeConstructor( serviceRegistry ) );
813840
}
814841
else {
815842
typeContributions.contributeJdbcType( OracleReflectionStructJdbcType.INSTANCE );

‎hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/OracleLegacySqlAstTranslator.java

Lines changed: 43 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
import java.util.ArrayList;
1010
import java.util.List;
1111

12+
import org.hibernate.dialect.OracleArrayJdbcType;
1213
import org.hibernate.engine.spi.SessionFactoryImplementor;
1314
import org.hibernate.internal.util.collections.Stack;
1415
import org.hibernate.metamodel.mapping.EmbeddableValuedModelPart;
@@ -53,6 +54,7 @@
5354
import org.hibernate.sql.exec.spi.JdbcOperation;
5455
import org.hibernate.sql.results.internal.SqlSelectionImpl;
5556
import org.hibernate.type.SqlTypes;
57+
import org.hibernate.type.descriptor.jdbc.JdbcType;
5658

5759
/**
5860
* A SQL AST translator for Oracle.
@@ -454,7 +456,8 @@ protected void renderComparison(Expression lhs, ComparisonOperator operator, Exp
454456
renderComparisonEmulateDecode( lhs, operator, rhs );
455457
return;
456458
}
457-
switch ( lhsExpressionType.getSingleJdbcMapping().getJdbcType().getDdlTypeCode() ) {
459+
final JdbcType jdbcType = lhsExpressionType.getSingleJdbcMapping().getJdbcType();
460+
switch ( jdbcType.getDdlTypeCode() ) {
458461
case SqlTypes.SQLXML:
459462
// In Oracle, XMLTYPE is not "comparable", so we have to use the xmldiff function for this purpose
460463
switch ( operator ) {
@@ -499,25 +502,51 @@ protected void renderComparison(Expression lhs, ComparisonOperator operator, Exp
499502
appendSql( ')' );
500503
break;
501504
case SqlTypes.ARRAY:
505+
final String arrayTypeName = ( (OracleArrayJdbcType) jdbcType ).getTypeName();
502506
switch ( operator ) {
503507
case DISTINCT_FROM:
504-
appendSql( "decode(" );
505-
arrayToString( lhs );
506-
appendSql( ',' );
507-
arrayToString( rhs );
508-
appendSql( ",0,1)=1" );
509-
break;
510508
case NOT_DISTINCT_FROM:
511-
appendSql( "decode(" );
512-
arrayToString( lhs );
509+
appendSql( arrayTypeName );
510+
appendSql( "_distinct(" );
511+
visitSqlSelectExpression( lhs );
513512
appendSql( ',' );
514-
arrayToString( rhs );
515-
appendSql( ",0,1)=0" );
513+
visitSqlSelectExpression( rhs );
514+
appendSql( ")" );
516515
break;
517516
default:
518-
arrayToString( lhs );
519-
appendSql( operator.sqlText() );
520-
arrayToString( rhs );
517+
appendSql( arrayTypeName );
518+
appendSql( "_cmp(" );
519+
visitSqlSelectExpression( lhs );
520+
appendSql( ',' );
521+
visitSqlSelectExpression( rhs );
522+
appendSql( ")" );
523+
break;
524+
}
525+
switch ( operator ) {
526+
case DISTINCT_FROM:
527+
appendSql( "=1" );
528+
break;
529+
case NOT_DISTINCT_FROM:
530+
appendSql( "=0" );
531+
break;
532+
case EQUAL:
533+
appendSql( "=0" );
534+
break;
535+
case NOT_EQUAL:
536+
appendSql( "<>0" );
537+
break;
538+
case LESS_THAN:
539+
appendSql( "=-1" );
540+
break;
541+
case GREATER_THAN:
542+
appendSql( "=1" );
543+
break;
544+
case LESS_THAN_OR_EQUAL:
545+
appendSql( "<=0" );
546+
break;
547+
case GREATER_THAN_OR_EQUAL:
548+
appendSql( ">=0" );
549+
break;
521550
}
522551
break;
523552
default:
@@ -526,19 +555,6 @@ protected void renderComparison(Expression lhs, ComparisonOperator operator, Exp
526555
}
527556
}
528557

529-
private void arrayToString(Expression expression) {
530-
appendSql("case when ");
531-
expression.accept( this );
532-
appendSql(" is not null then (select listagg(column_value||',')");
533-
if ( !getDialect().getVersion().isSameOrAfter( 18 ) ) {
534-
// The within group clause became optional in 18
535-
appendSql(" within group(order by rownum)");
536-
}
537-
appendSql("||';' from table(");
538-
expression.accept( this );
539-
appendSql(")) else null end");
540-
}
541-
542558
@Override
543559
protected void renderSelectTupleComparison(
544560
List<SqlSelection> lhsExpressions,

‎hibernate-community-dialects/src/main/java/org/hibernate/community/dialect/PostgreSQLLegacyDialect.java

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,7 @@
8484
import org.hibernate.type.descriptor.jdbc.UUIDJdbcType;
8585
import org.hibernate.type.descriptor.jdbc.XmlJdbcType;
8686
import org.hibernate.type.descriptor.jdbc.spi.JdbcTypeRegistry;
87+
import org.hibernate.type.descriptor.sql.internal.ArrayDdlTypeImpl;
8788
import org.hibernate.type.descriptor.sql.internal.CapacityDependentDdlType;
8889
import org.hibernate.type.descriptor.sql.internal.DdlTypeImpl;
8990
import org.hibernate.type.descriptor.sql.internal.Scale6IntervalSecondDdlType;
@@ -228,6 +229,9 @@ protected void registerColumnTypes(TypeContributions typeContributions, ServiceR
228229
super.registerColumnTypes( typeContributions, serviceRegistry );
229230
final DdlTypeRegistry ddlTypeRegistry = typeContributions.getTypeConfiguration().getDdlTypeRegistry();
230231

232+
// We need to configure that the array type uses the raw element type for casts
233+
ddlTypeRegistry.addDescriptor( new ArrayDdlTypeImpl( this, true ) );
234+
231235
// Register this type to be able to support Float[]
232236
// The issue is that the JDBC driver can't handle createArrayOf( "float(24)", ... )
233237
// It requires the use of "real" or "float4"
@@ -577,6 +581,8 @@ public void initializeFunctionRegistry(FunctionContributions functionContributio
577581
functionFactory.locate_positionSubstring();
578582
functionFactory.windowFunctions();
579583
functionFactory.listagg_stringAgg( "varchar" );
584+
functionFactory.array_casting();
585+
functionFactory.arrayAggregate();
580586

581587
if ( getVersion().isSameOrAfter( 9, 4 ) ) {
582588
functionFactory.makeDateTimeTimestamp();

‎hibernate-core/src/main/java/org/hibernate/dialect/CockroachDialect.java

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -448,6 +448,8 @@ public void initializeFunctionRegistry(FunctionContributions functionContributio
448448
functionFactory.listagg_stringAgg( "string" );
449449
functionFactory.inverseDistributionOrderedSetAggregates();
450450
functionFactory.hypotheticalOrderedSetAggregates_windowEmulation();
451+
functionFactory.array_casting();
452+
functionFactory.arrayAggregate();
451453

452454
functionContributions.getFunctionRegistry().register(
453455
"trunc",

0 commit comments

Comments
 (0)