gorm.tools.databinding.RelaxedConversionService.groovy Maven / Gradle / Ivy
/*
* Copyright 2020 Yak.Works - Licensed under the Apache License, Version 2.0 (the "License")
* You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
*/
package gorm.tools.databinding
import groovy.transform.CompileStatic
import org.springframework.core.convert.ConversionFailedException
import org.springframework.core.convert.ConversionService
import org.springframework.core.convert.TypeDescriptor
import org.springframework.core.convert.converter.Converter
import org.springframework.core.convert.converter.ConverterFactory
import org.springframework.core.convert.support.GenericConversionService
import org.springframework.util.Assert
/**
* copied in from old spring boot 1.5 code.
* Allows for case insensitive enums right now but more can be added.
*/
@CompileStatic
class RelaxedConversionService implements ConversionService {
private final ConversionService conversionService
private final GenericConversionService additionalConverters
/**
* Create a new {@link RelaxedConversionService} instance.
* @param conversionService and option root conversion service
*/
RelaxedConversionService(ConversionService conversionService) {
this.conversionService = conversionService
this.additionalConverters = new GenericConversionService()
// DefaultConversionService.addCollectionConverters(this.additionalConverters)
this.additionalConverters.addConverterFactory(new StringToEnumIgnoringCaseConverterFactory())
}
@Override
public boolean canConvert(Class sourceType, Class targetType) {
return (this.conversionService != null &&
this.conversionService.canConvert(sourceType, targetType)) ||
this.additionalConverters.canConvert(sourceType, targetType)
}
@Override
public boolean canConvert(TypeDescriptor sourceType, TypeDescriptor targetType) {
return (this.conversionService != null &&
this.conversionService.canConvert(sourceType, targetType)) ||
this.additionalConverters.canConvert(sourceType, targetType)
}
@Override
@SuppressWarnings("unchecked")
public T convert(Object source, Class targetType) {
Assert.notNull(targetType, "The targetType to convert to cannot be null")
return (T) convert(source, TypeDescriptor.forObject(source),
TypeDescriptor.valueOf(targetType))
}
@Override
@SuppressWarnings(['EmptyCatchBlock'])
public Object convert(Object source, TypeDescriptor sourceType,
TypeDescriptor targetType) {
if (this.conversionService != null) {
try {
return this.conversionService.convert(source, sourceType, targetType)
}
catch (ConversionFailedException ex) {
// Ignore and try the additional converters
}
}
return this.additionalConverters.convert(source, sourceType, targetType)
}
/**
* Clone of Spring's package private StringToEnumConverterFactory, but ignoring the
* case of the source.
*/
@CompileStatic
private static class StringToEnumIgnoringCaseConverterFactory
implements ConverterFactory {
@Override
public Converter getConverter(Class targetType) {
Class enumType = targetType
while (enumType != null && !enumType.isEnum()) {
enumType = enumType.getSuperclass()
}
Assert.notNull(enumType, "The target type " + targetType.getName()
+ " does not refer to an enum")
return new StringToEnum(enumType)
}
@CompileStatic
private class StringToEnum implements Converter {
private final Class enumType
StringToEnum(Class enumType) {
this.enumType = enumType
}
@Override
public T convert(String source) {
if (source.length() == 0) {
// It's an empty enum identifier: reset the enum value to null.
return null
}
source = source.trim()
for (T candidate : (Set)EnumSet.allOf(this.enumType)) {
RelaxedNames names = new RelaxedNames(
candidate.name().replace("_", "-").toLowerCase())
for (String name : names) {
if (name == source) {
return candidate
}
}
if (candidate.name().equalsIgnoreCase(source)) {
return candidate
}
}
throw new IllegalArgumentException("No enum constant "
+ this.enumType.getCanonicalName() + "." + source)
}
}
}
}