返回介绍

2.4 用户定义

发布于 2025-04-26 13:16:43 字数 16880 浏览 0 评论 0 收藏

在前面的案例中,我们的登录用户是基于配置文件来配置的(本质是基于内存),但是在实际开发中,这种方式肯定是不可取的,在实际项目中,用户信息肯定要存入数据库之中。

Spring Security 支持多种用户定义方式,接下来我们就逐个来看一下这些定义方式。通过前面的介绍(参见 2.1.3 小节),大家对于 UserDetailsService 以及它的子类都有了一定的了解,自定义用户其实就是使用 UserDetailsService 的不同实现类来提供用户数据,同时将配置好的 UserDetailsService 配置给 AuthenticationManagerBuilder,系统再将 UserDetailsService 提供给 AuthenticationProvider 使用。

2.4.1 基于内存

前面案例中用户的定义本质上还是基于内存,只是我们没有将 InMemoryUserDetails Manager 类明确抽出来自定义,现在我们通过自定义 InMemoryUserDetailsManager 来看一下基于内存的用户是如何自定义的。

重写 WebSecurityConfigurerAdapter 类的 configure(AuthenticationManagerBuilder) 方法,内容如下:

    @Override
    protected void configure(AuthenticationManagerBuilder auth) throws Exception {
       InMemoryUserDetailsManager manager = new InMemoryUserDetailsManager();
       manager.createUser(User.withUsername("javaboy")
                                 .password("{noop}123").roles("admin").build());
       manager.createUser(User.withUsername("sang")
                                  .password("{noop}123").roles("user").build());
       auth.userDetailsService(manager);
    }

首先构造了一个 InMemoryUserDetailsManager 实例,调用该实例的 createUser 方法来创建用户对象,我们在这里分别设置了用户名、密码以及用户角色。需要注意的是,用户密码加了一个{noop}前缀,表示密码不加密,明文存储(关于密码加密问题,会在后面的章节中专门介绍)。

配置完成后,启动项目,此时就可以使用这里配置的两个用户登录了。

InMemoryUserDetailsManager 的实现原理很简单,它间接实现了 UserDetailsService 接口并重写了它里边的 loadUserByUsername 方法,同时它里边维护了一个 HashMap 变量,Map 的 key 就是用户名,value 则是用户对象,createUser 就是往这个 Map 中存储数据,loadUserBy Username 方法则是从该 Map 中读取数据,这里的源码比较简单,就不贴出来了,读者可以自行查看。

2.4.2 基于 JdbcUserDetailsManager

JdbcUserDetailsManager 支持将用户数据持久化到数据库,同时它封装了一系列操作用户的方法,例如用户的添加、更新、查找等。

Spring Security 中为 JdbcUserDetailsManager 提供了数据库脚本,位置在 org/springframework/security/core/userdetails/jdbc/users.ddl,内容如下:

    create table users(username varchar_ignorecase(50) not null
    primary key,
    password varchar_ignorecase(500) not null,
    enabled boolean not null);
  
    create table authorities (username varchar_ignorecase(50) not null,
    authority varchar_ignorecase(50) not null,
    constraint fk_authorities_users
    foreign key(username) references users(username));
  
    create unique index ix_auth_username on authorities (username,authority);

可以看到这里一共创建了两张表,users 表就是存放用户信息的表,authorities 则是存放用户角色的表。但是大家注意 SQL 的数据类型中有一个 varchar_ignorecase,这个其实是针对 HSQLDB 的数据类型,我们这里使用的是 MySQL 数据库,所以这里手动将 varchar_ignorecase 类型修改为 varchar 类型,然后去数据库中执行修改后的脚本。

另一方面,由于要将数据存入数据库中,所以我们的项目也要提供数据库支持,JdbcUserDetailsManager 底层实际上是使用 JdbcTemplate 来完成的,所以这里主要添加两个依赖:

    <dependency>
       <groupId>org.springframework.boot</groupId>
       <artifactId>spring-boot-starter-jdbc</artifactId>
    </dependency>
    <dependency>
       <groupId>mysql</groupId>
       <artifactId>mysql-connector-java</artifactId>
    </dependency>

然后在 resources/application.properties 中配置数据库连接信息:

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

配置完成后,我们重写 WebSecurityConfigurerAdapter 类的 configure(AuthenticationManagerBuilder) 方法,内容如下:

