Spring Security---将用户数据存入数据库详解

x33g5p2x  于2021-11-19 转载在 Spring  
字(5.2k)|赞(0)|评价(0)|浏览(766)

1.UserDetailService

Spring Security 支持多种不同的数据源,这些不同的数据源最终都将被封装成 UserDetailsService 的实例

可以看到,在几个能直接使用的实现类中,除了 InMemoryUserDetailsManager 之外,还有一个 JdbcUserDetailsManager,使用 JdbcUserDetailsManager 可以让我们通过 JDBC 的方式将数据库和 Spring Security 连接起来。

2.JdbcUserDetailsManager

JdbcUserDetailsManager 自己提供了一个数据库模型,这个数据库模型保存在如下位置:

  1. org/springframework/security/core/userdetails/jdbc/users.ddl

这里存储的脚本内容如下:

  1. create table users
  2. (username varchar_ignorecase(50) not null
  3. primary key,password varchar_ignorecase(500) not null,
  4. enabled boolean not null);
  5. create table authorities
  6. (username varchar_ignorecase(50) not null,
  7. authority varchar_ignorecase(50) not null,
  8. //外键关联
  9. constraint fk_authorities_users foreign key(username) references users(username));
  10. //复合索引
  11. create unique index ix_auth_username on authorities (username,authority);

可以看到,脚本中有一种数据类型 varchar_ignorecase,这个其实是针对 HSQLDB 数据库创建的,而我们使用的 MySQL 并不支持这种数据类型,所以这里需要大家手动调整一下数据类型,将 varchar_ignorecase 改为 varchar 即可。

修改完成后,创建数据库,执行完成后的脚本。

执行完 SQL 脚本后,我们可以看到一共创建了两张表:users 和 authorities

  • users 表中保存用户的基本信息,包括用户名、用户密码以及账户是否可用。
  • authorities 中保存了用户的角色。
  • authorities 和 users 通过 username 关联起来。

配置完成后,接下来,我们将上篇文章中通过 InMemoryUserDetailsManager 提供的用户数据用 JdbcUserDetailsManager 代替掉,如下:

  1. @Autowired
  2. DataSource dataSource;
  3. @Override
  4. @Bean
  5. protected UserDetailsService userDetailsService() {
  6. JdbcUserDetailsManager manager = new JdbcUserDetailsManager();
  7. manager.setDataSource(dataSource);
  8. if (!manager.userExists("dhy")) {
  9. manager.createUser(User.withUsername("dhy").password("123").roles("admin").build());
  10. }
  11. if (!manager.userExists("大忽悠")) {
  12. manager.createUser(User.withUsername("大忽悠").password("123").roles("user").build());
  13. }
  14. return manager;
  15. }

这段配置的含义如下:

  1. 首先构建一个 JdbcUserDetailsManager 实例。
  2. 给 JdbcUserDetailsManager 实例添加一个 DataSource 对象。
  3. 调用 userExists 方法判断用户是否存在,如果不存在,就创建一个新的用户出来(因为每次项目启动时这段代码都会执行,所以加一个判断,避免重复创建用户)。
  4. 用户的创建方法和我们之前 InMemoryUserDetailsManager 中的创建方法基本一致。

这里的 createUser 或者 userExists 方法其实都是调用写好的 SQL 去判断的,我们从它的源码里就能看出来(部分):

  1. public class JdbcUserDetailsManager extends JdbcDaoImpl implements UserDetailsManager,
  2. GroupManager {
  3. public static final String DEF_USER_EXISTS_SQL = "select username from users where username = ?";
  4. private String userExistsSql = DEF_USER_EXISTS_SQL;
  5. public boolean userExists(String username) {
  6. List<String> users = getJdbcTemplate().queryForList(userExistsSql,
  7. new String[] { username }, String.class);
  8. if (users.size() > 1) {
  9. throw new IncorrectResultSizeDataAccessException(
  10. "More than one user found with name '" + username + "'", 1);
  11. }
  12. return users.size() == 1;
  13. }
  14. }

