Spring Security：如何优化UserDetailsService实现多用户类型邮箱认证

算力阿峰

2026-8-5

优化方案

关键改进点

消除重复数据库查询：每个仓库仅执行一次查询，减少不必要的IO开销
简化逻辑流程：按顺序检查User、Admin，逻辑清晰易维护
遵循接口规范：抛出UserDetailsService要求的UsernameNotFoundException而非自定义异常

优化后代码

@Service
@RequiredArgsConstructor
public class UserDetailsServiceImpl implements UserDetailsService {

    private final UserRepository userRepository;
    private final AdminRepository adminRepository;

    @Override
    public UserDetails loadUserByUsername(String email) throws UsernameNotFoundException {
        // 先查询用户，缓存结果避免重复调用
        Optional<? extends UserDetails> user = userRepository.findByEmail(email);
        if (user.isPresent()) {
            return user.get();
        }

        // 用户不存在时查询管理员
        Optional<? extends UserDetails> admin = adminRepository.findByEmail(email);
        if (admin.isPresent()) {
            return admin.get();
        }

        // 两者都不存在，抛出标准异常
        throw new UsernameNotFoundException("未找到与该邮箱匹配的用户或管理员");
    }
}

补充说明

查询复用：原代码中每个判断分支都重复调用findByEmail，优化后每个仓库仅查询一次，提升性能。
异常合规：loadUserByUsername接口明确要求抛出UsernameNotFoundException，替换原代码中的EntityNotFoundException更符合Spring Security的规范。
类型适配：请确保User和Admin实体类都正确实现了UserDetails接口，否则需要在返回时做类型转换（或在仓库查询方法中返回UserDetails类型）。
日志优化：原代码中的System.out.println建议替换为SLF4J日志（如log.debug("查询用户: {}", email)），便于生产环境排查问题。