SpringBoot整合Spring Security(六)

Spring Boot真的香系列

发布于：2020年10月10日

次浏览

一、OverView

在上一节中，使用了第一种方式：JDBC 连接数据库。这一节中将使用 Spring Data JPA 进行数据库操作。如果对 JPA 的环境和操作没了解过的，可以适当参考这个。

在上一节也提到 Spring Security 支持多种数据源，这些不同来源的数据被共同封装成了一个 UserDetailService 接口

在这一节中，还是这个思路，这一次不用 Spring Security 提供的类, 可以自定义一个类去实现 UserDetailService

二、环境搭建

思路清晰，先看依赖

application.properties

spring.datasource.username=root
spring.datasource.password=
spring.datasource.url=jdbc:mysql://127.0.0.1:3306/security?useUnicode=true&characterEncoding=UTF-8&serverTimezone=Asia/Shanghai

spring.jpa.database=mysql
spring.jpa.database-platform=mysql
spring.jpa.hibernate.ddl-auto=update
spring.jpa.show-sql=true
spring.jpa.properties.hibernate.dialect=org.hibernate.dialect.MySQL8Dialect

实体类

User

@Entity(name = "t_user")
public class User implements UserDetails {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private String username;
    private String password;
    private boolean accountNonExpired;
    private boolean accountNonLocked;
    private boolean credentialsNonExpired;
    private boolean enabled;

    // User 和 Role 是多对多关系
    @ManyToMany(fetch = FetchType.EAGER,cascade = CascadeType.PERSIST)
    private List<Role> roles;

    // 返回用户权限
    @Override
    public Collection<? extends GrantedAuthority> getAuthorities() {
        List<SimpleGrantedAuthority> authorities = new ArrayList<>();
        for (Role role : getRoles()) {
            authorities.add(new SimpleGrantedAuthority(role.getName()));
        }
        return authorities;
    }

    @Override
    public String getPassword() {
        return password;
    }

    @Override
    public String getUsername() {
        return username;
    }

    // 用户是否过期
    @Override
    public boolean isAccountNonExpired() {
        return accountNonExpired;
    }

    // 用户是否被锁
    @Override
    public boolean isAccountNonLocked() {
        return accountNonLocked;
    }

    // 密码是否过期
    @Override
    public boolean isCredentialsNonExpired() {
        return credentialsNonExpired;
    }

    // 账户是否可用
    @Override
    public boolean isEnabled() {
        return enabled;
    }
    
    // set get 省略
}

注：这里为什么 User 要实现 UserDetails ? 主要是由于在自定义类实现 UserDetailService，其中只有唯一的抽象方法loadUserByUsername，而该方法就是返回的 UserDetails

Role

@Entity(name = "t_role")
public class Role {
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    private Long id;
    private String name;
    private String nameZh;
}

UserDao

public interface UserDao extends JpaRepository<User, Long> {

    /**
     * 通过用户名找到用户
     *
     * @param username 用户名
     * @return {@link User}
     */
    User findAllByUsername(String username);
}

不懂为啥这么写的可以参考一下 JPA 的 CRUD 操作

UserService

@Service
public class UserService implements UserDetailsService{

    private UserDao userDao;

    @Autowired
    public void setUserDao(UserDao userDao) {
        this.userDao = userDao;
    }

    @Override
    public UserDetails loadUserByUsername(String s) throws UsernameNotFoundException {
        User user = userDao.findAllByUsername(s);
        if (StringUtils.isEmpty(user)) {
            throw new UsernameNotFoundException("用户不存在");
        }
        return user;
    }

    /**
     * 保存用户
     *
     * @param user 用户
     * @return {@link User}
     */
    User save(User user) {
        return userDao.save(user);
    }
}

注：

实现了 UserDetailsService，重写其中的 loadUserByUsername 方法：通过传入的用户名查找用户信息

save 方法是为了后面写入不同角色用户

UserController

@RestController
public class UserController {

    private UserService userService;

    @Autowired
    public void setUserService(UserService userService) {
        this.userService = userService;
    }

    @PutMapping("/save")
    public User save(@RequestBody User user) {
        return userService.save(user);
    }
}

将 SecurityConfig 中 configure 方法改为：

private UserService userService;

@Autowired
public void setUserService(UserService userService) {
    this.userService = userService;
}

@Override
protected void configure(AuthenticationManagerBuilder auth) throws Exception {
    auth.userDetailsService(userService);
}