MybatisPlus学习之（四）SQL注入

想让自定义的方法启动时直接注入sql语句到configuration中

参考：https://www.cnblogs.com/liuyangfirst/p/9744011.html

1.在mapper接口中先定义方法

@Mapper
public interface TblEmployeeMapper extends BaseMapper<TblEmployee> {
    /**
     *  自定义deleteAll操作
     *  在启动时 就自动注入，和BaseMapper里面的其他方法一样
     */
    int deleteAll();
}

2.自定义MySqlInjector类继承DefaultSqlInjector

/**
 *  自定义MySqlInjector类
 *  模仿LogicDelete方法继承
 */
public class MySqlInjector extends DefaultSqlInjector {

    public MySqlInjector() {
        super();
    }
    /**
     * 添加  deleteAll方法
     * @param mapperClass
     * @return
     */
    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass) {

        List<AbstractMethod> methodList = super.getMethodList(mapperClass);
        methodList.add(new DeleteAll());

        return methodList;
    }
}

3.创建 DeleteAll 的SQL语句，在添加到MappedStatement中，最后再添加进configuration中

这里直接创建了SQL语句，所以就不需要再新建一个method类继承AbstractMethod方法

public class DeleteAll extends AbstractMethod {

    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo){
		//创建SQL语句
        String sql = "delete from " + tableInfo.getTableName();
        String method = "deleteAll";
        //MySqlMethod mySqlMethod = MySqlMethod.DELETE_ALL;

        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
        return addUpdateMappedStatement(mapperClass, modelClass, method, sqlSource);
    }
}

4.全局配置

再applicationContext.xml的全局配置文件中注入

<!--定义MP的全局策略配置-->
    <bean id="globalConfig" class="com.baomidou.mybatisplus.core.config.GlobalConfig">
        <!--全局配置注入-->
        <property name="dbConfig" ref="dbConfig"/>

        <!-- 1.定义自定义注入器 -->
        <property name="sqlInjector" ref="mySqlInjector"></property>

        <!-- 2.注入逻辑删除 -->
        <!--<property name="sqlInjector" ref="logicSqlInjector"></property>-->

        <!-- 3.字段填充-->
        <property name="metaObjectHandler" ref="myMetaObjectHandler"></property>

    </bean>
    
    <!-- 定义自定义注入器 -->
    <bean id="mySqlInjector" class="com.Alice.mp.injecter.MySqlInjector"></bean>
    

5.测试

mapper对象可以直接利用自定义的deleteAll方法

    /**
     * 测试自定义全局操作
     *   deleteAll是自定义的方法，把它注入到 mappersatement中
     */
    @Test
    public void  testMySqlInjector() {
        Integer result = tblEmployeeMapper.deleteAll();
        System.out.println("result: " +result );
    }