diff --git a/aircraft-system/src/main/java/com/aircraft/modules/security/service/AUserDetailsService.java b/aircraft-system/src/main/java/com/aircraft/modules/security/service/AUserDetailsService.java index 6551bc0..d7ed335 100644 --- a/aircraft-system/src/main/java/com/aircraft/modules/security/service/AUserDetailsService.java +++ b/aircraft-system/src/main/java/com/aircraft/modules/security/service/AUserDetailsService.java @@ -46,7 +46,7 @@ public class AUserDetailsService implements UserDetailsService { userDto.setEnabled(true); userDto.setIsAdmin(false); // 创建一个 JwtCustomerDto 对象 - employeeDto = new JwtUserDto(userDto, UserTypeEnum.EMPLOYEES, new ArrayList<>()); + employeeDto = new JwtUserDto(userDto, UserTypeEnum.EMPLOYEES, new ArrayList<>(),null); // 添加缓存数据 userCacheManager.addEmployeeCache(username, employeeDto); } diff --git a/aircraft-system/src/main/java/com/aircraft/modules/security/service/BUserDetailsService.java b/aircraft-system/src/main/java/com/aircraft/modules/security/service/BUserDetailsService.java index 14b3175..1da68ec 100644 --- a/aircraft-system/src/main/java/com/aircraft/modules/security/service/BUserDetailsService.java +++ b/aircraft-system/src/main/java/com/aircraft/modules/security/service/BUserDetailsService.java @@ -45,7 +45,7 @@ public class BUserDetailsService implements UserDetailsService { userDto.setEnabled(true); userDto.setIsAdmin(false); // 创建一个 JwtCustomerDto 对象 - customerDto = new JwtUserDto(userDto, UserTypeEnum.CUSTOMER, new ArrayList<>()); + customerDto = new JwtUserDto(userDto, UserTypeEnum.CUSTOMER, new ArrayList<>(), customerDto.getCustomerType()); // 添加缓存数据 userCacheManager.addCustomerCache(username, customerDto); } diff --git a/aircraft-system/src/main/java/com/aircraft/modules/security/service/UserDetailsServiceImpl.java b/aircraft-system/src/main/java/com/aircraft/modules/security/service/UserDetailsServiceImpl.java index 11005db..65f3d63 100644 --- a/aircraft-system/src/main/java/com/aircraft/modules/security/service/UserDetailsServiceImpl.java +++ b/aircraft-system/src/main/java/com/aircraft/modules/security/service/UserDetailsServiceImpl.java @@ -60,7 +60,7 @@ public class UserDetailsServiceImpl implements UserDetailsService { // 获取用户的权限 List authorities = roleService.buildPermissions(user); // 初始化JwtUserDto - jwtUserDto = new JwtUserDto(BeanUtil.copyProperties(user, LoginUserDto.class), UserTypeEnum.getByValue(user.getUserType()), authorities); + jwtUserDto = new JwtUserDto(BeanUtil.copyProperties(user, LoginUserDto.class), UserTypeEnum.getByValue(user.getUserType()), authorities,null); // 添加缓存数据 userCacheManager.addUserCache(username, jwtUserDto); } diff --git a/aircraft-system/src/main/java/com/aircraft/modules/security/service/dto/JwtUserDto.java b/aircraft-system/src/main/java/com/aircraft/modules/security/service/dto/JwtUserDto.java index 6b1e981..8a654e5 100644 --- a/aircraft-system/src/main/java/com/aircraft/modules/security/service/dto/JwtUserDto.java +++ b/aircraft-system/src/main/java/com/aircraft/modules/security/service/dto/JwtUserDto.java @@ -50,6 +50,9 @@ public class JwtUserDto implements UserDetails { @ApiModelProperty(value = "角色") private final List authorities; + @ApiModelProperty("客户类型") + private final String customerType; + public Set getRoles() { return authorities.stream().map(AuthorityDto::getAuthority).collect(Collectors.toSet()); }