Mybatis-plus 自动装配

image.png

Mysql 分页

image.png

mybatis-plus 自动填充数据（默认开启）

image.png
image.png

多租户自动配置

image.png

  1. package com.np.pm.common.core.properties;
  3. import lombok.Getter;
  4. import lombok.Setter;
  5. import org.springframework.boot.context.properties.ConfigurationProperties;
  6. import org.springframework.cloud.context.config.annotation.RefreshScope;
  8. import java.util.ArrayList;
  9. import java.util.List;
  11. /**
  12.  * 多租户配置
  13.  *
  14.  * @author tangxingpeng
  15.  * @date 2021/1/26 23:32
  16.  */
  17. @Setter
  18. @Getter
  19. @ConfigurationProperties(prefix = "pm.tenant")
  20. @RefreshScope
  21. public class TenantProperties {
  23.     /**
  24.      * 是否开启多租户
  25.      */
  26.     private Boolean enable = false;
  28.     /**
  29.      * 配置不进行多租户隔离的表名
  30.      */
  31.     private List<String> ignoreTables = new ArrayList<>();
  33.     /**
  34.      * 配置不进行多租户隔离的sql
  35.      * 需要配置mapper的全路径如：com.central.user.mapper.SysUserMapper.findList
  36.      */
  37.     private List<String> ignoreSqls = new ArrayList<>();
  39. }

image.png

  1. package com.np.pm.common.datasource.configure;
  3. import com.baomidou.mybatisplus.extension.plugins.handler.TenantLineHandler;
  4. import com.np.pm.common.core.context.TenantContextHolder;
  5. import com.np.pm.common.core.properties.TenantProperties;
  6. import net.sf.jsqlparser.expression.Expression;
  7. import net.sf.jsqlparser.expression.NullValue;
  8. import net.sf.jsqlparser.expression.StringValue;
  9. import org.springframework.beans.factory.annotation.Autowired;
  10. import org.springframework.boot.context.properties.EnableConfigurationProperties;
  11. import org.springframework.context.annotation.Bean;
  13. /**
  14.  * 多租户自动配置
  15.  *
  16.  * @author tangxingpeng
  17.  * @date 2021/1/27 14:17
  18.  */
  19. @EnableConfigurationProperties(TenantProperties.class)
  20. public class TenantAutoConfigure {
  22.     @Autowired
  23.     private TenantProperties tenantProperties;
  25.     @Bean
  26.     public TenantLineHandler tenantHandler() {
  27.         return new TenantLineHandler() {
  29.             /**
  30.              * 获取租户id
  31.              */
  32.             @Override
  33.             public Expression getTenantId() {
  34.                 String tenant = TenantContextHolder.getTenant();
  35.                 if (tenant != null) {
  36.                     return new StringValue(TenantContextHolder.getTenant());
  37.                 }
  38.                 return new NullValue();
  39.             }
  41.             /**
  42.              * 获取租户列名
  43.              */
  44.             @Override
  45.             public String getTenantIdColumn() {
  46.                 return "tenant_id";
  47.             }
  49.             /**
  50.              * 过滤不需要根据租户隔离的表
  51.              * @param tableName 表名
  52.              */
  53.             @Override
  54.             public boolean ignoreTable(String tableName) {
  55.                 return tenantProperties.getIgnoreTables().stream().anyMatch(
  56.                         (e) -> e.equalsIgnoreCase(tableName)
  57.                 );
  58.             }
  59.         };
  60.     }
  61. }

update增强

项目的mybatis-plus默认的所有策略都是NOT_NULL(见pm-configure)

image.png

如果遇到向导更新null值时会忽略更新的问题

  1. 方式一（手动更新null）：

         UpdateWrapper<T> tLambdaUpdateWrapper = new UpdateWrapper<T>();
     tLambdaUpdateWrapper.set("field", null);
     tLambdaUpdateWrapper.eq(idTableFeild.value(), id);
     return SqlHelper.retBool(baseMapper.update(null, tLambdaUpdateWrapper));

  2. 方式二（前端传全部数据，后端全更新）：

官方文档：https://mp.baomidou.com/guide/sql-injector.html

public void updateEntity(SysMenu entity) {
    entity = this.beforeTreeSave(entity);
    baseMapper.updateAllById(entity);
}

