Add tables suggestion for ALTER, INSERT, UPDATE

chaisql · Aug 26, 2020 · 3766f80 · 3766f80
1 parent 051eb5e
commit 3766f80
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/sql/parser/alter.go b/sql/parser/alter.go
@@ -19,7 +19,9 @@ func (p *Parser) parseAlterStatement() (query.AlterStmt, error) {
 	// Parse table name.
 	stmt.TableName, err = p.parseIdent()
 	if err != nil {
-		return stmt, err
+		pErr := err.(*ParseError)
+		pErr.Expected = []string{"table_name"}
+		return stmt, pErr
 	}
 
 	// Parse "RENAME".

diff --git a/sql/parser/insert.go b/sql/parser/insert.go
@@ -22,7 +22,9 @@ func (p *Parser) parseInsertStatement() (query.InsertStmt, error) {
 	// Parse table name
 	stmt.TableName, err = p.parseIdent()
 	if err != nil {
-		return stmt, err
+		pErr := err.(*ParseError)
+		pErr.Expected = []string{"table_name"}
+		return stmt, pErr
 	}
 
 	valueParser := p.parseParamOrDocument

diff --git a/sql/parser/update.go b/sql/parser/update.go
@@ -15,7 +15,9 @@ func (p *Parser) parseUpdateStatement() (*planner.Tree, error) {
 	// Parse table name
 	cfg.TableName, err = p.parseIdent()
 	if err != nil {
-		return nil, err
+		pErr := err.(*ParseError)
+		pErr.Expected = []string{"table_name"}
+		return nil, pErr
 	}
 
 	// Parse clause: SET or UNSET.