Workers-RS 项目中 Socket 连接问题的分析与解决

问题背景

在使用 Workers-RS 项目（版本 0.4.1）与 Wrangler（版本 3.80.2）进行开发时，开发者遇到了一个关于 Socket 连接的报错问题。具体表现为尝试建立 PostgreSQL 数据库连接时，系统抛出错误信息："Specified address is empty string, contains unsupported characters or is too long"。

错误分析

该错误发生在使用 Socket::builder() 方法尝试连接 PostgreSQL 数据库时。开发者最初尝试直接传入完整的 PostgreSQL 连接字符串（包含协议、用户名、密码等信息），但这种方式不被 Socket 连接器接受。

解决方案

经过技术讨论和验证，正确的做法应该是：

1. 分离连接参数：将主机名、端口与认证信息分开处理
2. 分步构建连接：
   • 首先使用 Socket::builder() 仅指定主机和端口建立基础连接
   • 然后在 PostgreSQL 客户端配置中单独设置用户名、密码等认证信息

实现示例

以下是经过验证的正确实现方式：

#[derive(Clone)]
pub struct AppState {
    env: Env
}

impl AppState {
    pub fn new(env: Env) -> Self {
        AppState { env }
    }
    
    pub async fn new_pg_client(self) -> Client {
        // 从环境变量获取数据库连接字符串
        let database_url = match self.env.secret("DATABASE_URL") {
            Ok(v) => v.to_string(),
            Err(err) => panic!("env DATABASE_URL not found: {}", err),
        };

        // 解析连接配置
        let config: Config = Config::from_str(&database_url).unwrap_or_else(|e| {
            panic!("Config::from_str failed: {:?}", e);
        });

        // 提取主机和端口信息
        let host = match config.get_hosts()[0].clone() {
            Host::Tcp(v) => v,
            _ => panic!("Config::get_hosts should not be unix"),
        };
        let port = config.get_ports()[0];
        
        // 建立Socket连接
        let socket = Socket::builder()
            .secure_transport(SecureTransport::StartTls)
            .connect(host, port).unwrap_or_else(|e| {
                panic!("Socket::builder failed: {:?}", e);
            });
        
        // 使用配置和Socket建立数据库连接
        let (client, connection) = config
            .connect_raw(socket, PassthroughTls)
            .await.unwrap_or_else(|e| {
                panic!("Config::connect_raw failed: {:?}", e);
            });
        
        // 异步处理连接
        wasm_bindgen_futures::spawn_local(async move {
            if let Err(error) = connection.await {
                panic!("connection error: {:?}", error);
            }
        });
        
        client
    }
}

关键点说明

1. 连接字符串解析：使用 Config::from_str 方法解析完整的 PostgreSQL 连接字符串，从中提取出需要的连接参数。

2. Socket连接建立：仅使用主机名和端口号建立基础的 Socket 连接，这是解决原始报错的关键。

3. 安全传输设置：通过 .secure_transport(SecureTransport::StartTls) 指定使用 TLS 加密传输。

4. 异步连接处理：使用 wasm_bindgen_futures::spawn_local 来异步处理数据库连接，避免阻塞主线程。

总结

在 Workers-RS 项目中建立数据库连接时，需要注意连接参数的分阶段处理。直接传入完整连接字符串到 Socket 连接器会导致错误，正确的做法是将连接参数分拆，先建立基础 Socket 连接，再通过数据库客户端配置处理认证信息。这种方式不仅解决了报错问题，也使代码结构更加清晰合理。