java - 如何在 Spring 中从自定义数据源访问 Http Session？

Question

有没有办法从自定义数据源中的 WebApplicationContext 访问 HttpSession？我实现了一个自定义身份验证处理过滤器，它在 HttpSession 中存储一些信息。然后，数据源使用此信息来获取数据库连接。

另一个选项是使用 SecurityContextHolder 来获取经过自定义以包含其他属性的身份验证 token 。我不确定这是否是正确的方法。

这是我到目前为止所拥有的:

public class CustomDataSource extends DriverManagerDataSource implements ApplicationContextAware {

protected Connection getConnectionFromDriverManager(String url,
        Properties props) throws SQLException {

    // want to use the web context to get the http session

    // Authentication has a getAttribute(String name) method
    SecurityContext securityContext = SecurityContextHolder.getContext();
    CustomAuthenticationToken authentication = (CustomAuthenticationToken) securityContext.getAuthentication();
    Object attribute = authentication.getAttribute("db");


    // create a connection object here
    Object conn = getConnectionFromAttribute(attribute);
    return (Connection)conn;
}

private WebApplicationContext context;

@Override
public void setApplicationContext(ApplicationContext applicationContext)
        throws BeansException {
    this.context = (WebApplicationContext)applicationContext;        
}
}

更新:

我定义了一个名为 AuthInfo 的新类，它只有用户名和密码。然后将ThreadLocal创建为实用程序接口(interface)的静态最终变量:

public interface WebUtils{
    public static final ThreadLocal<AuthInfo> authInfo = new ThreadLocal<AuthInfo>();
}

然后在过滤器的attemptAuthentication方法中设置ThreadLocal的值

AuthInfo info = new AuthInfo();
info.setName(username);
info.setPass(password);

WebAttributes.authInfo.set(info);

现在，在自定义数据源中

protected Connection getConnectionFromDriverManager(String url,
        Properties props) throws SQLException {

    AuthInfo info = WebAttributes.authInfo.get();
    Connection conn = getConnFromAuthInfo(info);
    return conn;
}

这不是与使用 SecurityContextHolder 和 CustomAuthenticationToken 相同吗？

Answer 1

在典型的多层应用程序中，您的数据访问层不应该了解更高层(例如 HTTP 接口(interface))。

我建议您使用 Spring 的 session 或请求范围进行调查。您可以在这些范围之一中创建一个范围代理 bean，将身份验证信息放入其中，然后将其注入(inject)到数据源中。