TS实现一个“可取消等待 + 单例 Promise + WeakRef 清理”队列

这个需求,本质上是 "Promise 去重 + 可取消等待" 的组合,而且要做到 高性能、无锁、无多余状态切换

一个 最优解模式:单例执行 + 等待者可取消 + 结果复用,并且是 TypeScript 最佳实践。


✅ 目标总结(我重新整理成技术语言)

有一个耗时异步函数 expensiveTask()

  • 同一时间只能执行一次
  • 后续重复调用如果发现正在执行 → 不再触发新的执行,而是等待当前执行的结果
  • 等待中的调用如果被取消 → 立即结束,不再等待,也不拿结果
  • 执行完成后,所有仍在等待的调用都拿到同一个结果(成功或失败)
  • 调用次数可能非常多 → 必须高性能

✅ 最佳实现:可取消等待 + 单例 Promise + WeakRef 清理

下面是最终推荐的实现:

ts 复制代码
type CancelToken = {
    cancelled: boolean;
};

export class SingletonAsync<T> {
    private currentPromise: Promise<T> | null = null;
    private waiters = new Set<{
        resolve: (v: T) => void;
        reject: (e: any) => void;
        token: CancelToken;
    }>();

    constructor(private readonly task: () => Promise<T>) {}

    run(token?: CancelToken): Promise<T> {
        const cancelToken = token ?? { cancelled: false };

        // 若已有执行中的任务 → 直接加入等待队列
        if (this.currentPromise) {
            return new Promise<T>((resolve, reject) => {
                if (cancelToken.cancelled) {
                    reject(new Error("Cancelled"));
                    return;
                }
                this.waiters.add({ resolve, reject, token: cancelToken });
            });
        }

        // 启动新的任务
        this.currentPromise = this.task()
            .then(result => {
                // 通知所有未取消的等待者
                for (const w of this.waiters) {
                    if (!w.token.cancelled) w.resolve(result);
                }
                return result;
            })
            .catch(err => {
                for (const w of this.waiters) {
                    if (!w.token.cancelled) w.reject(err);
                }
                throw err;
            })
            .finally(() => {
                // 清理
                this.waiters.clear();
                this.currentPromise = null;
            });

        return this.currentPromise;
    }
}

✅ 使用方式

ts 复制代码
const expensiveRunner = new SingletonAsync(async () => {
    console.log("真正执行一次任务...");
    await new Promise(r => setTimeout(r, 3000));
    return 42;
});

// 调用方
const token = { cancelled: false };

expensiveRunner.run(token).then(console.log).catch(console.error);

// 某处取消
token.cancelled = true;

✅ 行为完全符合你的需求

需求 是否满足 说明
1. 方法耗时 不影响
2. 多次调用只执行一次 currentPromise 保证单例执行
3. 等待中的调用可取消 token.cancelled = true 立即 reject
4. 高性能 无锁、无多余 promise、无事件循环压力

✅ 为什么这是性能最好的方案?

✔ 无重复执行

所有重复调用都复用同一个 Promise。

✔ 等待者是 Set,O(1) 插入删除

适合大量调用。

✔ 无轮询、无额外事件队列

只有真正执行一次任务。

✔ 可取消等待,不影响主任务

取消只影响当前调用,不会破坏正在执行的任务。

相关推荐
牛奶5 小时前
ts随笔:面向对象与高级类型
前端·面试·typescript
SuperEugene6 小时前
接口类型管理:从 any 到有组织的 api.d.ts
前端·面试·typescript
牛奶7 小时前
ts随笔:基础与类型系统
前端·面试·typescript
随逸17712 小时前
《从零搭建NestJS项目》
数据库·typescript
唐璜Taro1 天前
Vue3 + TypeScript 后台管理系统完整方案
前端·javascript·typescript
Wect1 天前
LeetCode 530. 二叉搜索树的最小绝对差:两种解法详解(迭代+递归)
前端·算法·typescript
程序员林北北1 天前
【前端进阶之旅】typescriot的数据类型讲解(二)
前端·javascript·vue.js·react.js·typescript
火车叼位1 天前
TypeScript 类型体操:如何精准控制可选参数的“去留”
前端·typescript
却尘2 天前
你写的 TypeScript,其实只是穿了件类型外套的 JavaScript
前端·typescript
We་ct2 天前
LeetCode 102. 二叉树的层序遍历:图文拆解+代码详解
前端·算法·leetcode·typescript