image.png
image.png

package com.np.pm.common.datasource.inteceptor;

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.np.pm.common.datasource.inteceptor.methods.UpdateAllById;

import java.util.List;

/**
 * @author tangxingpeng
 * @date 2021/8/3 10:14
 */
public class UpdateSqlInjector extends DefaultSqlInjector {

    @Override
    public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
        List<AbstractMethod> methodList = super.getMethodList(mapperClass);
        methodList.add(new UpdateAllById());
        return methodList;
    }

}

package com.np.pm.common.datasource.inteceptor.methods;

import com.baomidou.mybatisplus.core.enums.SqlMethod;
import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.core.toolkit.sql.SqlScriptUtils;
import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.SqlSource;

import java.util.Objects;

import static java.util.stream.Collectors.joining;

/**
 * 根据 ID 更新所有字段
 *
 * @author tangxingpeng
 * @date 2021/8/3 10:19
 */
public class UpdateAllById extends AbstractMethod {

    public UpdateAllById() {
    }

    @Override
    public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
        SqlMethod sqlMethod = SqlMethod.UPDATE_BY_ID;
        final String additional = optlockVersion(tableInfo) + tableInfo.getLogicDeleteSql(true, true);
        String sql = String.format(sqlMethod.getSql(), tableInfo.getTableName(),
                sqlSet(tableInfo.isWithLogicDelete(), false, tableInfo, false, ENTITY, ENTITY_DOT),
                tableInfo.getKeyColumn(), ENTITY_DOT + tableInfo.getKeyProperty(), additional);
        SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
        return addUpdateMappedStatement(mapperClass, modelClass, getMethod(sqlMethod), sqlSource);
    }


    /**
     * SQL 更新 set 语句
     *
     * @param logic  是否逻辑删除注入器
     * @param ew     是否存在 UpdateWrapper 条件
     * @param table  表信息
     * @param alias  别名
     * @param prefix 前缀
     * @return sql
     */
    @Override
    protected String sqlSet(boolean logic, boolean ew, TableInfo table, boolean judgeAliasNull, final String alias,
                            final String prefix) {
        String sqlScript = getAllSqlSet(table, logic, prefix);
        if (judgeAliasNull) {
            sqlScript = SqlScriptUtils.convertIf(sqlScript, String.format("%s != null", alias), true);
        }
        if (ew) {
            sqlScript += NEWLINE;
            sqlScript += SqlScriptUtils.convertIf(SqlScriptUtils.unSafeParam(U_WRAPPER_SQL_SET),
                    String.format("%s != null and %s != null", WRAPPER, U_WRAPPER_SQL_SET), false);
        }
        sqlScript = SqlScriptUtils.convertSet(sqlScript);
        return sqlScript;
    }


    /**
     * 获取所有的 sql set 片段
     *
     * @param ignoreLogicDelFiled 是否过滤掉逻辑删除字段
     * @param prefix              前缀
     * @return sql 脚本片段
     */
    public String getAllSqlSet(TableInfo table, boolean ignoreLogicDelFiled, final String prefix) {
        final String newPrefix = prefix == null ? EMPTY : prefix;
        return table.getFieldList().stream()
                .filter(i -> {
                    if (ignoreLogicDelFiled) {
                        return !(table.isWithLogicDelete() && i.isLogicDelete());
                    }
                    return true;
                }).map(i -> i.getSqlSet(true, newPrefix)).filter(Objects::nonNull).collect(joining(NEWLINE));
    }

    @Override
    public String getMethod(SqlMethod sqlMethod) {
        return "updateAllById";
    }

}

package com.np.pm.common.springboot.starter.mapper;

import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.core.toolkit.Constants;
import org.apache.ibatis.annotations.Param;

/**
 * <p>mapper 基础接口</p>
 *
 * @Author: zhaojiwei
 * @Versions 1.0
 * @Date: 2021/7/26 下午12:48
 */
public interface PMBaseMapper<T> extends BaseMapper<T> {

    /**
     * 根据 ID 全部修改
     *
     * @param entity 实体对象
     */
    int updateAllById(@Param(Constants.ENTITY) T entity);

}
  1. 方式三（手写xml来更新复杂业务需求）

    数据权限（未完成）