Read raw result sets in compareResults

This moves reading of the result sets to the `compareResults` method
since raw result sets don't need to be read for interpreted results and
the `findResultSetsToCompare` method is shared between the two types
of results.

Author: koesie10

extensions/ql-vscode/src/compare/compare-view.ts

@@ -144,7 +144,7 @@ export class CompareView extends AbstractWebview<
     panel.reveal(undefined, true);
 
     await this.waitForPanelLoaded();
-    const { currentResultSetDisplayName, fromResultSet, toResultSet } =
+    const { currentResultSetDisplayName, fromResultSetName, toResultSetName } =
       await this.findResultSetsToCompare(
         this.comparePair,
         selectedResultSetName,
@@ -153,7 +153,11 @@ export class CompareView extends AbstractWebview<
       let result: RawQueryCompareResult | undefined;
       let message: string | undefined;
       try {
-        result = this.compareResults(fromResultSet, toResultSet);
+        result = await this.compareResults(
+          this.comparePair,
+          fromResultSetName,
+          toResultSetName,
+        );
       } catch (e) {
         message = getErrorMessage(e);
       }
@@ -230,7 +234,7 @@ export class CompareView extends AbstractWebview<
   }
 
   private async findResultSetsToCompare(
-    { from, fromInfo, to, toInfo, commonResultSetNames }: ComparePair,
+    { fromInfo, toInfo, commonResultSetNames }: ComparePair,
     selectedResultSetName: string | undefined,
   ) {
     const { currentResultSetDisplayName, fromResultSetName, toResultSetName } =
@@ -241,20 +245,10 @@ export class CompareView extends AbstractWebview<
         selectedResultSetName,
       );
 
-    const fromResultSet = await this.getResultSet(
-      fromInfo.schemas,
-      fromResultSetName,
-      from.completedQuery.query.resultsPaths.resultsPath,
-    );
-    const toResultSet = await this.getResultSet(
-      toInfo.schemas,
-      toResultSetName,
-      to.completedQuery.query.resultsPaths.resultsPath,
-    );
     return {
       currentResultSetDisplayName,
-      fromResultSet,
-      toResultSet,
+      fromResultSetName,
+      toResultSetName,
     };
   }
 
@@ -276,12 +270,24 @@ export class CompareView extends AbstractWebview<
     return await this.cliServer.bqrsDecode(resultsPath, resultSetName);
   }
 
-  private compareResults(
-    fromResults: DecodedBqrsChunk,
-    toResults: DecodedBqrsChunk,
-  ): RawQueryCompareResult {
+  private async compareResults(
+    { from, fromInfo, to, toInfo }: ComparePair,
+    fromResultSetName: string,
+    toResultSetName: string,
+  ): Promise<RawQueryCompareResult> {
+    const fromResultSet = await this.getResultSet(
+      fromInfo.schemas,
+      fromResultSetName,
+      from.completedQuery.query.resultsPaths.resultsPath,
+    );
+    const toResultSet = await this.getResultSet(
+      toInfo.schemas,
+      toResultSetName,
+      to.completedQuery.query.resultsPaths.resultsPath,
+    );
+
     // Only compare columns that have the same name
-    return resultsDiff(fromResults, toResults);
+    return resultsDiff(fromResultSet, toResultSet);
   }
 
   private async openQuery(kind: "from" | "to") {