使用 Roslyn,从字符串文字参数中找到调用方法

With Roslyn, find calling method from string literal parameter

我正在尝试使用 Roslyn 替换实用程序中的一些旧的、缓慢的代码,该实用程序在我们的源代码中搜索未包装在 X() 函数调用中的字符串文字(包装在 X() 将被 t运行 预定)。

我能够使用语法树很容易地获取字符串文字,并确定了 大多数 将它们包裹在 X() 中的地方。我在做什么:给定一个 LiteralExpressionSyntax 对象,我发现这给了我函数调用,我可以用正则表达式匹配它。

s.Parent.Parent.Parent.ToFullString()

当字符串文字被分成两行时,我很快 运行 遇到了一个问题。那时我意识到我检查它是否在 X() 调用中的方法很差,因为我必须继续将 .Parent 添加到链中。虽然我可以写一些东西来向后爬树,但这似乎不是正确的方法(而且可能不会表现很好)。

我一直在尝试找到一种方法,给定一个字符串文字语法节点,以确定它是否是方法调用中的参数。我一直无法找到一种从语法树到语义模型的合适方法来找到我正在寻找的东西。我什至不确定这是否是正确的方法,或者我是否遗漏了一些明显的东西。

我能够使用 SymbolFinder.FindSymbolAtPositionAsync 来获取符号,但是遇到同样的问题 - 我不能只传递字符串文字参数的位置,我必须传递(正确的)parent 我又回到了起点。

我希望避免多次遍历语法树,因为这会减慢速度。我可以在大约 1 秒内解析 553 个文件,但是一旦我尝试循环以解决这些多行情况,我最多需要 12-13 秒。

以防万一我在这篇中篇小说中迷失了你(抱歉),这是我希望弄清楚的: 将字符串文字作为参数传递给一种方法,有没有一种简单的方法可以确定该方法是什么?

这是一些示例代码 - 我已经用 Convert.ToString 替换了对 X() 函数的调用,只是为了模拟我正在搜索的代码(我不得不添加对我们的一个 DLL 的引用,所以我将调用切换到 Convert.ToString(),这样我就可以在这个例子中引用 mscorlib。

static void TestAttempt()
{

  string source = @"
  Imports System
  Namespace Exceptions
    Public NotInheritable Class ExampleException
      Inherits Validation

      Public Sub New()
        Convert.ToString(""Ignore me 1"")
        Console.WriteLine(""Report me"")
        Console.WriteLine(Convert.ToString(""Ignore me 2""))

        MyBase.New(Convert.ToString(""Ignore me 3, "" & _
                                    ""Because I'm already translated.""))
      End Sub
    End Class
  End Namespace";

  var tree = VisualBasicSyntaxTree.ParseText(source);

  var syntaxRoot = tree.GetRoot();
  int i = 0;

  foreach (var s in syntaxRoot.DescendantNodes().OfType<LiteralExpressionSyntax>())
  {
    // things to skip:
    if (s == null) { continue; }
    if (s.Kind() != SyntaxKind.StringLiteralExpression) { continue; }

    var Mscorlib = PortableExecutableReference.CreateFromFile(@"C:\Windows\Microsoft.NET\Framework\v4.0.30319\mscorlib.dll");
    var compilation = VisualBasicCompilation.Create("MyCompilation", syntaxTrees: new[] { tree }, references: new[] { Mscorlib });
    var model = compilation.GetSemanticModel(tree);

    var symbol = SymbolFinder.FindSymbolAtPositionAsync(model, s.Parent.Parent.Parent.Span.Start, new AdhocWorkspace()).Result;

    if (symbol.ToDisplayString().EndsWith("Convert")) { continue; }

    Console.WriteLine(symbol);
    Console.WriteLine($"  Reported: {s.ToString()}");
    i++;
  }

  Console.WriteLine();
  Console.WriteLine($"Total: {i}");
}

警告说我以前从未使用过 Roslyn,而且我通常在 VB.NET 中编写代码,我想我拼凑了一些似乎可以做你想做的事情(使用 LINQPad 和很多 Dump() 电话有助于查明发生了什么事)。

void TestAttempt()
{

    string source = @"
  Imports System
  Namespace Exceptions
    Public NotInheritable Class ExampleException
      Inherits Validation

      Public Sub New()
        X(""Ignore me 1"")
        Console.WriteLine(""Report me"")
        Console.WriteLine(X(""Ignore me 2""))

        MyBase.New(X(""Ignore me 3, "" & _
                   ""Because I'm already translated.""))
      End Sub
    End Class
  End Namespace";

    var tree = VisualBasicSyntaxTree.ParseText(source);

    var syntaxRoot = tree.GetRoot();
    int i = 0, notWrapped = 0;

    foreach (var s in syntaxRoot.DescendantNodes().OfType<LiteralExpressionSyntax>())
    {
        // things to skip:
        if (s == null) { continue; }
        if (s.Kind() != SyntaxKind.StringLiteralExpression) { continue; }

        if (!IsWrappedInCallToX(s))
        {
            Console.WriteLine($"  Reported: {s.ToString()}");
            notWrapped++;
        }
        i++;
    }

    Console.WriteLine();
    Console.WriteLine($"Total: {i}, Not Wrapped In X: {notWrapped}");
}

bool IsWrappedInCallToX(SyntaxNode node)
{
    var invocation = node as InvocationExpressionSyntax;
    if (invocation != null)
    {
        var exp = invocation.Expression as IdentifierNameSyntax;
        if (exp != null && exp.ToString() == "X")
        {
            return true;
        }
    }
    if (node.Parent != null)
    {
        return IsWrappedInCallToX(node.Parent);
    }
    return false;
}

这导致:

  Reported: "Report me"

Total: 5, Not Wrapped In X: 1

IsWrappedInCallToX 函数只是向上递归树,为 X 函数寻找 InvocationExpressionSyntax。我知道你说 "While I could write something to crawl backwards up the tree, that didn't seem like it was the right way (and probably wouldn't perform very well)",但对我来说这似乎是正确的方法 - 如果你的代码库的性能很糟糕,也许不是!

同样,我对 Roslyn 一无所知(这听起来很有趣),所以这很可能是一个糟糕的解决方案! :-)