(1)当引入 spring-boot-starter-jdbc 并配置了数据库连接信息后,一个 DataSource 实例就有了,这里首先引入 DataSource 实例。

(2)在 configure 方法中,创建一个 JdbcUserDetailsManager 实例,在创建时传入 DataSource 实例。通过 userExists 方法可以判断一个用户是否存在,该方法本质上就是去数据库中查询对应的用户;如果用户不存在,则通过 createUser 方法可以创建一个用户,该方法本质上就是向数据库中添加一个用户。

(3)最后将 manager 实例设置到 auth 对象中。

配置完成后,重启项目,如果项目启动成功,数据库中就会自动添加进来两条数据,如图 2-22、图 2-23 所示。

图 2-22 数据库中自动存入两条 users 数据

图 2-23 数据库中自动存入两条角色数据

此时,我们就可以使用 javaboy/123、sang/123 进行登录测试了。

在 JdbcUserDetailsManager 的继承体系中,首先是 JdbcDaoImpl 实现了 UserDetailsService 接口,并实现了基本的 loadUserByUsername 方法。JdbcUserDetailsManager 则继承自 JdbcDaoImpl,同时完善了数据库操作,又封装了用户的增删改查方法。这里,我们以 loadUserByUsername 为例,看一下源码,其余的增删改操作相对来说都比较容易,这里就不再赘述了。

JdbcDaoImpl#loadUserByUsername:

(1)首先根据用户名,调用 loadUsersByUsername 方法去数据库中查询用户,查询出来的是一个 List 集合,集合中如果没有数据,说明用户不存在,则直接抛出异常。

(2)如果集合中存在数据,则将集合中的第一条数据拿出来,然后再去查询用户角色,最后根据这些信息创建一个新的 UserDetails 出来。

(3)需要注意的是,这里还引入了分组的概念,不过考虑到 JdbcUserDetailsManager 并非我们实际项目中的主流方案,因此这里不做过多介绍。

这就是使用 JdbcUserDetailsManager 做数据持久化。这种方式看起来简单,都不用开发者自己写 SQL,但是局限性比较大,无法灵活地定义用户表、角色表等,而在实际开发中,我们还是希望能够灵活地掌控数据表结构,因此 JdbcUserDetailsManager 使用场景非常有限。

2.4.3 基于 MyBatis

使用 MyBatis 做数据持久化是目前大多数企业应用采取的方案,Spring Security 中结合 MyBatis 可以灵活地定制用户表以及角色表,我们对此进行详细介绍。

首先需要设计三张表,分别是用户表、角色表以及用户角色关联表,三张表的关系如图 2-24 所示。

图 2-24 用户表、角色表以及用户角色关联表

用户和角色是多对多的关系,我们使用 user_role 来将两者关联起来。

