work/book-api
4
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化代码

This commit is contained in:
阿沐 2022-10-25 11:35:24 +08:00
parent ccc71825c7
commit c9f32be88e
2 changed files with 26 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
maku-framework/src/main/java/net/maku/framework/common/service/impl/BaseServiceImpl.java
View File

@ -23,20 +23,21 @@ import java.util.List;
*
* @author 阿沐 babamu@126.com
*/
public class BaseServiceImpl<M extends BaseMapper<T>, T> extends ServiceImpl<M, T> implements BaseService<T> {
public class BaseServiceImpl<M extends BaseMapper<T>, T> extends ServiceImpl<M, T> implements BaseService<T> {
/**
* 获取分页对象
* @param query 分页参数
*
* @param query 分页参数
*/
protected IPage<T> getPage(Query query) {
Page<T> page = new Page<>(query.getPage(), query.getLimit());
// 排序
if(StringUtils.isNotBlank(query.getOrder())){
if(query.isAsc()) {
if (StringUtils.isNotBlank(query.getOrder())) {
if (query.isAsc()) {
return page.addOrder(OrderItem.asc(query.getOrder()));
}else {
} else {
return page.addOrder(OrderItem.desc(query.getOrder()));
}
}
@ -45,26 +46,37 @@ public class BaseServiceImpl<M extends BaseMapper<T>, T> extends ServiceImpl<M,
}
/**
* MyBatis-Plus 数据权限
*/
protected void dataScopeWrapper(LambdaQueryWrapper<T> queryWrapper) {
DataScope dataScope = getDataScope(null, null);
if (dataScope != null) {
queryWrapper.apply(dataScope.getSqlFilter());
}
}
/**
* 原生SQL 数据权限
*
* @param tableAlias 表别名，多表关联时，需要填写表别名
* @param orgIdAlias 机构ID别名，null：表示org_id
* @return 返回数据权限
*/
protected DataScope getDataScope(String tableAlias, String orgIdAlias) {
protected DataScope getDataScope(String tableAlias, String orgIdAlias) {
UserDetail user = SecurityUser.getUser();
// 如果是超级管理员，则不进行数据过滤
if(user.getSuperAdmin().equals(Constant.SUPER_ADMIN)) {
if (user.getSuperAdmin().equals(Constant.SUPER_ADMIN)) {
return null;
}
// 如果为null，则设置成空字符串
if(tableAlias == null){
if (tableAlias == null) {
tableAlias = "";
}
// 获取表的别名
if(StringUtils.isNotBlank(tableAlias)){
tableAlias += ".";
if (StringUtils.isNotBlank(tableAlias)) {
tableAlias += ".";
}
StringBuilder sqlFilter = new StringBuilder();
@ -73,12 +85,12 @@ public class BaseServiceImpl<M extends BaseMapper<T>, T> extends ServiceImpl<M,
// 数据权限范围
List<Long> dataScopeList = user.getDataScopeList();
// 全部数据权限
if (dataScopeList == null){
if (dataScopeList == null) {
return null;
}
// 数据过滤
if(dataScopeList.size() > 0){
if(StringUtils.isBlank(orgIdAlias)){
if (dataScopeList.size() > 0) {
if (StringUtils.isBlank(orgIdAlias)) {
orgIdAlias = "org_id";
}
sqlFilter.append(tableAlias).append(orgIdAlias);
@ -95,15 +107,4 @@ public class BaseServiceImpl<M extends BaseMapper<T>, T> extends ServiceImpl<M,
return new DataScope(sqlFilter.toString());
}
/**
* MyBatis-Plus 数据权限
*/
protected void dataScopeWrapper(LambdaQueryWrapper<T> queryWrapper) {
DataScope dataScope = getDataScope(null, null);
if (dataScope != null){
queryWrapper.apply(dataScope.getSqlFilter());
}
}
}

2
maku-framework/src/main/java/net/maku/framework/security/config/SecurityConfig.java
View File

@ -72,7 +72,7 @@ public class SecurityConfig {
}
@Bean
SecurityFilterChain filterChain(HttpSecurity http) throws Exception {
SecurityFilterChain securityFilterChain(HttpSecurity http) throws Exception {
// 忽略授权的地址列表
List<String> permitList = permitResource.getPermitList();
String[] permits = permitList.toArray(new String[0]);