From 0dd49779e3aa7bbaf337ca9d1c8155edf4eebb40 Mon Sep 17 00:00:00 2001
From: Yaiba <4yaiba@gmail.com>
Date: Thu, 7 Mar 2024 17:24:46 -0600
Subject: [PATCH] SafeToSql util fn

---
 schema/validator.go | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/schema/validator.go b/schema/validator.go
index 5df56c8..0aa9b35 100644
--- a/schema/validator.go
+++ b/schema/validator.go
@@ -244,22 +244,22 @@ func (c *ContextValidator) visitActions(actions []Action) error {
 		}
 
 		for _, statement := range a.Statements {
-			stmt, err := actparser.Parse(statement)
+			actStmt, err := actparser.Parse(statement)
 			if err != nil {
 				return fmt.Errorf("%w: %s", err, statement)
 			}
 
-			switch s := stmt.(type) {
+			switch s := actStmt.(type) {
 			case *actparser.DMLStmt:
-				astTree, err := sqlparser.ParseSql(statement, 1, nil, false)
+				sqlStmt, err := sqlparser.ParseSql(statement, 1, nil, false, false)
 				if err != nil {
 					return fmt.Errorf("%w: %s", err, statement)
 				}
-				if _, err := astTree.ToSQL(); err != nil {
+				if _, err := tree.SafeToSQL(sqlStmt); err != nil {
 					return fmt.Errorf("%w: %s", err, statement)
 				}
 				// TODO: validate reference in SQL statement
-				//switch t := astTree.(type) {
+				//switch t := sqlStmt.(type) {
 				//case *tree.Select:
 			case *actparser.ActionCallStmt:
 				if _, ok := c.actionCtx[s.Method]; !ok {