MybatisParameterHandler

mybatis-plus-core/src/main/java/com/baomidou/mybatisplus/core/MybatisDefaultParameterHandler.java

@@ -1,269 +0,0 @@
-/*
- * Copyright (c) 2011-2020, baomidou (jobob@qq.com).
- * <p>
- * 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
- * <p>
- * https://www.apache.org/licenses/LICENSE-2.0
- * <p>
- * 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.
- */
-package com.baomidou.mybatisplus.core;
-
-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.mapping.*;
-import org.apache.ibatis.reflection.MetaObject;
-import org.apache.ibatis.scripting.defaults.DefaultParameterHandler;
-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.*;
-
-/**
- * 自定义 ParameterHandler 重装构造函数，填充插入方法主键 ID
- *
- * @author hubin
- * @since 2016-03-11
- * @deprecated 3.3.3 {@link MybatisParameterHandler}
- */
-@Deprecated
-public class MybatisDefaultParameterHandler extends DefaultParameterHandler {
-
-    private final TypeHandlerRegistry typeHandlerRegistry;
-    private final MappedStatement mappedStatement;
-    private final Object parameterObject;
-    private final BoundSql boundSql;
-    private final Configuration configuration;
-
-    public MybatisDefaultParameterHandler(MappedStatement mappedStatement, Object parameterObject, BoundSql boundSql) {
-        super(mappedStatement, processParameter(mappedStatement, parameterObject), boundSql);
-        this.mappedStatement = mappedStatement;
-        this.configuration = mappedStatement.getConfiguration();
-        this.typeHandlerRegistry = mappedStatement.getConfiguration().getTypeHandlerRegistry();
-        this.parameterObject = parameterObject;
-        this.boundSql = boundSql;
-    }
-
-    /**
-     * 处理参数
-     *
-     * @param ms              MappedStatement
-     * @param parameterObject 插入数据库对象
-     * @return ignore
-     */
-    public static Object processParameter(MappedStatement ms, Object parameterObject) {
-        /* 只处理插入或更新操作 */
-        if (parameterObject != null
-            && (SqlCommandType.INSERT == ms.getSqlCommandType() || SqlCommandType.UPDATE == ms.getSqlCommandType())) {
-            //检查 parameterObject
-            if (ReflectionKit.isPrimitiveOrWrapper(parameterObject.getClass())
-                || parameterObject.getClass() == String.class) {
-                return parameterObject;
-            }
-            Collection<Object> parameters = getParameters(parameterObject);
-            if (null != parameters) {
-                // 感觉这里可以稍微优化一下，理论上都是同一个.
-                parameters.forEach(obj -> process(ms, obj));
-            } else {
-                process(ms, parameterObject);
-            }
-        }
-        return parameterObject;
-    }
-
-    private static void process(MappedStatement ms, Object parameterObject) {
-        if (parameterObject != null) {
-            TableInfo tableInfo = null;
-            Object entity = parameterObject;
-            if (parameterObject instanceof Map) {
-                Map<?, ?> map = (Map<?, ?>) parameterObject;
-                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(parameterObject.getClass());
-            }
-            Configuration configuration = ms.getConfiguration();
-            if (tableInfo != null) {
-                //到这里就应该转换到实体参数对象了,因为填充和ID处理都是争对实体对象处理的,不用传递原参数对象下去.
-                MetaObject metaObject = configuration.newMetaObject(entity);
-                if (SqlCommandType.INSERT == ms.getSqlCommandType()) {
-                    populateKeys(tableInfo, metaObject, entity, configuration);
-                    insertFill(metaObject, tableInfo, configuration);
-                } else {
-                    updateFill(metaObject, tableInfo, configuration);
-                }
-            }
-        }
-    }
-
-
-    /**
-     * 处理正常批量插入逻辑
-     * <p>
-     * org.apache.ibatis.session.defaults.DefaultSqlSession$StrictMap 该类方法
-     * wrapCollection 实现 StrictMap 封装逻辑
-     * </p>
-     *
-     * @param parameter 插入数据库对象
-     * @return
-     */
-    @SuppressWarnings({"rawtypes", "unchecked"})
-    protected static Collection<Object> getParameters(Object parameter) {
-        Collection<Object> parameters = null;
-        if (parameter instanceof Collection) {
-            parameters = (Collection) parameter;
-        } else if (parameter instanceof Map) {
-            Map parameterMap = (Map) parameter;
-            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;
-    }
-
-    /**
-     * 填充主键
-     *
-     * @param tableInfo  数据库表反射信息
-     * @param metaObject 元数据对象
-     * @param entity     实体信息
-     * @deprecated 3.3.3 {@link #populateKeys(TableInfo, MetaObject, Object, Configuration)}
-     */
-    @Deprecated
-    protected static void populateKeys(TableInfo tableInfo, MetaObject metaObject, Object entity) {
-        populateKeys(tableInfo, metaObject, entity, tableInfo.getConfiguration());
-    }
-
-    protected static void populateKeys(TableInfo tableInfo, MetaObject metaObject, Object entity, Configuration configuration) {
-        final IdType idType = tableInfo.getIdType();
-        final String keyProperty = tableInfo.getKeyProperty();
-        if (StringUtils.isNotBlank(keyProperty) && null != idType && idType.getKey() >= 3) {
-            final IdentifierGenerator identifierGenerator = GlobalConfigUtils.getGlobalConfig(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));
-                }
-            }
-        }
-    }
-
-    /**
-     *
-     * @param metaObject
-     * @param tableInfo
-     * @deprecated 3.3.3 {@link #insertFill(MetaObject, TableInfo, Configuration)}
-     */
-    @Deprecated
-    protected static void insertFill(MetaObject metaObject, TableInfo tableInfo) {
-        insertFill(metaObject, tableInfo, tableInfo.getConfiguration());
-    }
-
-    protected static void insertFill(MetaObject metaObject, TableInfo tableInfo,Configuration configuration) {
-        GlobalConfigUtils.getMetaObjectHandler(configuration).ifPresent(metaObjectHandler -> {
-            if (metaObjectHandler.openInsertFill()) {
-                if (tableInfo.isWithInsertFill()) {
-                    metaObjectHandler.insertFill(metaObject);
-                } else {
-                    // 兼容旧操作 id类型为input或none的要用填充器处理一下
-                    if (metaObjectHandler.compatibleFillId()) {
-                        String keyProperty = tableInfo.getKeyProperty();
-                        if (StringUtils.isNotBlank(keyProperty)) {
-                            Object value = metaObject.getValue(keyProperty);
-                            if (value == null && (IdType.NONE == tableInfo.getIdType() || IdType.INPUT == tableInfo.getIdType())) {
-                                metaObjectHandler.insertFill(metaObject);
-                            }
-                        }
-                    }
-                }
-            }
-        });
-    }
-
-    /**
-     *
-     * @param metaObject
-     * @param tableInfo
-     * @deprecated 3.3.3 {@link #updateFill(MetaObject, TableInfo, Configuration)}
-     */
-    protected static void updateFill(MetaObject metaObject, TableInfo tableInfo) {
-        updateFill(metaObject, tableInfo, tableInfo.getConfiguration());
-    }
-
-    protected static void updateFill(MetaObject metaObject, TableInfo tableInfo, Configuration configuration) {
-        GlobalConfigUtils.getMetaObjectHandler(configuration).ifPresent(metaObjectHandler -> {
-            if (metaObjectHandler.openUpdateFill() && tableInfo.isWithUpdateFill()) {
-                metaObjectHandler.updateFill(metaObject);
-            }
-        });
-    }
-
-    @Override
-    @SuppressWarnings("unchecked")
-    public void setParameters(PreparedStatement ps) {
-        ErrorContext.instance().activity("setting parameters").object(mappedStatement.getParameterMap().getId());
-        List<ParameterMapping> parameterMappings = 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 (boundSql.hasAdditionalParameter(propertyName)) { // issue #448 ask first for additional params
-                        value = boundSql.getAdditionalParameter(propertyName);
-                    } else if (parameterObject == null) {
-                        value = null;
-                    } else if (typeHandlerRegistry.hasTypeHandler(parameterObject.getClass())) {
-                        value = parameterObject;
-                    } else {
-                        MetaObject metaObject = configuration.newMetaObject(parameterObject);
-                        value = metaObject.getValue(propertyName);
-                    }
-                    TypeHandler typeHandler = parameterMapping.getTypeHandler();
-                    JdbcType jdbcType = parameterMapping.getJdbcType();
-                    if (value == null && jdbcType == null) {
-                        jdbcType = 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);
-                    }
-                }
-            }
-        }
-    }
-}