|
|
|
|
@ -0,0 +1,314 @@
|
|
|
|
|
/*
|
|
|
|
|
* SPDX-License-Identifier: LGPL-2.1-or-later
|
|
|
|
|
* Copyright Red Hat Inc. and Hibernate Authors
|
|
|
|
|
*/
|
|
|
|
|
package org.hibernate.dialect.aggregate;
|
|
|
|
|
|
|
|
|
|
import org.hibernate.dialect.Dialect;
|
|
|
|
|
import org.hibernate.internal.util.StringHelper;
|
|
|
|
|
import org.hibernate.mapping.Column;
|
|
|
|
|
import org.hibernate.metamodel.mapping.JdbcMapping;
|
|
|
|
|
import org.hibernate.metamodel.mapping.SelectableMapping;
|
|
|
|
|
import org.hibernate.metamodel.mapping.SelectablePath;
|
|
|
|
|
import org.hibernate.metamodel.mapping.SqlTypedMapping;
|
|
|
|
|
import org.hibernate.sql.ast.SqlAstNodeRenderingMode;
|
|
|
|
|
import org.hibernate.sql.ast.SqlAstTranslator;
|
|
|
|
|
import org.hibernate.sql.ast.spi.SqlAppender;
|
|
|
|
|
import org.hibernate.type.BasicPluralType;
|
|
|
|
|
import org.hibernate.type.spi.TypeConfiguration;
|
|
|
|
|
|
|
|
|
|
import java.util.LinkedHashMap;
|
|
|
|
|
import java.util.Map;
|
|
|
|
|
|
|
|
|
|
import static org.hibernate.type.SqlTypes.ARRAY;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.BIGINT;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.BINARY;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.BOOLEAN;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.DOUBLE;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.FLOAT;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.INTEGER;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.JSON;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.JSON_ARRAY;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.LONG32VARBINARY;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.SMALLINT;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.TINYINT;
|
|
|
|
|
import static org.hibernate.type.SqlTypes.VARBINARY;
|
|
|
|
|
|
|
|
|
|
public class CockroachDBAggregateSupport extends AggregateSupportImpl {
|
|
|
|
|
|
|
|
|
|
private static final AggregateSupport INSTANCE = new CockroachDBAggregateSupport();
|
|
|
|
|
|
|
|
|
|
public static AggregateSupport valueOf(Dialect dialect) {
|
|
|
|
|
return CockroachDBAggregateSupport.INSTANCE;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public String aggregateComponentCustomReadExpression(
|
|
|
|
|
String template,
|
|
|
|
|
String placeholder,
|
|
|
|
|
String aggregateParentReadExpression,
|
|
|
|
|
String columnExpression,
|
|
|
|
|
int aggregateColumnTypeCode,
|
|
|
|
|
SqlTypedMapping column) {
|
|
|
|
|
switch ( aggregateColumnTypeCode ) {
|
|
|
|
|
case JSON_ARRAY:
|
|
|
|
|
case JSON:
|
|
|
|
|
switch ( column.getJdbcMapping().getJdbcType().getDefaultSqlTypeCode() ) {
|
|
|
|
|
case JSON:
|
|
|
|
|
case JSON_ARRAY:
|
|
|
|
|
return template.replace(
|
|
|
|
|
placeholder,
|
|
|
|
|
aggregateParentReadExpression + "->'" + columnExpression + "'"
|
|
|
|
|
);
|
|
|
|
|
case BINARY:
|
|
|
|
|
case VARBINARY:
|
|
|
|
|
case LONG32VARBINARY:
|
|
|
|
|
// We encode binary data as hex, so we have to decode here
|
|
|
|
|
return template.replace(
|
|
|
|
|
placeholder,
|
|
|
|
|
"decode(" + aggregateParentReadExpression + "->>'" + columnExpression + "','hex')"
|
|
|
|
|
);
|
|
|
|
|
case ARRAY:
|
|
|
|
|
final BasicPluralType<?, ?> pluralType = (BasicPluralType<?, ?>) column.getJdbcMapping();
|
|
|
|
|
switch ( pluralType.getElementType().getJdbcType().getDefaultSqlTypeCode() ) {
|
|
|
|
|
case BOOLEAN:
|
|
|
|
|
case TINYINT:
|
|
|
|
|
case SMALLINT:
|
|
|
|
|
case INTEGER:
|
|
|
|
|
case BIGINT:
|
|
|
|
|
case FLOAT:
|
|
|
|
|
case DOUBLE:
|
|
|
|
|
// For types that are natively supported in jsonb we can use jsonb_array_elements,
|
|
|
|
|
// but note that we can't use that for string types,
|
|
|
|
|
// because casting a jsonb[] to text[] will not omit the quotes of the jsonb text values
|
|
|
|
|
return template.replace(
|
|
|
|
|
placeholder,
|
|
|
|
|
"cast(array(select jsonb_array_elements(" + aggregateParentReadExpression + "->'" + columnExpression + "')) as " + column.getColumnDefinition() + ')'
|
|
|
|
|
);
|
|
|
|
|
case BINARY:
|
|
|
|
|
case VARBINARY:
|
|
|
|
|
case LONG32VARBINARY:
|
|
|
|
|
// We encode binary data as hex, so we have to decode here
|
|
|
|
|
return template.replace(
|
|
|
|
|
placeholder,
|
|
|
|
|
"array(select decode(jsonb_array_elements_text(" + aggregateParentReadExpression + "->'" + columnExpression + "'),'hex'))"
|
|
|
|
|
);
|
|
|
|
|
default:
|
|
|
|
|
return template.replace(
|
|
|
|
|
placeholder,
|
|
|
|
|
"cast(array(select jsonb_array_elements_text(" + aggregateParentReadExpression + "->'" + columnExpression + "')) as " + column.getColumnDefinition() + ')'
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
default:
|
|
|
|
|
return template.replace(
|
|
|
|
|
placeholder,
|
|
|
|
|
"cast(" + aggregateParentReadExpression + "->>'" + columnExpression + "' as " + column.getColumnDefinition() + ')'
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
throw new IllegalArgumentException( "Unsupported aggregate SQL type: " + aggregateColumnTypeCode );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static String jsonCustomWriteExpression(String customWriteExpression, JdbcMapping jdbcMapping) {
|
|
|
|
|
final int sqlTypeCode = jdbcMapping.getJdbcType().getDefaultSqlTypeCode();
|
|
|
|
|
switch ( sqlTypeCode ) {
|
|
|
|
|
case BINARY:
|
|
|
|
|
case VARBINARY:
|
|
|
|
|
case LONG32VARBINARY:
|
|
|
|
|
// We encode binary data as hex
|
|
|
|
|
return "to_jsonb(encode(" + customWriteExpression + ",'hex'))";
|
|
|
|
|
case ARRAY:
|
|
|
|
|
final BasicPluralType<?, ?> pluralType = (BasicPluralType<?, ?>) jdbcMapping;
|
|
|
|
|
switch ( pluralType.getElementType().getJdbcType().getDefaultSqlTypeCode() ) {
|
|
|
|
|
case BINARY:
|
|
|
|
|
case VARBINARY:
|
|
|
|
|
case LONG32VARBINARY:
|
|
|
|
|
// We encode binary data as hex
|
|
|
|
|
return "to_jsonb(array(select encode(unnest(" + customWriteExpression + "),'hex')))";
|
|
|
|
|
default:
|
|
|
|
|
return "to_jsonb(" + customWriteExpression + ")";
|
|
|
|
|
}
|
|
|
|
|
default:
|
|
|
|
|
return "to_jsonb(" + customWriteExpression + ")";
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public String aggregateComponentAssignmentExpression(
|
|
|
|
|
String aggregateParentAssignmentExpression,
|
|
|
|
|
String columnExpression,
|
|
|
|
|
int aggregateColumnTypeCode,
|
|
|
|
|
Column column) {
|
|
|
|
|
switch ( aggregateColumnTypeCode ) {
|
|
|
|
|
case JSON:
|
|
|
|
|
case JSON_ARRAY:
|
|
|
|
|
// For JSON we always have to replace the whole object
|
|
|
|
|
return aggregateParentAssignmentExpression;
|
|
|
|
|
}
|
|
|
|
|
throw new IllegalArgumentException( "Unsupported aggregate SQL type: " + aggregateColumnTypeCode );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public boolean requiresAggregateCustomWriteExpressionRenderer(int aggregateSqlTypeCode) {
|
|
|
|
|
switch ( aggregateSqlTypeCode ) {
|
|
|
|
|
case JSON:
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public WriteExpressionRenderer aggregateCustomWriteExpressionRenderer(
|
|
|
|
|
SelectableMapping aggregateColumn,
|
|
|
|
|
SelectableMapping[] columnsToUpdate,
|
|
|
|
|
TypeConfiguration typeConfiguration) {
|
|
|
|
|
final int aggregateSqlTypeCode = aggregateColumn.getJdbcMapping().getJdbcType().getDefaultSqlTypeCode();
|
|
|
|
|
switch ( aggregateSqlTypeCode ) {
|
|
|
|
|
case JSON:
|
|
|
|
|
return jsonAggregateColumnWriter( aggregateColumn, columnsToUpdate );
|
|
|
|
|
}
|
|
|
|
|
throw new IllegalArgumentException( "Unsupported aggregate SQL type: " + aggregateSqlTypeCode );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private WriteExpressionRenderer jsonAggregateColumnWriter(
|
|
|
|
|
SelectableMapping aggregateColumn,
|
|
|
|
|
SelectableMapping[] columns) {
|
|
|
|
|
return new RootJsonWriteExpression( aggregateColumn, columns );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
interface JsonWriteExpression {
|
|
|
|
|
void append(
|
|
|
|
|
SqlAppender sb,
|
|
|
|
|
String path,
|
|
|
|
|
SqlAstTranslator<?> translator,
|
|
|
|
|
AggregateColumnWriteExpression expression);
|
|
|
|
|
}
|
|
|
|
|
private static class AggregateJsonWriteExpression implements JsonWriteExpression {
|
|
|
|
|
private final LinkedHashMap<String, JsonWriteExpression> subExpressions = new LinkedHashMap<>();
|
|
|
|
|
|
|
|
|
|
protected void initializeSubExpressions(SelectableMapping[] columns) {
|
|
|
|
|
for ( SelectableMapping column : columns ) {
|
|
|
|
|
final SelectablePath selectablePath = column.getSelectablePath();
|
|
|
|
|
final SelectablePath[] parts = selectablePath.getParts();
|
|
|
|
|
AggregateJsonWriteExpression currentAggregate = this;
|
|
|
|
|
for ( int i = 1; i < parts.length - 1; i++ ) {
|
|
|
|
|
currentAggregate = (AggregateJsonWriteExpression) currentAggregate.subExpressions.computeIfAbsent(
|
|
|
|
|
parts[i].getSelectableName(),
|
|
|
|
|
k -> new AggregateJsonWriteExpression()
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
final String customWriteExpression = column.getWriteExpression();
|
|
|
|
|
currentAggregate.subExpressions.put(
|
|
|
|
|
parts[parts.length - 1].getSelectableName(),
|
|
|
|
|
new BasicJsonWriteExpression(
|
|
|
|
|
column,
|
|
|
|
|
jsonCustomWriteExpression( customWriteExpression, column.getJdbcMapping() )
|
|
|
|
|
)
|
|
|
|
|
);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void append(
|
|
|
|
|
SqlAppender sb,
|
|
|
|
|
String path,
|
|
|
|
|
SqlAstTranslator<?> translator,
|
|
|
|
|
AggregateColumnWriteExpression expression) {
|
|
|
|
|
sb.append( "||jsonb_build_object" );
|
|
|
|
|
char separator = '(';
|
|
|
|
|
for ( Map.Entry<String, JsonWriteExpression> entry : subExpressions.entrySet() ) {
|
|
|
|
|
final String column = entry.getKey();
|
|
|
|
|
final JsonWriteExpression value = entry.getValue();
|
|
|
|
|
final String subPath = path + "->'" + column + "'";
|
|
|
|
|
sb.append( separator );
|
|
|
|
|
if ( value instanceof AggregateJsonWriteExpression ) {
|
|
|
|
|
sb.append( '\'' );
|
|
|
|
|
sb.append( column );
|
|
|
|
|
sb.append( "',coalesce(" );
|
|
|
|
|
sb.append( subPath );
|
|
|
|
|
sb.append( ",'{}')" );
|
|
|
|
|
value.append( sb, subPath, translator, expression );
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
value.append( sb, subPath, translator, expression );
|
|
|
|
|
}
|
|
|
|
|
separator = ',';
|
|
|
|
|
}
|
|
|
|
|
sb.append( ')' );
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
private static class RootJsonWriteExpression extends AggregateJsonWriteExpression
|
|
|
|
|
implements WriteExpressionRenderer {
|
|
|
|
|
private final boolean nullable;
|
|
|
|
|
private final String path;
|
|
|
|
|
|
|
|
|
|
RootJsonWriteExpression(SelectableMapping aggregateColumn, SelectableMapping[] columns) {
|
|
|
|
|
this.nullable = aggregateColumn.isNullable();
|
|
|
|
|
this.path = aggregateColumn.getSelectionExpression();
|
|
|
|
|
initializeSubExpressions( columns );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void render(
|
|
|
|
|
SqlAppender sqlAppender,
|
|
|
|
|
SqlAstTranslator<?> translator,
|
|
|
|
|
AggregateColumnWriteExpression aggregateColumnWriteExpression,
|
|
|
|
|
String qualifier) {
|
|
|
|
|
final String basePath;
|
|
|
|
|
if ( qualifier == null || qualifier.isBlank() ) {
|
|
|
|
|
basePath = path;
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
basePath = qualifier + "." + path;
|
|
|
|
|
}
|
|
|
|
|
if ( nullable ) {
|
|
|
|
|
sqlAppender.append( "coalesce(" );
|
|
|
|
|
sqlAppender.append( basePath );
|
|
|
|
|
sqlAppender.append( ",'{}')" );
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
sqlAppender.append( basePath );
|
|
|
|
|
}
|
|
|
|
|
append( sqlAppender, basePath, translator, aggregateColumnWriteExpression );
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
private static class BasicJsonWriteExpression implements JsonWriteExpression {
|
|
|
|
|
|
|
|
|
|
private final SelectableMapping selectableMapping;
|
|
|
|
|
private final String customWriteExpressionStart;
|
|
|
|
|
private final String customWriteExpressionEnd;
|
|
|
|
|
|
|
|
|
|
BasicJsonWriteExpression(SelectableMapping selectableMapping, String customWriteExpression) {
|
|
|
|
|
this.selectableMapping = selectableMapping;
|
|
|
|
|
if ( customWriteExpression.equals( "?" ) ) {
|
|
|
|
|
this.customWriteExpressionStart = "";
|
|
|
|
|
this.customWriteExpressionEnd = "";
|
|
|
|
|
}
|
|
|
|
|
else {
|
|
|
|
|
final String[] parts = StringHelper.split( "?", customWriteExpression );
|
|
|
|
|
assert parts.length == 2;
|
|
|
|
|
this.customWriteExpressionStart = parts[0];
|
|
|
|
|
this.customWriteExpressionEnd = parts[1];
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@Override
|
|
|
|
|
public void append(
|
|
|
|
|
SqlAppender sb,
|
|
|
|
|
String path,
|
|
|
|
|
SqlAstTranslator<?> translator,
|
|
|
|
|
AggregateColumnWriteExpression expression) {
|
|
|
|
|
sb.append( '\'' );
|
|
|
|
|
sb.append( selectableMapping.getSelectableName() );
|
|
|
|
|
sb.append( "'," );
|
|
|
|
|
sb.append( customWriteExpressionStart );
|
|
|
|
|
// We use NO_UNTYPED here so that expressions which require type inference are casted explicitly,
|
|
|
|
|
// since we don't know how the custom write expression looks like where this is embedded,
|
|
|
|
|
// so we have to be pessimistic and avoid ambiguities
|
|
|
|
|
translator.render( expression.getValueExpression( selectableMapping ), SqlAstNodeRenderingMode.NO_UNTYPED );
|
|
|
|
|
sb.append( customWriteExpressionEnd );
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
}
|
Christian Beikov