MybatisPlus' Sql injector

We already know that in MP,AbstractSqlInjectorWillBaseMapperThe methods in it are injected into the Mybatis container so that these methods can be executed normally.

Then, if we need to expandBaseMapperHow to implement the methods in this article?

Let’s learn from the extension findAll method as an example.

1. Write MyBaseMapper

/*



General mapper interface. When creating other mapper interfaces in the future, the BaseMapper will no longer be inherited, but MyBaseMapper will be inherited.



 */




public interface MyBaseMapper<T> extends BaseMapper<T> {



 



    /*




Query all users



    */



    public List<User> findAll();



}

AI writing code

OthersMapperAll can inherit the mapper, thus achieving unified extension.

like:

public interface UserMapper extends MyBaseMapper<User> {



    /*




Custom findById method



     */



    public User findById(Long id);



}

AI writing code

2. Write MySqlInjector

If you inherit the AbstractSqlInjector directly, the original BaseMapper method will be invalid, so we choose to inherit.DefaultSqlInjectorExpand.

/*



Custom sql injector



 */




public class MySqlInjector extends DefaultSqlInjector {



    @Override




    public List<AbstractMethod> getMethodList() {



 



        List<AbstractMethod> methodList = super.getMethodList();



        // Expand custom methods




        (new FindAll());



        return methodList;



    }



}

AI writing code

3. Write FindAll

public class FindAll extends AbstractMethod {



    @Override




    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {



 



        String sql = "select * from " + ();



        SqlSource sqlSource = (configuration, sql, modelClass);



        return this.addSelectMappedStatement(mapperClass, "findAll", sqlSource, modelClass, tableInfo);



 



    }



}

AI writing code

4. Register to Spring Container

    /*




Custom sql injector



    */



 



   @Bean




   public MySqlInjector mySqlInjector(){



       return new MySqlInjector();



   }

AI writing code

5. Test

@Test




public void testFindAll(){



    List<User> users = this.();



    for (User user : users) {



        (user);



      }



}

AI writing code

The output SQL:

[main] []-[DEBUG] ==> Preparing: select
* from tb_user
[main] []-[DEBUG] ==> Parameters:
[main] []-[DEBUG] <== Total: 10

At this point, we have achieved global expansionSQL Injectionutensil.