Merge pull request #12017 from Snuffleupagus/api-intentStates-Map

Convert the `PDFPageProxy.intentStates` property from an `Object` to a `Map`

src/core/chunked_stream.js

@@ -591,7 +591,7 @@ class ChunkedStreamManager {
     if (this.pdfNetworkStream) {
       this.pdfNetworkStream.cancelAllRequests(reason);
     }
-    for (const [, capability] of this._promisesByRequest) {
+    for (const capability of this._promisesByRequest.values()) {
       capability.reject(reason);
     }
   }

src/display/api.js

@@ -898,7 +898,7 @@ class PDFPageProxy {
 
     this.cleanupAfterRender = false;
     this.pendingCleanup = false;
-    this.intentStates = Object.create(null);
+    this._intentStates = new Map();
     this.destroyed = false;
   }
 
@@ -1003,10 +1003,11 @@ class PDFPageProxy {
     // this call to render.
     this.pendingCleanup = false;
 
-    if (!this.intentStates[renderingIntent]) {
+    let intentState = this._intentStates.get(renderingIntent);
-      this.intentStates[renderingIntent] = Object.create(null);
+    if (!intentState) {
+      intentState = Object.create(null);
+      this._intentStates.set(renderingIntent, intentState);
     }
-    const intentState = this.intentStates[renderingIntent];
 
     // Ensure that a pending `streamReader` cancel timeout is always aborted.
     if (intentState.streamReaderCancelTimeout) {
@@ -1128,14 +1129,15 @@ class PDFPageProxy {
     }
 
     const renderingIntent = "oplist";
-    if (!this.intentStates[renderingIntent]) {
+    let intentState = this._intentStates.get(renderingIntent);
-      this.intentStates[renderingIntent] = Object.create(null);
+    if (!intentState) {
+      intentState = Object.create(null);
+      this._intentStates.set(renderingIntent, intentState);
     }
-    const intentState = this.intentStates[renderingIntent];
     let opListTask;
 
     if (!intentState.opListReadCapability) {
-      opListTask = {};
+      opListTask = Object.create(null);
       opListTask.operatorListChanged = operatorListChanged;
       intentState.opListReadCapability = createPromiseCapability();
       intentState.renderTasks = [];
@@ -1222,8 +1224,7 @@ class PDFPageProxy {
     this._transport.pageCache[this._pageIndex] = null;
 
     const waitOn = [];
-    Object.keys(this.intentStates).forEach(intent => {
+    for (const [intent, intentState] of this._intentStates) {
-      const intentState = this.intentStates[intent];
       this._abortOperatorList({
         intentState,
         reason: new Error("Page was destroyed."),
@@ -1232,16 +1233,13 @@ class PDFPageProxy {
 
       if (intent === "oplist") {
         // Avoid errors below, since the renderTasks are just stubs.
-        return;
+        continue;
       }
-      intentState.renderTasks.forEach(function (renderTask) {
+      for (const internalRenderTask of intentState.renderTasks) {
-        const renderCompleted = renderTask.capability.promise.catch(
+        waitOn.push(internalRenderTask.completed);
-          function () {}
+        internalRenderTask.cancel();
-        ); // ignoring failures
+      }
-        waitOn.push(renderCompleted);
+    }
-        renderTask.cancel();
-      });
-    });
     this.objs.clear();
     this.annotationsPromise = null;
     this.pendingCleanup = false;
@@ -1264,22 +1262,16 @@ class PDFPageProxy {
    * @private
    */
   _tryCleanup(resetStats = false) {
-    if (
+    if (!this.pendingCleanup) {
-      !this.pendingCleanup ||
-      Object.keys(this.intentStates).some(intent => {
-        const intentState = this.intentStates[intent];
-        return (
-          intentState.renderTasks.length !== 0 ||
-          !intentState.operatorList.lastChunk
-        );
-      })
-    ) {
       return false;
     }
+    for (const { renderTasks, operatorList } of this._intentStates.values()) {
+      if (renderTasks.length !== 0 || !operatorList.lastChunk) {
+        return false;
+      }
+    }
 
-    Object.keys(this.intentStates).forEach(intent => {
+    this._intentStates.clear();
-      delete this.intentStates[intent];
-    });
     this.objs.clear();
     this.annotationsPromise = null;
     if (resetStats && this._stats) {
@@ -1293,7 +1285,7 @@ class PDFPageProxy {
    * @private
    */
   _startRenderPage(transparency, intent) {
-    const intentState = this.intentStates[intent];
+    const intentState = this._intentStates.get(intent);
     if (!intentState) {
       return; // Rendering was cancelled.
     }
@@ -1343,7 +1335,7 @@ class PDFPageProxy {
     );
     const reader = readableStream.getReader();
 
-    const intentState = this.intentStates[args.intent];
+    const intentState = this._intentStates.get(args.intent);
     intentState.streamReader = reader;
 
     const pump = () => {
@@ -1428,13 +1420,12 @@ class PDFPageProxy {
     }
     // Remove the current `intentState`, since a cancelled `getOperatorList`
     // call on the worker-thread cannot be re-started...
-    Object.keys(this.intentStates).some(intent => {
+    for (const [intent, curIntentState] of this._intentStates) {
-      if (this.intentStates[intent] === intentState) {
+      if (curIntentState === intentState) {
-        delete this.intentStates[intent];
+        this._intentStates.delete(intent);
-        return true;
+        break;
       }
-      return false;
+    }
-    });
     // ... and force clean-up to ensure that any old state is always removed.
     this.cleanup();
   }
@@ -2650,6 +2641,13 @@ const InternalRenderTask = (function InternalRenderTaskClosure() {
       this._canvas = params.canvasContext.canvas;
     }
 
+    get completed() {
+      return this.capability.promise.catch(function () {
+        // Ignoring errors, since we only want to know when rendering is
+        // no longer pending.
+      });
+    }
+
     initializeGraphics(transparency = false) {
       if (this.cancelled) {
         return;