참고소스 수정본

참고/playwright-main/packages/isomorphic/timeoutRunner.ts

@@ -0,0 +1,62 @@
+/**
+ * Copyright (c) Microsoft Corporation.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Hopefully, this file is never used in injected sources,
+// because it does not use `builtins.setTimeout` and similar,
+// and can break when clock emulation is engaged.
+
+/* eslint-disable no-restricted-globals */
+
+import { monotonicTime } from './time';
+
+export async function raceAgainstDeadline<T>(cb: () => Promise<T>, deadline: number): Promise<{ result: T, timedOut: false } | { timedOut: true }> {
+  let timer: NodeJS.Timeout | undefined;
+  // Note: not including "await" here truncates the async stack inside cb(), so always include it.
+  return await Promise.race([
+    cb().then(result => {
+      return { result, timedOut: false };
+    }),
+    new Promise<{ timedOut: true }>(resolve => {
+      if (!deadline)
+        return;
+      timer = setTimeout(() => resolve({ timedOut: true }), deadline - monotonicTime());
+    }),
+  ]).finally(() => {
+    clearTimeout(timer);
+  });
+}
+
+export async function pollAgainstDeadline<T>(callback: () => Promise<{ continuePolling: boolean, result: T }>, deadline: number, pollIntervals: number[] = [100, 250, 500, 1000]): Promise<{ result?: T, timedOut: boolean }> {
+  const lastPollInterval = pollIntervals.pop() ?? 1000;
+  let lastResult: T|undefined;
+  const wrappedCallback = () => Promise.resolve().then(callback);
+  while (true) {
+    const time = monotonicTime();
+    if (deadline && time >= deadline)
+      break;
+    const received = await raceAgainstDeadline(wrappedCallback, deadline);
+    if (received.timedOut)
+      break;
+    lastResult = (received as any).result.result;
+    if (!(received as any).result.continuePolling)
+      return { result: lastResult, timedOut: false };
+    const interval = pollIntervals!.shift() ?? lastPollInterval;
+    if (deadline && deadline <= monotonicTime() + interval)
+      break;
+    await new Promise(x => setTimeout(x, interval));
+  }
+  return { timedOut: true, result: lastResult };
+}