fix: cannot parse SQLite upsert statement (#1732)

Author: hakobera

go.mod

@@ -3,7 +3,7 @@ module github.com/kyleconroy/sqlc
 go 1.18
 
 require (
-	github.com/antlr/antlr4/runtime/Go/antlr v0.0.0-20220209173558-ad29539cd2e9
+	github.com/antlr/antlr4/runtime/Go/antlr v0.0.0-20220626175859-9abda183db8e
 	github.com/bytecodealliance/wasmtime-go v0.38.1
 	github.com/davecgh/go-spew v1.1.1
 	github.com/go-sql-driver/mysql v1.6.0

go.sum

@@ -1,8 +1,8 @@
 github.com/BurntSushi/toml v0.3.1 h1:WXkYYl6Yr3qBf1K79EBnL4mak0OimBfB0XUf9Vl28OQ=
 github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
 github.com/Masterminds/semver/v3 v3.1.1/go.mod h1:VPu/7SZ7ePZ3QOrcuXROw5FAcLl4a0cBrbBpGY/8hQs=
-github.com/antlr/antlr4/runtime/Go/antlr v0.0.0-20220209173558-ad29539cd2e9 h1:zvkJv+9Pxm1nnEMcKnShREt4qtduHKz4iw4AB4ul0Ao=
-github.com/antlr/antlr4/runtime/Go/antlr v0.0.0-20220209173558-ad29539cd2e9/go.mod h1:F7bn7fEU90QkQ3tnmaTx3LTKLEDqnwWODIYppRQ5hnY=
+github.com/antlr/antlr4/runtime/Go/antlr v0.0.0-20220626175859-9abda183db8e h1:bt6SW1eSSvdmmsG0KqyxYXorcTnFBTX7hfVR1+68+jg=
+github.com/antlr/antlr4/runtime/Go/antlr v0.0.0-20220626175859-9abda183db8e/go.mod h1:F7bn7fEU90QkQ3tnmaTx3LTKLEDqnwWODIYppRQ5hnY=
 github.com/benbjohnson/clock v1.1.0 h1:Q92kusRqC1XV2MjkWETPvjJVqKetz1OzxZB7mHJLju8=
 github.com/benbjohnson/clock v1.1.0/go.mod h1:J11/hYXuz8f4ySSvYwY0FKfm+ezbsZBKZxNJlLklBHA=
 github.com/bytecodealliance/wasmtime-go v0.38.1 h1:eqrJmy1nR/uyxZC4OydXWpvGeEHzprW1Fao6eXvNnEE=

internal/endtoend/testdata/upsert/sqlite/go/db.go

@@ -0,0 +1,27 @@
+-- https://github.com/kyleconroy/sqlc/issues/1728
+
+CREATE TABLE IF NOT EXISTS locations (
+    id              INTEGER PRIMARY KEY,
+    name            TEXT    NOT NULL,
+    address         TEXT    NOT NULL,
+    zip_code        INT     NOT NULL,
+    latitude        REAL    NOT NULL,
+    longitude       REAL    NOT NULL,
+    UNIQUE(name)
+);
+
+/* name: UpsertLocation :exec */
+INSERT INTO locations (
+    name,
+    address,
+    zip_code,
+    latitude,
+    longitude
+)
+VALUES (?, ?, ?, ?, ?)
+ON CONFLICT(name) DO UPDATE SET 
+    name = excluded.name,
+    address = excluded.address,
+    zip_code = excluded.zip_code,
+    latitude = excluded.latitude,
+    longitude = excluded.longitude;

internal/endtoend/testdata/upsert/sqlite/go/models.go

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

internal/endtoend/testdata/upsert/sqlite/go/query.sql.go

@@ -355,8 +355,8 @@ upsert_clause:
     )? DO_ (
         NOTHING_
         | UPDATE_ SET_ (
-            (column_name | column_name_list) EQ expr (
-                COMMA (column_name | column_name_list) EQ expr
+            (column_name | column_name_list) ASSIGN expr (
+                COMMA (column_name | column_name_list) ASSIGN expr
             )* (WHERE_ expr)?
         )
     )