sviolet.slate.common.x.conversion.beanutil.BeanizationFactory Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of slate-common Show documentation
Show all versions of slate-common Show documentation
Comprehensive Java library for spring framework (Java8+)
/*
* Copyright (C) 2015-2018 S.Violet
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* Project GitHub: https://github.com/shepherdviolet/slate
* Email: [email protected]
*/
package sviolet.slate.common.x.conversion.beanutil;
import sviolet.thistle.util.conversion.BeanMethodNameUtils;
import sviolet.thistle.util.conversion.PrimitiveUtils;
import sviolet.thistle.util.judge.CheckUtils;
import java.lang.reflect.Method;
import java.util.*;
/**
* Bean矫正器
*
* @author S.Violet
*/
class BeanizationFactory {
private Class templateType;
private BeanConverter converter;
private Map templateProperties;
BeanizationFactory(Class templateType, BeanConverter converter) {
this.templateType = templateType;
this.converter = converter;
init();
}
private void init(){
if (templateType == null) {
throw new NullPointerException("templateType is null");
}
if (converter == null) {
throw new NullPointerException("converter is null");
}
Map> properties = new HashMap<>();
Map definiteProperties = new HashMap<>();
Method[] methods = templateType.getMethods();
if (methods != null) {
for (Method method : methods) {
String methodName = method.getName();
String fieldName = BeanMethodNameUtils.methodToField(methodName);
if (CheckUtils.isEmptyOrBlank(fieldName)) {
continue;
}
Class[] paramTypes = method.getParameterTypes();
if (methodName.startsWith("set")) {
//setter
if (paramTypes.length != 1) {
continue;
}
Set classSet = properties.get(fieldName);
if (classSet == null) {
classSet = new HashSet<>();
properties.put(fieldName, classSet);
}
classSet.add(PrimitiveUtils.toWrapperType(paramTypes[0]));
} else if (methodName.startsWith("get") || methodName.startsWith("is")) {
//getter
if (paramTypes.length != 0) {
continue;
}
if (void.class.isAssignableFrom(method.getReturnType())) {
continue;
}
definiteProperties.put(fieldName, PrimitiveUtils.toWrapperType(method.getReturnType()));
}
}
}
for (Map.Entry entry : definiteProperties.entrySet()) {
Set classSet = new HashSet<>(1);
classSet.add(entry.getValue());
properties.put(entry.getKey(), classSet);
}
this.templateProperties = new HashMap<>(properties.size());
for (Map.Entry> entry : properties.entrySet()) {
Class[] classArray = new Class[entry.getValue().size()];
this.templateProperties.put(entry.getKey(), entry.getValue().toArray(classArray));
}
}
Map beanization(Map map, boolean convert, boolean throwExceptionIfFails){
Map result = new HashMap<>(templateProperties.size());
for (Map.Entry entry : templateProperties.entrySet()) {
String entryKey = entry.getKey();
Object value = map.get(entryKey);
if (value == null) {
continue;
}
Class valueType = value.getClass();
try {
boolean found = false;
for (Class type : entry.getValue()) {
if (type.isAssignableFrom(valueType)) {
if (convert) {
value = converter.onConvert(BeanConverter.Cause.BEANIZATION, value, new Class[]{type});
if (value == null && throwExceptionIfFails) {
throw new MappingRuntimeException("SlateBeanUtils: Error while pre-mapping (check and conversion) Map to " + templateType.getName() + ", field \"" + entryKey + "\" convert failed (In PropMapper for " + valueType.getName() + " to " + type.getName() + "), map data:" + map, null, "java.util.Map", templateType.getName(), entryKey);
}
}
result.put(entryKey, value);
found = true;
break;
}
}
if (found) {
continue;
}
//fallback
if (convert) {
value = converter.onConvert(BeanConverter.Cause.BEANIZATION, value, entry.getValue());
if (value == null && throwExceptionIfFails) {
throw new MappingRuntimeException("SlateBeanUtils: Error while pre-mapping (check and conversion) Map to " + templateType.getName() + ", field \"" + entryKey + "\" convert failed (No PropMapper for " + valueType.getName() + " to" + getClassNames(entry.getValue()) + "), map data:" + map, null, "java.util.Map", templateType.getName(), entryKey);
}
result.put(entryKey, value);
} else if (throwExceptionIfFails){
throw new MappingRuntimeException("SlateBeanUtils: Error while pre-mapping (check and conversion) Map to " + templateType.getName() + ", field \"" + entryKey + "\" convert failed (No PropMapper for " + valueType.getName() + " to" + getClassNames(entry.getValue()) + "), map data:" + map, null, "java.util.Map", templateType.getName(), entryKey);
} else {
result.put(entryKey, null);
}
} catch (MappingRuntimeException e) {
if (throwExceptionIfFails) {
//补上field名
String fieldName = BeanMethodNameUtils.methodToField(entryKey);
e.setFieldName(fieldName);
throw e;
}
} catch (Exception e) {
if (throwExceptionIfFails) {
throw new MappingRuntimeException("SlateBeanUtils: Error while pre-mapping (check and conversion) Map to " + templateType.getName() + ", problem property \"" + entryKey + "\" (No PropMapper for " + valueType.getName() + " to" + getClassNames(entry.getValue()) + "), map data:" + map, e, "java.util.Map", templateType.getName(), entryKey);
}
}
}
return result;
}
private String getClassNames(Class[] classes) {
if (classes.length == 1) {
return " " + classes[0].getName();
}
StringBuilder stringBuilder = new StringBuilder();
for (Class clazz : classes) {
stringBuilder.append(" ");
stringBuilder.append(clazz.getSimpleName());
}
return stringBuilder.toString();
}
}