Add maximum statement count limit to SQL syntax checker

Signed-off-by: worksofliam <[email protected]>

Author: worksofliam

src/connection/syntaxChecker/checker.ts

@@ -1,6 +1,7 @@
 export const VALIDATOR_NAME = `VALIDATE_STATEMENT`;
 export const WRAPPER_NAME = `CHECKSTMTWRAPPED`;
 export const VALID_STATEMENT_LENGTH = 32740;
+export const MAX_STATEMENT_COUNT = 2000;
 
 export function getValidatorSource(schema: string, version: number) {
   return /*sql*/`

src/language/providers/problemProvider.ts

@@ -6,7 +6,7 @@ import { remoteAssistIsEnabled } from "./logic/available";
 import Configuration from "../../configuration";
 import { SQLStatementChecker, SqlSyntaxError } from "../../connection/syntaxChecker";
 import { StatementGroup, StatementType } from "../sql/types";
-import { VALID_STATEMENT_LENGTH } from "../../connection/syntaxChecker/checker";
+import { MAX_STATEMENT_COUNT, VALID_STATEMENT_LENGTH } from "../../connection/syntaxChecker/checker";
 import { getSqlDocument, isSafeDocument } from "./logic/parse";
 import path from "path";
 
@@ -125,8 +125,8 @@ async function validateSqlDocument(document: TextDocument, specificStatement?: n
     const sqlDocument = getSqlDocument(document);
 
     if (sqlDocument) {
-
       const allGroups = sqlDocument.getStatementGroups();
+
       let statementRanges: StatementRange[] = [];
 
       for (let i = 0; i < allGroups.length; i++) {
@@ -165,6 +165,13 @@ async function validateSqlDocument(document: TextDocument, specificStatement?: n
 
       if (statementRanges.length > 0) {
         const validStatements = statementRanges.filter(r => r.validate);
+
+        if (validStatements.length > MAX_STATEMENT_COUNT) {
+          const basename = document.fileName ? path.basename(document.fileName) : `Untitled`;
+          window.showWarningMessage(`${basename}: the SQL syntax checker cannot run because the statement limit has been reached (${validStatements.length} of ${MAX_STATEMENT_COUNT} max).`);
+          return;
+        }
+
         const invalidStatements = statementRanges.filter(r => !r.validate);
         const sqlStatementContents = validStatements.map(range => sqlDocument.content.substring(range.start, range.end));