fix: 修复update指定表别名时审核错误的问题(fix #249)

hanchuanchuan · Jul 23, 2020 · c3d39d8 · c3d39d8
1 parent 67e1ed1
commit c3d39d8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/session/session_inception.go b/session/session_inception.go
@@ -6333,7 +6333,11 @@ func (s *session) checkUpdate(node *ast.UpdateStmt, sql string) {
 			for _, l := range node.List {
 				// 未指定表别名时加默认设置
 				if l.Column.Table.L == "" && len(tableInfoList) == 1 {
-					l.Column.Table = model.NewCIStr(s.myRecord.TableInfo.Name)
+					if s.myRecord.TableInfo.AsName != "" {
+						l.Column.Table = model.NewCIStr(s.myRecord.TableInfo.AsName)
+					} else {
+						l.Column.Table = model.NewCIStr(s.myRecord.TableInfo.Name)
+					}
 				}
 
 				if s.checkFieldItem(l.Column, tableInfoList) {