Question

我想了解Roslyn Compiler是如何工作的。

我正在尝试编写简单的函数，它将分析代码并将循环语句从增加变为减少。

例如更改：

for(int i=0; i<10; i++)
    int a = i;

为：

for(int i=9; i>=0; i--)
    int a=i;

我已编写arleady代码来查找该循环，但我不知道如何编辑它。

这是我到目前为止所做的：

SyntaxTree tree = CSharpSyntaxTree.ParseText(
    @"using System;
    using System.Collections.Generic;
    using System.Text;

        static void Main(string[] args)
        {
        for(int i=0; i<10; i++)
        int a = i;
        }");
var root = (CompilationUnitSyntax)tree.GetRoot();

IEnumerable<ForStatementSyntax> forStatementSyntaxs = root.DescendantNodes().OfType<ForStatementSyntax>();

ForStatementSyntax forStatementSyntax = forStatementSyntaxs.First();
ExpressionSyntax expressionSyntax = forStatementSyntax.Incrementors.First();

我想知道如何改变Roslyn'for loop'声明。另外，如何更改该循环包含的表达式。

Answer 1

创建一个SyntaxRewriter，它是一个访问者，可用于使用您创建的新节点替换语法树中的节点。还有一些方法可用于创建新节点并对其进行更改的所有节点。

在您的情况下，您希望创建一个具有更改的条件和语句的新ForStatementSyntax节点。因此，您可以使用WithCondition()和WithStatement()方法分别进行更新。您可以手动创建新节点，也可以使用多种SyntaxFactory方法之一解析字符串。

如，

class Rewriter : CSharpSyntaxRewriter
{
    public override SyntaxNode VisitForStatement(ForStatementSyntax node)
    {
        // update the current node with the new condition and statement
        return node.WithCondition(
            SyntaxFactory.ParseExpression("i>=0")
        ).WithStatement(
            SyntaxFactory.ParseStatement(@"{
              Console.WriteLine(i);
              Console.WriteLine(i*2);
            }")
        );
    }
}

创建访问者后，您只需使用它即可。

var root = SyntaxFactory.ParseCompilationUnit(
@"using System;
using System.Collections.Generic;
using System.Text;

static void Main(string[] args)
{
    for(int i=0; i<10; i++)
        int a = i;
}");
var rewritten = new Rewriter().Visit(root);

另一方面，对于更简单的重写任务，您不一定需要重写器。您可以使用替换扩展方法替换树中的节点。

var forStmt = root.DescendantNodes().OfType<ForStatementSyntax>().Single();
var rewritten = root.ReplaceNode(forStmt,
    forStmt.WithCondition(
        SyntaxFactory.ParseExpression("i>=0")
    ).WithStatement(
        SyntaxFactory.ParseStatement(@"    {
          Console.WriteLine(i);
          Console.WriteLine(i*2);
        }
      ")
    ));

在罗斯林编辑循环

1 个答案: