fix: replace table_name with qualified_table_name

sqlc-dev · Nov 26, 2024 · 5849df4 · 5849df4
1 parent 5b6ffd3
commit 5849df4
Show file tree

Hide file tree

Showing 4 changed files with 728 additions and 786 deletions.
diff --git a/internal/engine/sqlite/convert.go b/internal/engine/sqlite/convert.go
@@ -1003,7 +1003,7 @@ func (c *cc) convertTablesOrSubquery(n []parser.ITable_or_subqueryContext) []ast
 }
 
 type Update_stmt interface {
-	Table_name() parser.ITable_nameContext
+	Qualified_table_name() parser.IQualified_table_nameContext
 	GetStart() antlr.Token
 	AllColumn_name() []parser.IColumn_nameContext
 	WHERE_() antlr.TerminalNode
@@ -1017,7 +1017,7 @@ func (c *cc) convertUpdate_stmtContext(n Update_stmt) ast.Node {
 	}
 
 	relations := &ast.List{}
-	tableName := n.Table_name().GetText()
+	tableName := n.Qualified_table_name().GetText()
 	rel := ast.RangeVar{
 		Relname:  &tableName,
 		Location: n.GetStart().GetStart(),

diff --git a/internal/engine/sqlite/parser/SQLiteParser.g4 b/internal/engine/sqlite/parser/SQLiteParser.g4
@@ -468,7 +468,7 @@ compound_operator:
 update_stmt:
     with_clause? UPDATE_ (
         OR_ (ROLLBACK_ | ABORT_ | REPLACE_ | FAIL_ | IGNORE_)
-    )? table_name (AS_? table_alias)? SET_ (column_name | column_name_list) ASSIGN expr (
+    )? qualified_table_name SET_ (column_name | column_name_list) ASSIGN expr (
         COMMA (column_name | column_name_list) ASSIGN expr
     )* (FROM_ table_or_subquery)? (WHERE_ expr)? returning_clause?
 ;

diff --git a/internal/engine/sqlite/parser/SQLiteParser.interp b/internal/engine/sqlite/parser/SQLiteParser.interp