TypeScript中如何优雅处理异常与WebSocket资源清理？

我写了一段源自Java模式的TypeScript代码：

let socket: WebSocket
try {
  socket = new WebSocket(path, options)
  // do some more stuff with socket
} catch (e) {
  socket.terminate()
  throw e
}

但IDE报错TS2454: Variable 'socket' is used before being assigned.——因为TypeScript无法确定catch块执行时socket已经完成赋值。

我试过两种替代方案，但感觉过于繁琐：

let socket: WebSocket | null = null
try {
  socket = new WebSocket(path, options)
  // do some more stuff with socket
} catch (e) {
  socket?.terminate()
  throw e
}

let socket: WebSocket | undefined
try {
  socket = new WebSocket(path, options)
  // do some more stuff with socket
} catch (e) {
  socket?.terminate()
  throw e
}

以下是几种更优的实现方式：

方案1：拆分实例化与赋值逻辑 #

通过内部变量先完成WebSocket实例化，再赋值给外部变量，让TypeScript的控制流分析更清晰：

let socket: WebSocket
try {
  const ws = new WebSocket(path, options)
  socket = ws
  // do some more stuff with socket
} catch (e) {
  socket?.terminate()
  throw e
}

这种写法和之前的可选链逻辑本质类似，但通过拆分步骤让代码意图更明确。

方案2：用IIFE封装逻辑（推荐） #

将整个逻辑封装在立即执行函数中，既避免外部变量的类型歧义，又能让TypeScript精准判断变量初始化状态：

const socket = (() => {
  const ws = new WebSocket(path, options)
  try {
    // do some more stuff with ws
    return ws
  } catch (e) {
    ws.terminate()
    throw e
  }
})()

这里new WebSocket放在try块外，确保后续catch块执行时ws一定已初始化，无需可选链；外部的socket始终是WebSocket类型，不会混入null或undefined。

如果new WebSocket本身可能抛出异常，可以调整为：

const socket = (() => {
  try {
    const ws = new WebSocket(path, options)
    // do some more stuff with ws
    return ws
  } catch (e) {
    // 实例化失败时无需调用terminate
    throw e
  }
})()

方案3：非空断言（不推荐） #

如果你能100%确认catch块执行时socket已赋值（实际存在运行时风险），可以用非空断言绕过类型检查：

let socket: WebSocket
try {
  socket = new WebSocket(path, options)
  // do some more stuff with socket
} catch (e) {
  socket!.terminate()
  throw e
}

这种写法会忽略TypeScript的安全检查，不建议在生产代码中使用。

内容的提问来源于stack exchange，提问作者Michael Wiles