Roslyn - 查找所有符号

Question

我正在使用以下代码来检索代码块中使用的所有符号。这包括声明和对符号的引用。 不幸的是，GetSymbolInfo 调用非常慢，因此该方法花费的总时间可能很长。有没有办法加快速度？

    public static IEnumerable<ISymbol> GetAllSymbols(CSharpCompilation compilation, SyntaxNode root)
    {
        var noDuplicates = new HashSet<ISymbol>();

        var model = compilation.GetSemanticModel(root.SyntaxTree);

        foreach (var node in root.DescendantNodesAndSelf())
        {
            ISymbol symbol = model.GetDeclaredSymbol(node) ??
                model.GetSymbolInfo(node).Symbol;

            if (symbol != null)
            {
                if (noDuplicates.Add(symbol))
                    yield return symbol; 
            }
        }
    }

Answer 1

我发现您的代码存在一些问题。

第一个有点详细，但您说您正在寻找您“使用”的符号。您是否认为声明符号是一种使用符号？如果没有，你可以摆脱model.GetDeclaredSymbol(node)。

第二个问题更重要：你经常得到相同的符号。

以下面的语句为例：

SomeMethod();

这是一个ExpressionStatement 节点，其中有一个InvocationExpression，其中有一个IdentifierName。您正在这三个节点上调用model.GetSymbolInfo(node)。你应该想办法避免这种情况。

如果您只在 SimpleNameSyntax（或其后代，IdentifierNameSyntax 和 GenericNameSyntax）类型的节点上调用 model.GetSymbolInfo(node)，您会得到很多符号。

类似：

public static IEnumerable<ISymbol> GetAllSymbols(CSharpCompilation compilation, SyntaxNode root)
{
    var noDuplicates = new HashSet<ISymbol>();

    var model = compilation.GetSemanticModel(root.SyntaxTree);

    foreach (var node in root.DescendantNodesAndSelf())
    {
        switch (node.Kind())
        {
            case SyntaxKind.IdentifierName:
            case SyntaxKind.GenericName:
                ISymbol symbol = model.GetSymbolInfo(node).Symbol;

                if (symbol != null && noDuplicates.Add(symbol))
                {
                    yield return symbol;
                }
                break;
        }
    }
}

它不会得到 all 符号。例如，将找不到运算符的符号。

这让我想到了第三点：你真的应该考虑你所追求的符号。你真的需要所有个符号吗？

即使答案是“是”，您也可以通过反转上述逻辑来避免许多冗余查找的情况。

例如：

public static IEnumerable<ISymbol> GetAllSymbols(CSharpCompilation compilation, SyntaxNode root)
{
    var noDuplicates = new HashSet<ISymbol>();

    var model = compilation.GetSemanticModel(root.SyntaxTree);

    foreach (var node in root.DescendantNodesAndSelf())
    {
        switch (node.Kind())
        {
            case SyntaxKind.ExpressionStatement:
            case SyntaxKind.InvocationExpression:
                break;
            default:
                ISymbol symbol = model.GetSymbolInfo(node).Symbol;

                if (symbol != null)
                {
                    if (noDuplicates.Add(symbol))
                        yield return symbol;
                }
                break;
        }
    }
}

在这种情况下，我只过滤掉了上面示例中的ExpressionStatement 和InvocationExpression。您可以安全地过滤掉更多内容，但我将其留给您作为练习。