MyBatis-Plus：如何通过扩展baseMapper，添加一些自定义的查询方法

一、引言

先来看一段SQLException

### Error updating database.  Cause: com.microsoft.sqlserver.jdbc.SQLServerException: 事务(进程 ID 271)与另一个进程被死锁在 锁 资源上，并且已被选作死锁牺牲品。请重新运行该事务。

由于公司db工程师的建议，针对sqlserver的查询语句尽量添加上with(nolock)来解决上述事务死锁问题；
针对db工程师的建议，如何通过扩展baseMapper，添加一些自定义的查询方法呢？我们以方法如下

二、自定义方法实现

步骤1.继承抽象的注入方法类

/***
 * 针对sqlserver增加with(nolock) 防止事务死锁
 */
public class SelectByIdNoLockMethod extends AbstractMethod {
    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        String sql = "SELECT %s FROM %s with(nolock) WHERE %s=#{%s} %s";
        String method = "selectByIdNoLock";
        SqlSource sqlSource = new RawSqlSource(configuration, String.format(sql,
                sqlSelectColumns(tableInfo, false),
                tableInfo.getTableName(), tableInfo.getKeyColumn(), tableInfo.getKeyProperty(),
                tableInfo.getLogicDeleteSql(true, true)), Object.class);
        return addSelectMappedStatementForTable(mapperClass, method, sqlSource, tableInfo);
    }
}

步骤2.创建注入器，把自定义方法添加到集合当中。

@Component
public class MySqlInjector extends DefaultSqlInjector {

    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass);
        // 添加自定义方法类
        methodList.add(new SelectByIdNoLockMethod());
        return methodList;
    }
}

步骤3.新建一个自定义的Mapper接口继承baseMapper接口，并添加我们扩展的查询

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

    /**
     * 自定义通用方法
     * @return
     */
    T selectByIdNoLock(Serializable id);
}

三、结尾

接下来直接调用就可以发现自动添加了with(nolock)关键字

public interface BusCommonFileLogMapper extends MyMapper<BusCommonFileLog> {

}

@Test
    public void selectByIdNoLock() {
        busCommonFileLogMapper.selectByIdNoLock("1465926655269212162");
    }

使用Mybatis Log Plugin 插件查看打印出来的sql就可以发现已经自动加上with(nolock)关键字啦

四、注意事项

由于是在启动时一次性注入的，所以没办法判断当前数据源，所以使用时需谨慎，只有在支持with(nolock)语法的数据库才能使用