想让自定义的方法启动时直接注入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 );
}