mirror of
https://gitee.com/best_handsome/mybatis-plus-join
synced 2025-07-11 00:02:22 +08:00
update
This commit is contained in:
admin
parent
03fac41eb3
commit
2e712036d9
@ -1,5 +1,6 @@
|
||||
package com.github.mybatisplus.interceptor;
|
||||
|
||||
import com.baomidou.mybatisplus.core.MybatisParameterHandler;
|
||||
import com.baomidou.mybatisplus.core.toolkit.ClassUtils;
|
||||
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
||||
import com.github.mybatisplus.toolkit.Constant;
|
||||
|
||||
@ -1,205 +0,0 @@
|
||||
package com.github.mybatisplus.interceptor;
|
||||
|
||||
import com.baomidou.mybatisplus.annotation.IdType;
|
||||
import com.baomidou.mybatisplus.core.incrementer.IdentifierGenerator;
|
||||
import com.baomidou.mybatisplus.core.metadata.TableInfo;
|
||||
import com.baomidou.mybatisplus.core.metadata.TableInfoHelper;
|
||||
import com.baomidou.mybatisplus.core.toolkit.Constants;
|
||||
import com.baomidou.mybatisplus.core.toolkit.GlobalConfigUtils;
|
||||
import com.baomidou.mybatisplus.core.toolkit.ReflectionKit;
|
||||
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
||||
import org.apache.ibatis.executor.ErrorContext;
|
||||
import org.apache.ibatis.executor.parameter.ParameterHandler;
|
||||
import org.apache.ibatis.mapping.*;
|
||||
import org.apache.ibatis.reflection.MetaObject;
|
||||
import org.apache.ibatis.session.Configuration;
|
||||
import org.apache.ibatis.type.JdbcType;
|
||||
import org.apache.ibatis.type.TypeException;
|
||||
import org.apache.ibatis.type.TypeHandler;
|
||||
import org.apache.ibatis.type.TypeHandlerRegistry;
|
||||
|
||||
import java.sql.PreparedStatement;
|
||||
import java.sql.SQLException;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collection;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
/**
|
||||
* copy {@link com.baomidou.mybatisplus.core.MybatisParameterHandler}
|
||||
* <p>
|
||||
* 兼容3.4以下
|
||||
*/
|
||||
public class MybatisParameterHandler implements ParameterHandler {
|
||||
|
||||
private final TypeHandlerRegistry typeHandlerRegistry;
|
||||
private final MappedStatement mappedStatement;
|
||||
private final Object parameterObject;
|
||||
private final BoundSql boundSql;
|
||||
private final Configuration configuration;
|
||||
private final SqlCommandType sqlCommandType;
|
||||
|
||||
public MybatisParameterHandler(MappedStatement mappedStatement, Object parameter, BoundSql boundSql) {
|
||||
this.typeHandlerRegistry = mappedStatement.getConfiguration().getTypeHandlerRegistry();
|
||||
this.mappedStatement = mappedStatement;
|
||||
this.boundSql = boundSql;
|
||||
this.configuration = mappedStatement.getConfiguration();
|
||||
this.sqlCommandType = mappedStatement.getSqlCommandType();
|
||||
this.parameterObject = processParameter(parameter);
|
||||
}
|
||||
|
||||
public Object processParameter(Object parameter) {
|
||||
/* 只处理插入或更新操作 */
|
||||
if (parameter != null
|
||||
&& (SqlCommandType.INSERT == this.sqlCommandType || SqlCommandType.UPDATE == this.sqlCommandType)) {
|
||||
//检查 parameterObject
|
||||
if (ReflectionKit.isPrimitiveOrWrapper(parameter.getClass())
|
||||
|| parameter.getClass() == String.class) {
|
||||
return parameter;
|
||||
}
|
||||
Collection<Object> parameters = getParameters(parameter);
|
||||
if (null != parameters) {
|
||||
// 感觉这里可以稍微优化一下，理论上都是同一个.
|
||||
parameters.forEach(this::process);
|
||||
} else {
|
||||
process(parameter);
|
||||
}
|
||||
}
|
||||
return parameter;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Object getParameterObject() {
|
||||
return this.parameterObject;
|
||||
}
|
||||
|
||||
private void process(Object parameter) {
|
||||
if (parameter != null) {
|
||||
TableInfo tableInfo = null;
|
||||
Object entity = parameter;
|
||||
if (parameter instanceof Map) {
|
||||
Map<?, ?> map = (Map<?, ?>) parameter;
|
||||
if (map.containsKey(Constants.ENTITY)) {
|
||||
Object et = map.get(Constants.ENTITY);
|
||||
if (et != null) {
|
||||
entity = et;
|
||||
tableInfo = TableInfoHelper.getTableInfo(entity.getClass());
|
||||
}
|
||||
}
|
||||
} else {
|
||||
tableInfo = TableInfoHelper.getTableInfo(parameter.getClass());
|
||||
}
|
||||
if (tableInfo != null) {
|
||||
//到这里就应该转换到实体参数对象了,因为填充和ID处理都是争对实体对象处理的,不用传递原参数对象下去.
|
||||
MetaObject metaObject = this.configuration.newMetaObject(entity);
|
||||
if (SqlCommandType.INSERT == this.sqlCommandType) {
|
||||
populateKeys(tableInfo, metaObject, entity);
|
||||
insertFill(metaObject, tableInfo);
|
||||
} else {
|
||||
updateFill(metaObject, tableInfo);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
protected void populateKeys(TableInfo tableInfo, MetaObject metaObject, Object entity) {
|
||||
final IdType idType = tableInfo.getIdType();
|
||||
final String keyProperty = tableInfo.getKeyProperty();
|
||||
if (StringUtils.isNotBlank(keyProperty) && null != idType && idType.getKey() >= 3) {
|
||||
final IdentifierGenerator identifierGenerator = GlobalConfigUtils.getGlobalConfig(this.configuration).getIdentifierGenerator();
|
||||
Object idValue = metaObject.getValue(keyProperty);
|
||||
if (StringUtils.checkValNull(idValue)) {
|
||||
if (idType.getKey() == IdType.ASSIGN_ID.getKey()) {
|
||||
if (Number.class.isAssignableFrom(tableInfo.getKeyType())) {
|
||||
metaObject.setValue(keyProperty, identifierGenerator.nextId(entity));
|
||||
} else {
|
||||
metaObject.setValue(keyProperty, identifierGenerator.nextId(entity).toString());
|
||||
}
|
||||
} else if (idType.getKey() == IdType.ASSIGN_UUID.getKey()) {
|
||||
metaObject.setValue(keyProperty, identifierGenerator.nextUUID(entity));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
protected void insertFill(MetaObject metaObject, TableInfo tableInfo) {
|
||||
GlobalConfigUtils.getMetaObjectHandler(this.configuration).ifPresent(metaObjectHandler -> {
|
||||
if (metaObjectHandler.openInsertFill() && tableInfo.isWithInsertFill()) {
|
||||
metaObjectHandler.insertFill(metaObject);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
protected void updateFill(MetaObject metaObject, TableInfo tableInfo) {
|
||||
GlobalConfigUtils.getMetaObjectHandler(this.configuration).ifPresent(metaObjectHandler -> {
|
||||
if (metaObjectHandler.openUpdateFill() && tableInfo.isWithUpdateFill()) {
|
||||
metaObjectHandler.updateFill(metaObject);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
/**
|
||||
* 处理正常批量插入逻辑
|
||||
* <p>
|
||||
* org.apache.ibatis.session.defaults.DefaultSqlSession$StrictMap 该类方法
|
||||
* wrapCollection 实现 StrictMap 封装逻辑
|
||||
* </p>
|
||||
*
|
||||
* @return 集合参数
|
||||
*/
|
||||
@SuppressWarnings({"rawtypes", "unchecked"})
|
||||
protected Collection<Object> getParameters(Object parameterObject) {
|
||||
Collection<Object> parameters = null;
|
||||
if (parameterObject instanceof Collection) {
|
||||
parameters = (Collection) parameterObject;
|
||||
} else if (parameterObject instanceof Map) {
|
||||
Map parameterMap = (Map) parameterObject;
|
||||
if (parameterMap.containsKey("collection")) {
|
||||
parameters = (Collection) parameterMap.get("collection");
|
||||
} else if (parameterMap.containsKey("list")) {
|
||||
parameters = (List) parameterMap.get("list");
|
||||
} else if (parameterMap.containsKey("array")) {
|
||||
parameters = Arrays.asList((Object[]) parameterMap.get("array"));
|
||||
}
|
||||
}
|
||||
return parameters;
|
||||
}
|
||||
|
||||
@Override
|
||||
@SuppressWarnings("unchecked")
|
||||
public void setParameters(PreparedStatement ps) {
|
||||
ErrorContext.instance().activity("setting parameters").object(this.mappedStatement.getParameterMap().getId());
|
||||
List<ParameterMapping> parameterMappings = this.boundSql.getParameterMappings();
|
||||
if (parameterMappings != null) {
|
||||
for (int i = 0; i < parameterMappings.size(); i++) {
|
||||
ParameterMapping parameterMapping = parameterMappings.get(i);
|
||||
if (parameterMapping.getMode() != ParameterMode.OUT) {
|
||||
Object value;
|
||||
String propertyName = parameterMapping.getProperty();
|
||||
if (this.boundSql.hasAdditionalParameter(propertyName)) { // issue #448 ask first for additional params
|
||||
value = this.boundSql.getAdditionalParameter(propertyName);
|
||||
} else if (this.parameterObject == null) {
|
||||
value = null;
|
||||
} else if (this.typeHandlerRegistry.hasTypeHandler(this.parameterObject.getClass())) {
|
||||
value = parameterObject;
|
||||
} else {
|
||||
MetaObject metaObject = this.configuration.newMetaObject(this.parameterObject);
|
||||
value = metaObject.getValue(propertyName);
|
||||
}
|
||||
TypeHandler typeHandler = parameterMapping.getTypeHandler();
|
||||
JdbcType jdbcType = parameterMapping.getJdbcType();
|
||||
if (value == null && jdbcType == null) {
|
||||
jdbcType = this.configuration.getJdbcTypeForNull();
|
||||
}
|
||||
try {
|
||||
typeHandler.setParameter(ps, i + 1, value, jdbcType);
|
||||
} catch (TypeException | SQLException e) {
|
||||
throw new TypeException("Could not set parameters for mapping: " + parameterMapping + ". Cause: " + e, e);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user