Remove inline comments from DDL files via ScriptRunner (cmu-db#159)

shaharuk-yb · Apr 26, 2022 · 29936f3 · 29936f3
1 parent 72f7d02
commit 29936f3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/java/com/oltpbenchmark/util/ScriptRunner.java b/src/main/java/com/oltpbenchmark/util/ScriptRunner.java
@@ -93,6 +93,8 @@ private void runScript(Connection conn, Reader reader) throws IOException, SQLEx
                     command = new StringBuffer();
                 }
                 String trimmedLine = line.trim();
+                line = line.replaceAll("\\-\\-.*$", ""); // remove comments in line;
+
                 if (trimmedLine.startsWith("--") || trimmedLine.startsWith("//")) {
                     LOG.debug(trimmedLine);
                 } else if (trimmedLine.length() < 1) {
@@ -103,8 +105,6 @@ private void runScript(Connection conn, Reader reader) throws IOException, SQLEx
 
                     try (Statement statement = conn.createStatement()) {
 
-                        // println(command);
-
                         boolean hasResults = false;
                         final String sql = command.toString().trim();
                         if (stopOnError) {