增加处理空列表情况

This commit is contained in:
s2042968 2025-01-07 12:42:03 +08:00
parent 5f52e7f020
commit 35e62a2c71
1 changed files with 7 additions and 5 deletions

View File

@ -9,11 +9,10 @@ import com.electromagnetic.industry.software.manage.pojo.models.RolePermission;
import com.electromagnetic.industry.software.manage.pojo.models.UserRole;
import com.electromagnetic.industry.software.manage.service.PermissionService;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import javax.annotation.Resource;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.*;
import java.util.stream.Collectors;
@Service
@ -32,16 +31,19 @@ public class PermissionServiceImpl implements PermissionService {
* @return
*/
@Override
@Transactional
public Map<String, Boolean> getUserPermission(String userId, String id) {
List<String> roleIds = getRoles(userId);
// 只有当 roleIds 不为空且不为 null 才添加 in 条件
if (roleIds == null || roleIds.isEmpty()) {
return transToMap(new ArrayList<>());
}
LambdaQueryWrapper<RolePermission> queryWrapper1 = new LambdaQueryWrapper<>();
queryWrapper1.select(RolePermission::getPermissionCode)
.eq(RolePermission::getFileId, id)
.in(RolePermission::getRoleId, roleIds);
List<String> permissionCodes = rolePermissionMapper.selectObjs(queryWrapper1).stream().map(Object::toString).collect(Collectors.toList());
return transToMap(permissionCodes);
}