fix

raw-labs · Jul 5, 2024 · daeb66b · daeb66b
1 parent 940cba0
commit daeb66b
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/sql-client/src/main/scala/raw/client/sql/antlr4/RawSqlSyntaxAnalyzer.scala b/sql-client/src/main/scala/raw/client/sql/antlr4/RawSqlSyntaxAnalyzer.scala
@@ -26,9 +26,8 @@ class RawSqlSyntaxAnalyzer(val positions: Positions) extends Parsers(positions)
     val source = StringSource(s)
     val rawErrorListener = new RawSqlErrorListener()
 
-    val striped = s.stripTrailing()
-    val withoutNewLine = if(striped.endsWith("\n")) striped.substring(0, striped.length - 1) else striped
-    val lexer = new PsqlLexer(CharStreams.fromString(withoutNewLine))
+    val striped = s.stripTrailing().stripSuffix("\r\n").stripSuffix("\n").stripSuffix("\r")
+    val lexer = new PsqlLexer(CharStreams.fromString(striped))
     lexer.removeErrorListeners()
     lexer.addErrorListener(rawErrorListener)