Skip to content
Closed
Changes from all commits
Commits
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
Original file line number Diff line number Diff line change
@@ -1,15 +1,12 @@
package io.quarkus.hibernate.orm.runtime.customized;

import java.util.List;
import java.util.Locale;
import java.util.Set;
import java.util.concurrent.atomic.AtomicBoolean;

import org.hibernate.boot.spi.MetadataImplementor;
import org.hibernate.mapping.Collection;
import org.hibernate.mapping.Column;
import org.hibernate.mapping.PersistentClass;
import org.hibernate.mapping.Property;
import org.hibernate.type.SqlTypes;
import org.hibernate.type.descriptor.java.spi.JsonJavaType;
import org.hibernate.type.descriptor.java.spi.XmlJavaType;
import org.jboss.logging.Logger;

import io.quarkus.arc.ArcContainer;
Expand Down Expand Up @@ -66,30 +63,33 @@ private static String message(String puName, String type) {
}

public static boolean hasJsonProperties(MetadataImplementor metadata) {
return hasXxxProperties(metadata, Set.of(SqlTypes.JSON, SqlTypes.JSON_ARRAY));
AtomicBoolean hasJsonProperties = new AtomicBoolean(false);
metadata.getTypeConfiguration().getJavaTypeRegistry().forEachDescriptor(javaType -> {
if (javaType instanceof JsonJavaType<?>) {
hasJsonProperties.set(true);
}
});
if (hasJsonProperties.get()) {
return true;
} else {
// for JSON_ARRAY we need to check the jdbc type registry instead
return metadata.getTypeConfiguration().getJdbcTypeRegistry().hasRegisteredDescriptor(SqlTypes.JSON_ARRAY);
}
}

public static boolean hasXmlProperties(MetadataImplementor metadata) {
return hasXxxProperties(metadata, Set.of(SqlTypes.SQLXML, SqlTypes.XML_ARRAY));
}

private static boolean hasXxxProperties(MetadataImplementor metadata, Set<Integer> sqlTypeCodes) {
for (PersistentClass persistentClass : metadata.getEntityBindings()) {
for (Property property : persistentClass.getProperties()) {
List<Column> columns = property.getColumns();
if (columns.isEmpty()) {
if (property.getValue() instanceof Collection c) {
columns = c.getElement().getColumns();
}
}
for (Column column : columns) {
if (sqlTypeCodes.contains(column.getSqlTypeCode(metadata))) {
return true;
}
}
AtomicBoolean hasXmlProperties = new AtomicBoolean(false);
metadata.getTypeConfiguration().getJavaTypeRegistry().forEachDescriptor(javaType -> {
if (javaType instanceof XmlJavaType<?>) {
hasXmlProperties.set(true);
}
});
if (hasXmlProperties.get()) {
return true;
} else {
// for XML_ARRAY we need to check the jdbc type registry instead
return metadata.getTypeConfiguration().getJdbcTypeRegistry().hasRegisteredDescriptor(SqlTypes.XML_ARRAY);
}
return false;
}

public void jsonApply(MetadataImplementor metadata, String puName, ArcContainer container,
Expand Down
Loading