从这段源码中就可以看出来,userExists 方法的执行逻辑其实就是调用 JdbcTemplate 来执行预定义好的 SQL 脚本,进而判断出用户是否存在,其他的判断方法都是类似,我就不再赘述。

3.数据库支持

通过前面的代码,大家看到这里需要数据库支持,所以我们在项目中添加如下两个依赖:

  1. <dependency>
  2. <groupId>org.springframework.boot</groupId>
  3. <artifactId>spring-boot-starter-jdbc</artifactId>
  4. </dependency>
  5. <dependency>
  6. <groupId>mysql</groupId>
  7. <artifactId>mysql-connector-java</artifactId>
  8. </dependency>

然后再在 application.properties 中配置一下数据库连接:

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

配置完成后,就可以启动项目。

项目启动成功后,我们就可以看到数据库中自动添加了两个用户进来,并且用户都配置了角色.

4.测试

接下来我们就可以进行测试了。

我们首先以 dhy的身份进行登录:

登录成功后,分别访问 /hello/admin/hello 以及 /user/hello 三个接口,其中:

  • /hello 因为登录后就可以访问,这个接口访问成功。
  • /admin/hello 需要 admin 身份,所以访问成功。
  • /user/hello 需要 user 身份,因为配置了角色继承,所以访问成功

在测试的过程中,如果在数据库中将用户的 enabled 属性设置为 false,表示禁用该账户,此时再使用该账户登录就会登录失败。

配置完成的代码如下:

  1. @Configuration
  2. public class SecurityConfig extends WebSecurityConfigurerAdapter {
  3. //fastjson
  4. private ObjectMapper objectMapper=new ObjectMapper();
  5. @Bean
  6. PasswordEncoder passwordEncoder() {
  7. //密码不加密
  8. return NoOpPasswordEncoder.getInstance();
  9. }
  10. //放行静态资源
  11. @Override
  12. public void configure(WebSecurity web) throws Exception {
  13. web.ignoring().antMatchers("/js/**", "/css/**","/images/**");
  14. }
  15. //数据源注入
  16. @Autowired
  17. DataSource dataSource;
  18. //用户配置
  19. @Override
  20. @Bean
  21. protected UserDetailsService userDetailsService() {
  22. JdbcUserDetailsManager manager = new JdbcUserDetailsManager();
  23. manager.setDataSource(dataSource);
  24. if (!manager.userExists("dhy")) {
  25. manager.createUser(User.withUsername("dhy").password("123").roles("admin").build());
  26. }
  27. if (!manager.userExists("大忽悠")) {
  28. manager.createUser(User.withUsername("大忽悠").password("123").roles("user").build());
  29. }
  30. return manager;
  31. }
  32. @Override
  33. protected void configure(HttpSecurity http) throws Exception {
  34. http.authorizeRequests()
  35. .antMatchers("/admin/**").hasRole("admin")
  36. .antMatchers("/user/**").hasRole("user")
  37. .anyRequest().authenticated()
  38. .and()
  39. .formLogin()
  40. .loginPage("/login.html").permitAll()
  41. .loginProcessingUrl("/login").permitAll()
  42. .successHandler(new AuthenticationSuccessHandler() {
  43. @Override
  44. public void onAuthenticationSuccess(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Authentication authentication) throws IOException, ServletException {
  45. httpServletResponse.setContentType("application/json;charset=UTF-8");
  46. httpServletResponse.getWriter().write(objectMapper.writeValueAsString(authentication));
  47. }
  48. }).permitAll()
  49. .and()
  50. .csrf().disable();
  51. }
  52. //角色继承
  53. @Bean
  54. RoleHierarchy roleHierarchy() {
  55. RoleHierarchyImpl hierarchy = new RoleHierarchyImpl();
  56. hierarchy.setHierarchy("ROLE_admin > ROLE_user");
  57. return hierarchy;
  58. }
  59. }

相关文章