Spring Security-从数据库获取当前用户字段

Question

CREATE TABLE users
(
  id_user serial NOT NULL,
  phone text,
  password text,
  balance numeric,
  email text,
  CONSTRAINT users_pkey PRIMARY KEY (id_user),
  CONSTRAINT users_login_key UNIQUE (phone)
)

我用的是Spring Security。配置的一部分：

<security:jdbc-user-service id="userService"
        data-source-ref="dataSource"
        users-by-username-query="select phone, password, true from users where phone=?"
        authorities-by-username-query="select phone,'ROLE_USER' from users where phone=?" />

例如，我如何在Spring MVC控制器中获得当前用户平衡？

Answer 1

您在Spring Security配置中定义的JDBC服务将仅使用user-by-username-query查询获取用户名、密码和启用状态，并使用authorities-by-username-query查询获取用户名的权限。这意味着任何额外的属性都不会映射到将创建用于填充安全上下文的用户主体对象。

获取用户所有信息的一种方法是创建userdetailsservice的自定义实现，该实现将能够检索用户数据，包括其所有自定义属性。下面是这种方法的一个示例（它假设您正在为您的数据层使用JPA/Hibernate）：

@Entity
@Table(name="users")
public class User implements UserDetails {
    private BigDecimal balance;
    // Other properties omitted ...
    // Getters and setters omitted ...
    // Implementation of UserDetails methods omitted ...
}

负责加载用户信息的服务。请注意，在现实世界中，您可能应该有单独的DAO查询用户、针对接口编程等。

@Service
public class UserService implements UserDetailsService {
    @Autowired
    private SessionFactory sessionFactory;

    @Override
    public User loadUserByUsername(String username) throws UsernameNotFoundException {
        Query query = sessionFactory.getCurrentSession().createQuery("FROM User u WHERE u.username = :username");
        query.setParameter("username", username);
        User user = (User) query.uniqueResult();
        if (user == null) {
            throw new UsernameNotFoundException("User with username '" + username + "' does not exist.");
        }
        return user;
    }
}

创建authenticationproviderbean并提供您的自定义用户服务作为其属性。

<bean id="daoAuthenticationProvider" class="org.springframework.security.authentication.dao.DaoAuthenticationProvider">
  <property name="userDetailsService" ref="userService" />
  <!-- Other required properties ... -->
</bean>

可以使用以下方法访问控制器中的当前用户及其属性：

User user = (User) SecurityContextHolder.getContext().getAuthentication().getPrincipal();
BigDecimal balance = user.getBalance();

作为最后的意见，请注意，这不是100%工作的复制和粘贴解决方案，您的问题，而是一个方法，您可以采取，以实现期望的结果。我建议查看核心类/接口的文档，如AuthenticationProvider、UserDetailsService、UserDetails和Spring Security guides。