fix(engine/mysql): Fix MySQL parser for query without trailing semicolon (#1282)

Author: timstudd

internal/endtoend/testdata/missing_semicolon/mysql/go/db.go

@@ -0,0 +1,11 @@
+-- https://github.com/kyleconroy/sqlc/issues/1198
+CREATE TABLE authors (
+  id   INT PRIMARY KEY,
+  name VARCHAR(255) NOT NULL,
+  bio  text
+);
+
+-- name: SetAuthor :exec
+UPDATE  authors
+SET     name = ?
+WHERE   id = ?

internal/endtoend/testdata/missing_semicolon/mysql/go/models.go

@@ -0,0 +1,12 @@
+{
+	"version": "1",
+	"packages": [
+		{
+			"path": "go",
+			"engine": "mysql",
+			"name": "querytest",
+			"schema": "query.sql",
+			"queries": "query.sql"
+		}
+	]
+}

internal/endtoend/testdata/missing_semicolon/mysql/go/query.sql.go

@@ -69,11 +69,16 @@ func (p *Parser) Parse(r io.Reader) ([]ast.Statement, error) {
 		text := stmtNodes[i].Text()
 		loc := strings.Index(string(blob), text)
 
+		stmtLen := len(text)
+		if text[stmtLen-1] == ';' {
+			stmtLen -= 1 // Subtract one to remove semicolon
+		}
+
 		stmts = append(stmts, ast.Statement{
 			Raw: &ast.RawStmt{
 				Stmt:         out,
 				StmtLocation: loc,
-				StmtLen:      len(text) - 1, // Subtract one to remove semicolon
+				StmtLen:      stmtLen,
 			},
 		})
 	}