数据库脚本如下:

    CREATE TABLE `role` (
     `id` int(11) NOT NULL AUTO_INCREMENT,
     `name` varchar(32) DEFAULT NULL,
     `nameZh` varchar(32) DEFAULT NULL,
     PRIMARY KEY (`id`)
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8;
 
    CREATE TABLE `user` (
     `id` int(11) NOT NULL AUTO_INCREMENT,
     `username` varchar(32) DEFAULT NULL,
     `password` varchar(255) DEFAULT NULL,
     `enabled` tinyint(1) DEFAULT NULL,
     `accountNonExpired` tinyint(1) DEFAULT NULL,
     `accountNonLocked` tinyint(1) DEFAULT NULL,
     `credentialsNonExpired` tinyint(1) DEFAULT NULL,
     PRIMARY KEY (`id`)
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8;
 
    CREATE TABLE `user_role` (
     `id` int(11) NOT NULL AUTO_INCREMENT,
     `uid` int(11) DEFAULT NULL,
     `rid` int(11) DEFAULT NULL,
     PRIMARY KEY (`id`),
     KEY `uid` (`uid`),
     KEY `rid` (`rid`)
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8;

对于角色表,三个字段从上往下含义分别为角色 id、角色英文名称以及角色中文名称。

对于用户表,七个字段从上往下含义依次为:用户 id、用户名、用户密码、账户是否可用、账户是否没有过期、账户是否没有锁定以及凭证(密码)是否没有过期。

数据库创建完成后,可以向数据库中添加几条模拟数据,代码如下:

    INSERT INTO `role` (`id`, `name`, `nameZh`)
    VALUES
       (1,'ROLE_dba','数据库管理员'),
       (2,'ROLE_admin','系统管理员'),
       (3,'ROLE_user','用户');
 
    INSERT INTO `user` (`id`, `username`, `password`, `enabled`,
           `accountNonExpired`, `accountNonLocked`, `credentialsNonExpired`)
    VALUES
       (1,'root','{noop}123',1,1,1,1),
       (2,'admin','{noop}123',1,1,1,1),
       (3,'sang','{noop}123',1,1,1,1);
 
    INSERT INTO `user_role` (`id`, `uid`, `rid`)
    VALUES
       (1,1,1),
       (2,1,2),
       (3,2,2),
       (4,3,3);

这样,数据库的准备工作就算完成了。

在 Spring Security 项目中,我们需要引入 MyBatis 和 MySQL 依赖,代码如下:

    <dependency>
       <groupId>org.mybatis.spring.boot</groupId>
       <artifactId>mybatis-spring-boot-starter</artifactId>
       <version>2.1.3</version>
    </dependency>
    <dependency>
       <groupId>mysql</groupId>
       <artifactId>mysql-connector-java</artifactId>
    </dependency>

同时在 resources/application.properties 中配置数据库基本连接信息:

    spring.datasource.username=root
    spring.datasource.password=123
    spring.datasource.url=jdbc:mysql:///security02?useUnicode=true&characterEncod
ing=UTF-8&serverTimezone=Asia/Shanghai

接下来创建用户类和角色类:

    public class User implements UserDetails {
       private Integer id;
       private String username;
   
       private String password;
       private Boolean enabled;
       private Boolean accountNonExpired;
       private Boolean accountNonLocked;
       private Boolean credentialsNonExpired;
       private List<Role> roles = new ArrayList<>();
       @Override
       public Collection<? extends GrantedAuthority> getAuthorities() {
           List<SimpleGrantedAuthority> authorities = new ArrayList<>();
           for (Role role : roles) {
               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;
       }
       //省略其他 getter/setter
    }
    public class Role {
       private Integer id;
       private String name;
       private String nameZh;
       //省略 getter/setter
    }

自定义用户类需要实现 UserDetails 接口,并实现接口中的方法,这些方法的含义我们在 2.1.3 小节中已经介绍过了,这里不再赘述。其中 roles 属性用来保存用户所具备的角色信息,由于系统获取用户角色调用的方法是 getAuthorities,所以我们在 getAuthorities 方法中,将 roles 中的角色转为系统可识别的对象并返回。

注意

User 类中的 isXXX 方法可以当成 get 方法对待,不需要再给这些属性生成 get 方法。

接下来我们自定义 UserDetailsService 以及对应的数据库查询方法:

    @Service
    public class MyUserDetailsService implements UserDetailsService {
       @Autowired
       UserMapper userMapper;
       @Override
       public UserDetails loadUserByUsername(String username)
                                                 throws UsernameNotFoundException {
           User user = userMapper.loadUserByUsername(username);
           if (user == null) {
               throw new UsernameNotFoundException("用户不存在");
           }
           user.setRoles(userMapper.getRolesByUid(user.getId()));
           return user;
       }
    }
    @Mapper
    public interface UserMapper {
       List<Role> getRolesByUid(Integer id);
       User loadUserByUsername(String username);
    }

自定义 MyUserDetailsService 实现 UserDetailsService 接口,并实现该接口中的方法。loadUserByUsername 方法经过前面章节的讲解,相信大家已经很熟悉了,该方法就是根据用户名去数据库中加载用户,如果从数据库中没有查到用户,则抛出 UsernameNotFoundException 异常;如果查询到用户了,则给用户设置 roles 属性。

UserMapper 中定义两个方法用于支持 MyUserDetailsService 中的查询操作。

最后,在 UserMapper.xml 中定义查询 SQL,代码如下:

    <!DOCTYPE mapper
           PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
           "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
    <mapper namespace="org.javaboy.formlogin.mapper.UserMapper">
       <select id="loadUserByUsername"
                                 resultType="org.javaboy.formlogin.model.User">
           select * from user where username=#{username};
       </select>
       <select id="getRolesByUid" resultType="org.javaboy.formlogin.model.Role">
           select r.* from role r,user_role ur where r.`id`=ur.`rid`
       </select>
    </mapper>

为了方便,我们将 UserMapper.xml 文件和 UserMapper 接口放在了相同的包下。为了防止 Maven 打包时自动忽略了 XML 文件,还需要在 pom.xml 中添加如下配置:

    <build>
       <resources>
           <resource>
               <directory>src/main/java</directory>
               <includes>
                   <include>**/*.xml</include>
               </includes>
           </resource>
           <resource>
               <directory>src/main/resources</directory>
           </resource>
       </resources>
    </build>

最后一步,就是在 SecurityConfig 中注入 UserDetailsService:

    @Configuration
    public class SecurityConfig extends WebSecurityConfigurerAdapter {
       @Autowired
       MyUserDetailsService myUserDetailsService;
       @Override
       protected void configure(AuthenticationManagerBuilder auth)
                                                                 throws Exception {
           auth.userDetailsService(myUserDetailsService);
       }
       @Override
       protected void configure(HttpSecurity http) throws Exception {
           http.authorizeRequests()
                   //省略
       }
    }

配置 UserDetailsService 的方式和前面配置 JdbcUserDetailsManager 的方式基本一致,只不过配置对象变成了 myUserDetailsService 而已。

至此,整个配置工作就完成了。

接下来启动项目,利用数据库中添加的模拟用户进行登录测试,就可以成功登录了,测试方式和前面章节一致,这里不再赘述。

2.4.4 基于 Spring Data JPA

考虑到在 Spring Boot 技术栈中也有不少人使用 Spring Data JPA,因此这里针对 Spring Security+Spring Data JPA 也做一个简单介绍,具体思路和基于 MyBatis 的整合类似。

首先引入 Spring Data JPA 的依赖和 MySQL 依赖:

    <dependency>
       <groupId>org.springframework.boot</groupId>
       <artifactId>spring-boot-starter-data-jpa</artifactId>
    </dependency>
    <dependency>
       <groupId>mysql</groupId>
       <artifactId>mysql-connector-java</artifactId>
    </dependency>

然后在 resources/application.properties 中配置数据库和 JPA,代码如下:

    spring.datasource.username=root
    spring.datasource.password=123
    spring.datasource.url=jdbc:mysql:///security03?useUnicode=true&characterEncod
ing=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

数据库的配置还是和以前一样,JPA 的配置则主要配置了数据库平台,数据表更新方式、是否打印 SQL 以及对应的数据库方言。

使用 Spring Data JPA 的好处是我们不用提前准备 SQL 脚本,所以接下来配置两个数据库实体类即可:

    @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;
       @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;
       }
       //省略 getter/setter
    }
    @Entity(name = "t_role")
    public class Role {
       @Id
       @GeneratedValue(strategy = GenerationType.IDENTITY)
       private Long id;
       private String name;
       private String nameZh;
       //省略 getter/setter
    }

这两个实体类和前面 MyBatis 中实体类的配置类似,需要注意的是 roles 属性上多了一个多对多配置。

接下来配置 UserDetailsService,并提供数据查询方法:

MyUserDetailsService 的定义也和前面的类似,不同之处在于数据查询方法的变化。定义 UserDao 继承自 JpaRepository,并定义一个 findUserByUsername 方法,剩下的事情 Spring Data JPA 框架会帮我们完成。

最后,再在 SecurityConfig 中配置 MyUserDetailsService,配置方式和 MyBatis 一模一样,这里就不再把代码贴出来了。

使用了 Spring Data JPA 之后,当项目启动时,会自动在数据库中创建相关的表,而不用我们自己去写脚本,这也是使用 Spring Data JPA 的方便之处。

为了测试方便,我们可以在单元测试中执行如下代码,向数据库中添加测试数据:

    @Autowired
    UserDao userDao;
    @Test
    void contextLoads() {
       User u1 = new User();
       u1.setUsername("javaboy");
       u1.setPassword("{noop}123");
       u1.setAccountNonExpired(true);
       u1.setAccountNonLocked(true);
       u1.setCredentialsNonExpired(true);
       u1.setEnabled(true);
       List<Role> rs1 = new ArrayList<>();
       Role r1 = new Role();
       r1.setName("ROLE_admin");
       r1.setNameZh("管理员");
       rs1.add(r1);
       u1.setRoles(rs1);
       userDao.save(u1);
    }

测试数据添加成功之后,接下来启动项目,使用测试数据进行登录测试,具体测试过程就不再赘述了。

至此,四种不同的用户定义方式就介绍完了。这四种方式,异曲同工,只是数据存储的方式不一样而已,其他的执行流程都是一样的。

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
    我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。