Merge pull request #1775 from visualfc/walk2

ast: fix forPhrase.end
goplus · Feb 23, 2024 · 36d3812 · 36d3812
2 parents 05a2651 + d9d7d4d
commit 36d3812
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ast/ast_gop.go b/ast/ast_gop.go
@@ -205,7 +205,12 @@ type ForPhrase struct {
 func (p *ForPhrase) Pos() token.Pos { return p.For }
 
 // End returns position of first character immediately after the node.
-func (p *ForPhrase) End() token.Pos { return p.X.End() }
+func (p *ForPhrase) End() token.Pos {
+	if p.Cond != nil {
+		return p.Cond.End()
+	}
+	return p.X.End()
+}
 
 func (p *ForPhrase) exprNode() {}