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

@@ -96,8 +96,8 @@ export class CompareView extends AbstractWebview<
     const {
       commonResultSetNames,
       currentResultSetDisplayName,
-      fromResultSet,
-      toResultSet,
+      fromResultSetName,
+      toResultSetName,
     } = await this.findResultSetsToCompare(
       this.comparePair,
       selectedResultSetName,
@@ -106,7 +106,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);
       }
@@ -200,7 +204,7 @@ export class CompareView extends AbstractWebview<
   }
 
   private async findResultSetsToCompare(
-    { from, fromInfo, to, toInfo }: ComparePair,
+    { fromInfo, toInfo }: ComparePair,
     selectedResultSetName: string | undefined,
   ) {
     const {
@@ -210,21 +214,11 @@ export class CompareView extends AbstractWebview<
       toResultSetName,
     } = await findResultSetNames(fromInfo, toInfo, 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 {
       commonResultSetNames,
       currentResultSetDisplayName,
-      fromResultSet,
-      toResultSet,
+      fromResultSetName,
+      toResultSetName,
     };
   }
 
@@ -246,12 +240,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") {