go/ast/astutil: update Equal to handle new ast.IndexListExpr type

dominikh · Mar 28, 2022 · 11e5f96 · 11e5f96
1 parent 27f1505
commit 11e5f96
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/go/ast/astutil/util.go b/go/ast/astutil/util.go
@@ -278,6 +278,17 @@ func Equal(a, b ast.Node) bool {
 	case *ast.IndexExpr:
 		b := b.(*ast.IndexExpr)
 		return Equal(a.X, b.X) && Equal(a.Index, b.Index)
+	case *ast.IndexListExpr:
+		b := b.(*ast.IndexListExpr)
+		if len(a.Indices) != len(b.Indices) {
+			return false
+		}
+		for i, v := range a.Indices {
+			if !Equal(v, b.Indices[i]) {
+				return false
+			}
+		}
+		return Equal(a.X, b.X)
 	case *ast.KeyValueExpr:
 		b := b.(*ast.KeyValueExpr)
 		return Equal(a.Key, b.Key) && Equal(a.Value, b.Value)