Tomcat 8中WebSocket端点无法获取数据库连接池连接的问题

这问题我之前在Tomcat环境开发WebSocket应用时也踩过坑，核心原因是Tomcat对@ServerEndpoint标注的WebSocket端点的依赖注入支持和Servlet完全不同，下面给你拆解原因和可行的解决办法：

为什么@Resource注入会返回Null？ #

Tomcat中，@Resource这类依赖注入注解的处理是由Servlet容器负责的——但WebSocket端点并不属于Servlet容器管理的组件范畴：

• Servlet实例是由Tomcat的Servlet容器初始化并管理生命周期的，初始化阶段容器会自动扫描并处理@Resource注入；
• 而WebSocket端点是由Tomcat的WebSocket API实现（比如TomcatWebSocketContainer）在有新连接建立时实例化的，这个过程默认不会触发任何依赖注入逻辑，所以你用@Resource注入的DataSource自然就是null。

至于用DriverManager能正常连接，是因为它直接通过JDBC驱动建立连接，完全不依赖容器的注入机制，和WebSocket端点的实例化过程没有关联，所以不受影响。

可行的解决方案 #

方案1：手动通过JNDI查找数据源（最直接，无需额外配置） #

跳过@Resource注入，在WebSocket端点需要使用数据源的地方（比如@OnOpen方法或者业务方法中），直接通过JNDI API获取数据源：

import javax.naming.Context;
import javax.naming.InitialContext;
import javax.sql.DataSource;

// 在你的端点方法中添加如下代码
Context initCtx = new InitialContext();
Context envCtx = (Context) initCtx.lookup("java:comp/env");
DataSource dataSource = (DataSource) envCtx.lookup("jdbc/TicTacToeDb");
// 之后就可以用dataSource.getConnection()获取连接了

这种方式完全复用你已经配置好的连接池，不需要修改任何容器配置，是最快捷的解决办法。

方案2：将WebSocket端点转为CDI Bean（适合有CDI环境的项目） #

如果你的项目已经使用CDI（比如添加了Weld依赖），可以给WebSocket端点添加CDI注解（比如@Named或@RequestScoped），让CDI容器管理端点的实例化和依赖注入：

import javax.inject.Named;
import javax.websocket.server.ServerEndpoint;

@Named
@ServerEndpoint(value = "/tictactoe")
public class TicTacToeEndpoint {
    @Resource(lookup = "jdbc/TicTacToeDb")
    private DataSource dataSource;
    
    // 后续方法正常使用dataSource即可
}

注意：需要确保Tomcat启用了CDI支持，通常需要在项目中添加Weld的Tomcat集成依赖（比如org.jboss.weld.servlet:weld-servlet-shaded），并在WEB-INF下添加相关配置文件。

方案3：自定义端点实例化器（灵活控制注入逻辑） #

通过实现ServerEndpointConfig.Configurator，手动控制WebSocket端点的实例化过程，并在实例化后注入数据源：

1. 编写自定义Configurator：

import javax.naming.Context;
import javax.naming.InitialContext;
import javax.sql.DataSource;
import javax.websocket.server.ServerEndpointConfig;
import java.lang.reflect.Method;

public class DataSourceInjectorConfigurator extends ServerEndpointConfig.Configurator {
    @Override
    public <T> T createEndpoint(Class<T> endpointClass) throws InstantiationException {
        T endpoint = super.createEndpoint(endpointClass);
        try {
            // 通过JNDI获取数据源
            Context initCtx = new InitialContext();
            Context envCtx = (Context) initCtx.lookup("java:comp/env");
            DataSource dataSource = (DataSource) envCtx.lookup("jdbc/TicTacToeDb");
            
            // 假设你的端点有setDataSource方法
            Method setDataSourceMethod = endpointClass.getMethod("setDataSource", DataSource.class);
            setDataSourceMethod.invoke(endpoint, dataSource);
        } catch (Exception e) {
            throw new InstantiationException("Failed to inject DataSource into WebSocket endpoint", e);
        }
        return endpoint;
    }
}

2. 在端点上指定这个Configurator：

import javax.websocket.server.ServerEndpoint;

@ServerEndpoint(value = "/tictactoe", configurator = DataSourceInjectorConfigurator.class)
public class TicTacToeEndpoint {
    private DataSource dataSource;
    
    // 提供setter方法供Configurator调用
    public void setDataSource(DataSource dataSource) {
        this.dataSource = dataSource;
    }
    
    // 后续方法正常使用dataSource即可
}

这种方式不需要依赖CDI，完全由自己控制注入逻辑，适合对容器配置有严格限制的场景。

内容的提问来源于stack exchange，提问作者Eduard Valentin