fix: find_params bug

Author: qykr

internal/compiler/find_params.go

@@ -111,6 +111,16 @@ func (p paramSearch) Visit(node ast.Node) astutils.Visitor {
 		}
 
 	case *ast.UpdateStmt:
+		var targetRV *ast.RangeVar
+		for _, relation := range n.Relations.Items {
+			rv, ok := relation.(*ast.RangeVar)
+			if !ok {
+				continue
+			}
+			targetRV = rv
+			break
+		}
+
 		for _, item := range n.TargetList.Items {
 			target, ok := item.(*ast.ResTarget)
 			if !ok {
@@ -120,13 +130,7 @@ func (p paramSearch) Visit(node ast.Node) astutils.Visitor {
 			if !ok {
 				continue
 			}
-			for _, relation := range n.Relations.Items {
-				rv, ok := relation.(*ast.RangeVar)
-				if !ok {
-					continue
-				}
-				*p.refs = append(*p.refs, paramRef{parent: target, ref: ref, rv: rv})
-			}
+			*p.refs = append(*p.refs, paramRef{parent: target, ref: ref, rv: targetRV})
 			p.seen[ref.Location] = struct{}{}
 		}
 		if n.LimitCount != nil {