feat(wait): 支持取消等待

src/wait.test.ts

@@ -11,3 +11,13 @@ test('表现正常', async () => {
     true,
   )
 })
+
+test('可取消等待', async () => {
+  const cb = jest.fn()
+  const w = wait(500)
+  w.then(cb)
+  w.cancel()
+  await wait(600)
+
+  expect(cb).toBeCalledTimes(0)
+})

src/wait.ts

@@ -1,3 +1,10 @@
+export interface WaitReturn extends Promise<void> {
+  /**
+   * 取消等待，不执行后续逻辑。
+   */
+  cancel: () => void,
+}
+
 /**
  * 等待一段时间。
  *
@@ -9,8 +16,13 @@
  * })
  * ```
  */
-export function wait(milliseconds: number): Promise<void> {
-  return new Promise(
-    resolve => setTimeout(resolve, milliseconds),
+export function wait(milliseconds: number): WaitReturn {
+  let timer!: number
+  const p = new Promise(
+    resolve => {
+      timer = setTimeout(resolve, milliseconds) as any
+    },
   )
+  ;(p as any).cancel = () => clearTimeout(timer)
+  return p as any
 }