fix: mysql delete join panic (#2197)

* fix:mysql delete join panic

* fix: test

Author: nusr

internal/compiler/output_columns.go

@@ -404,9 +404,7 @@ func sourceTables(qc *QueryCatalog, node ast.Node) ([]*Table, error) {
 	var list *ast.List
 	switch n := node.(type) {
 	case *ast.DeleteStmt:
-		list = &ast.List{
-			Items: []ast.Node{n.Relation},
-		}
+		list = n.Relations
 	case *ast.InsertStmt:
 		list = &ast.List{
 			Items: []ast.Node{n.Relation},

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

@@ -0,0 +1,25 @@
+CREATE TABLE author (
+  id INT UNSIGNED AUTO_INCREMENT PRIMARY KEY,
+  name VARCHAR(255) NOT NULL
+);
+
+CREATE TABLE book (
+  id INT UNSIGNED AUTO_INCREMENT PRIMARY KEY,
+  title VARCHAR(255) NOT NULL
+);
+
+CREATE TABLE author_book (
+  author_id INT UNSIGNED NOT NULL,
+  book_id INT UNSIGNED NOT NULL,
+  CONSTRAINT `pk-author_book` PRIMARY KEY (author_id, book_id),
+  CONSTRAINT `fk-author_book-author-id` FOREIGN KEY (author_id) REFERENCES author (id),
+  CONSTRAINT `fk-author_book-book-id` FOREIGN KEY (book_id) REFERENCES book (id)
+);
+
+/* name: RemoveAllAuthorsFromTheGreatGatsby :exec */
+DELETE author_book
+FROM
+  author_book
+  INNER JOIN book ON book.id = author_book.book_id
+WHERE
+  book.title = 'The Great Gatsby';

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

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

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

@@ -0,0 +1,43 @@
+CREATE TABLE primary_table (
+        id bigint(20) unsigned NOT NULL AUTO_INCREMENT,
+        user_id bigint(20) unsigned NOT NULL,
+        PRIMARY KEY (id)
+);
+
+CREATE TABLE join_table (
+        id bigint(20) unsigned NOT NULL AUTO_INCREMENT,
+        primary_table_id bigint(20) unsigned NOT NULL,
+        other_table_id bigint(20) unsigned NOT NULL,
+        is_active tinyint(1) NOT NULL DEFAULT '0',
+        PRIMARY KEY (id)
+);
+
+-- name: DeleteJoin :exec
+DELETE jt.*,
+pt.*
+FROM
+        join_table as jt
+        JOIN primary_table as pt ON jt.primary_table_id = pt.id
+WHERE
+        jt.id = ?
+        AND pt.user_id = ?;
+
+-- name: DeleteLeftJoin :exec
+DELETE jt.*,
+pt.*
+FROM
+        join_table as jt
+        LEFT JOIN primary_table as pt ON jt.primary_table_id = pt.id
+WHERE
+        jt.id = ?
+        AND pt.user_id = ?;
+
+-- name: DeleteRightJoin :exec
+DELETE jt.*,
+pt.*
+FROM
+        join_table as jt
+        RIGHT JOIN primary_table as pt ON jt.primary_table_id = pt.id
+WHERE
+        jt.id = ?
+        AND pt.user_id = ?;