Merge pull request #250 from hanchuanchuan/fix-update-alias-error

fix: 修复update指定表别名时审核错误的问题(fix #249)
hanchuanchuan · Jul 23, 2020 · ff008f0 · ff008f0
2 parents 67e1ed1 + c3d39d8
commit ff008f0
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) {