A. Create a class definition method
public class MyMethod extends AbstractMethod { @Override public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) { //自定义sql String sql = "delete from "+tableInfo.getTableName(); //mapper接口的方法名 String method = "deleteAllData"; SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass); return addDeleteMappedStatement(mapperClass,method,sqlSource); } }
II. Custom SQL injector
@Component public class MyInjector extends DefaultSqlInjector { @Override public List<AbstractMethod> getMethodList(Class<?> mapperClass) { List<AbstractMethod> methodList = super.getMethodList(mapperClass); methodList.add(new MyMethod()); return methodList; } }
III. Adding just defined method in your class Mapper
@Repository public interface UserMapper extends BaseMapper<User> { int deleteAllData(); }
IV. Test
@Test public void testInject(){ userMapper.deleteAllData(); }
operation result: