♻️ Refactoring code. 数据权限扩展支持 mybatis-plus 自带方法

pigx-common/pigx-common-data/src/main/java/com/pig4cloud/pigx/common/data/datascope/DataScopeMapper.java

@@ -0,0 +1,38 @@
+package com.pig4cloud.pigx.common.data.datascope;
+
+import com.baomidou.mybatisplus.core.conditions.Wrapper;
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
+import com.baomidou.mybatisplus.core.toolkit.Constants;
+import org.apache.ibatis.annotations.Param;
+
+import java.util.List;
+
+/**
+ * @author lengleng
+ * @date 2020/4/26
+ */
+public interface DataScopeMapper<T> extends BaseMapper<T> {
+
+	/**
+	 * 根据 entity 条件，查询全部记录
+	 *
+	 * @param queryWrapper 实体对象封装操作类（可以为 null）
+	 * @param scope        数据权限范围
+	 * @return List<T>
+	 */
+	List<T> selectListByScope(@Param(Constants.WRAPPER) Wrapper<T> queryWrapper, DataScope scope);
+
+
+	/**
+	 * 根据 entity 条件，查询全部记录（并翻页）
+	 *
+	 * @param page         分页查询条件（可以为 RowBounds.DEFAULT）
+	 * @param queryWrapper 实体对象封装操作类（可以为 null）
+	 * @param scope        数据权限范围
+	 * @return Page
+	 */
+	<E extends IPage<T>> E selectPageByScope(E page, @Param(Constants.WRAPPER) Wrapper<T> queryWrapper, DataScope scope);
+
+
+}

pigx-common/pigx-common-data/src/main/java/com/pig4cloud/pigx/common/data/datascope/DataScopeSqlInjector.java

@@ -0,0 +1,21 @@
+package com.pig4cloud.pigx.common.data.datascope;
+
+import com.baomidou.mybatisplus.core.injector.AbstractMethod;
+import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
+
+import java.util.List;
+
+/**
+ * @author lengleng
+ * @date 2020/4/26
+ */
+public class DataScopeSqlInjector extends DefaultSqlInjector {
+
+	@Override
+	public List<AbstractMethod> getMethodList(Class<?> mapperClass) {
+		List<AbstractMethod> methodList = super.getMethodList(mapperClass);
+		methodList.add(new SelectListByScope());
+		methodList.add(new SelectPageByScope());
+		return methodList;
+	}
+}

pigx-common/pigx-common-data/src/main/java/com/pig4cloud/pigx/common/data/datascope/SelectListByScope.java

@@ -0,0 +1,22 @@
+package com.pig4cloud.pigx.common.data.datascope;
+
+import com.baomidou.mybatisplus.core.enums.SqlMethod;
+import com.baomidou.mybatisplus.core.injector.AbstractMethod;
+import com.baomidou.mybatisplus.core.metadata.TableInfo;
+import org.apache.ibatis.mapping.MappedStatement;
+import org.apache.ibatis.mapping.SqlSource;
+
+/**
+ * @author lengleng
+ * @date 2020/4/26
+ */
+public class SelectListByScope extends AbstractMethod {
+	@Override
+	public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
+		SqlMethod sqlMethod = SqlMethod.SELECT_LIST;
+		String sql = String.format(sqlMethod.getSql(), sqlFirst(), sqlSelectColumns(tableInfo, true), tableInfo.getTableName(),
+				sqlWhereEntityWrapper(true, tableInfo), sqlComment());
+		SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
+		return this.addSelectMappedStatementForTable(mapperClass, "selectListByScope", sqlSource, tableInfo);
+	}
+}

pigx-common/pigx-common-data/src/main/java/com/pig4cloud/pigx/common/data/datascope/SelectPageByScope.java

@@ -0,0 +1,24 @@
+package com.pig4cloud.pigx.common.data.datascope;
+
+import com.baomidou.mybatisplus.core.enums.SqlMethod;
+import com.baomidou.mybatisplus.core.injector.AbstractMethod;
+import com.baomidou.mybatisplus.core.metadata.TableInfo;
+import org.apache.ibatis.mapping.MappedStatement;
+import org.apache.ibatis.mapping.SqlSource;
+
+/**
+ * @author lengleng
+ * @date 2020/4/26
+ */
+public class SelectPageByScope extends AbstractMethod {
+
+	@Override
+	public MappedStatement injectMappedStatement(Class<?> mapperClass, Class<?> modelClass, TableInfo tableInfo) {
+		SqlMethod sqlMethod = SqlMethod.SELECT_PAGE;
+		String sql = String.format(sqlMethod.getSql(), sqlFirst(), sqlSelectColumns(tableInfo, true),
+				tableInfo.getTableName(), sqlWhereEntityWrapper(true, tableInfo), sqlComment());
+		SqlSource sqlSource = languageDriver.createSqlSource(configuration, sql, modelClass);
+		return this.addSelectMappedStatementForTable(mapperClass, "selectPageByScope", sqlSource, tableInfo);
+	}
+
+}

pigx-common/pigx-common-data/src/main/java/com/pig4cloud/pigx/common/data/mybatis/MybatisPlusConfiguration.java

@@ -22,6 +22,7 @@ import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
 import com.baomidou.mybatisplus.extension.plugins.tenant.TenantSqlParser;
 import com.pig4cloud.pigx.common.data.datascope.DataScopeHandle;
 import com.pig4cloud.pigx.common.data.datascope.DataScopeInterceptor;
+import com.pig4cloud.pigx.common.data.datascope.DataScopeSqlInjector;
 import com.pig4cloud.pigx.common.data.datascope.PigxDefaultDatascopeHandle;
 import com.pig4cloud.pigx.common.data.tenant.PigxTenantHandler;
 import org.springframework.boot.autoconfigure.AutoConfigureAfter;
@@ -97,4 +98,10 @@ public class MybatisPlusConfiguration {
 	public DataScopeInterceptor dataScopeInterceptor() {
 		return new DataScopeInterceptor(dataScopeHandle());
 	}
+
+	@Bean
+	@ConditionalOnBean(DataScopeHandle.class)
+	public DataScopeSqlInjector dataScopeSqlInjector() {
+		return new DataScopeSqlInjector();
+	}
 }