This commit is contained in:
ovo 2024-12-07 20:09:31 +08:00
parent 12542ae115
commit 34b48cc27d
3 changed files with 46 additions and 26 deletions

View File

@ -1,7 +1,10 @@
package com.guwan.backend.security;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.guwan.backend.dto.user.UserDTO;
import com.guwan.backend.service.UserService;
import com.guwan.backend.entity.User;
import com.guwan.backend.mapper.UserMapper;
import com.guwan.backend.security.CustomUserDetails;
import lombok.RequiredArgsConstructor;
import org.springframework.security.core.authority.SimpleGrantedAuthority;
import org.springframework.security.core.userdetails.UserDetails;
@ -16,30 +19,33 @@ import java.util.List;
@RequiredArgsConstructor
public class UserDetailsServiceImpl implements UserDetailsService {
private final UserService userService;
private final UserMapper userMapper;
@Override
public UserDetails loadUserByUsername(String username) throws UsernameNotFoundException {
UserDTO user = userService.findByUsername(username);
User user = userMapper.selectOne(
new LambdaQueryWrapper<User>()
.eq(User::getUsername, username)
);
if (user == null) {
throw new UsernameNotFoundException("用户不存在");
}
return convertToUserDetailsWithoutPassword(user);
return convertToUserDetails(user);
}
public UserDetails loadUserById(Long userId) throws UsernameNotFoundException {
UserDTO user = userService.getUserById(userId);
User user = userMapper.selectById(userId);
if (user == null) {
throw new UsernameNotFoundException("用户不存在");
}
return convertToUserDetailsWithoutPassword(user);
return convertToUserDetails(user);
}
private UserDetails convertToUserDetailsWithoutPassword(UserDTO user) {
private UserDetails convertToUserDetails(User user) {
List<SimpleGrantedAuthority> authorities = new ArrayList<>();
authorities.add(new SimpleGrantedAuthority("Admin"));
authorities.add(new SimpleGrantedAuthority("ROLE_USER"));
return new CustomUserDetails(
user.getId(),

View File

@ -13,6 +13,7 @@ import com.guwan.backend.service.UserService;
import com.guwan.backend.util.JwtUtil;
import com.guwan.backend.util.RedisUtil;
import com.guwan.backend.util.RedisUtils;
import com.guwan.backend.util.SecurityUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeanUtils;
@ -37,6 +38,7 @@ public class UserServiceImpl implements UserService {
private final RedisUtil redisUtil;
private final RedisUtils redisUtils;
private final EmailService emailService;
private final SecurityUtil securityUtil;
private static final String USER_CACHE_KEY = "user:";
private static final long USER_CACHE_DURATION = 3600L; // 1小时
@ -161,22 +163,7 @@ public class UserServiceImpl implements UserService {
@Override
public Long getCurrentUserId() {
// 从SecurityContext中获取认证信息
Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
// 如果没有认证信息返回null
if (authentication == null || !authentication.isAuthenticated() ||
authentication instanceof AnonymousAuthenticationToken) {
return null;
}
// 获取UserDetails
Object principal = authentication.getPrincipal();
if (principal instanceof CustomUserDetails) {
return ((CustomUserDetails) principal).getUserId();
}
//demo
return null;
return securityUtil.getCurrentUserId();
}
@Override

View File

@ -0,0 +1,27 @@
package com.guwan.backend.util;
import com.guwan.backend.security.CustomUserDetails;
import org.springframework.security.authentication.AnonymousAuthenticationToken;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;
import org.springframework.stereotype.Component;
@Component
public class SecurityUtil {
public Long getCurrentUserId() {
Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
if (authentication == null || !authentication.isAuthenticated() ||
authentication instanceof AnonymousAuthenticationToken) {
return null;
}
Object principal = authentication.getPrincipal();
if (principal instanceof CustomUserDetails) {
return ((CustomUserDetails) principal).getUserId();
}
return null;
}
}