如何从过程中获取选择语句

Question

这是这里的后续问题。我正在尝试使用 DACExtension从Procedure中检索SelectStatement对象，以便我可以使用 T4 模板从我的 SSDT 项目生成包装器函数。问题是我的访客对象的节点属性中没有任何节点。我错过了什么？

这是我的访客：

public class SelectVisitor : TSqlFragmentVisitor
{
    public SelectVisitor() { this.Nodes = new List<SelectStatement>(); }
    public List<SelectStatement> Nodes { get; private set; }
    public override void Visit(SelectStatement node)
    {
        base.Visit(node);
        this.Nodes.Add(node);
    }
}

这是我尝试使用它的方式：

// Create the model
var procFiles = Directory.GetFiles(sqlPath, "*.sql", SearchOption.AllDirectories);
var model = new TSqlTypedModel(SqlServerVersion.Sql100, new TSqlModelOptions());
foreach(var procFile in procFiles)
{
    model.AddObjects(File.ReadAllText(procFile));
}
// Loop through the procs
var procs = model.GetObjects<TSqlProcedure>(DacQueryScopes.UserDefined);
foreach(var proc in procs){ 
    var selectVisitor = new SelectVisitor();
    var ast = proc.GetAst();
    ast.Accept(selectVisitor);
    foreach(var node in selectVisitor.Nodes){
        // Nodes has Count=0 :(
    }
}

Answer 1

使用 TSqlModelUtils.TryGetFragmentForAnalysis 应该确保你在模型中得到原始的 AST——希望它能得到你需要的东西。你可能想要调试并查看你的 AST 以及实际创建的内容 - 这就是我们在内部做事的方式，通常你可能会对实际生成的内容感到惊讶。最后请注意，DacpacExplorer 工具可能会使其更易于可视化 - 它现在应该支持在对象（如过程）后面显示 AST。