Microsoft Tsql 脚本中的哪个 API 允许我从 SQL 脚本中检索/存储所有表引用?

Question

这是我引用的MS文档：https://learn.microsoft.com/en-us/dotnet/api/microsoft.sqlserver.transactsql.scriptdom?view=sql-dacfx-140.3881.1

我已经多次浏览了这个，但似乎找不到我能够在 sql 脚本中检索每个表引用的实际名称的位置。 NamedTableReference是我找到的最接近的东西，但这只允许您查看表的别名，而我想要实际名称。

例：

SELECT * FROM Table1 INNER JOIN Table2 ON Table1.ID = Table2.ID

我想要以下输出：表1 表2

最后一部分不一定重要，但只是为了上下文。我正在尝试扩展开源 TSQLLint 项目的功能，这将使我能够在没有 NOLOCK 提示的情况下检查开发人员正在使用的我们公司中非常活跃的表名。

Answer 1

您可以使用访问者模式识别脚本中的所有NamedTableReference片段。片段的SchemaObject.BaseIdentifier.Value是引用的表名。

下面是一个派生自TSqlConcreteFragmentVisitor的 PowerShell 示例。这可以针对您选择的 .NET 语言进行调整。

$script = @"
SELECT * FROM Table1 INNER JOIN Table2 as t2 ON Table1.ID = Table2.ID
"@
class MyVisitor: Microsoft.SqlServer.TransactSql.ScriptDom.TSqlConcreteFragmentVisitor {
[void]Visit ([Microsoft.SqlServer.TransactSql.ScriptDom.NamedTableReference] $fragment) {
Write-Host "Found $($fragment.GetType().Name) at line $($fragment.StartLine), column $($fragment.StartColumn). Table name: $($fragment.SchemaObject.BaseIdentifier.Value)"
}
}
# ############
# ### MAIN ###
# ############
try {
$parser = New-Object Microsoft.SqlServer.TransactSql.ScriptDom.TSql150Parser($true)
$parseErrors = New-Object System.Collections.Generic.List[Microsoft.SqlServer.TransactSql.ScriptDom.ParseError]
$stringReader = New-Object System.IO.StringReader($script)
$frament = $parser.Parse($stringReader, [ref]$parseErrors)
if($parseErrors.Count -gt 0) {
throw "$($parseErrors.Count) parsing errors: $(($parseErrors | ConvertTo-Json))"
}
$visitor = [MyVisitor]::new()
$frament.Accept($visitor)
}
catch {
throw
}

输出：

Found NamedTableReference at line 1, column 15. Table name: Table1.
Found NamedTableReference at line 1, column 33. Table name: Table2.

在 PowerShell 中，必须首先使用 Add-Type 将定义基类型的外部程序集加载到应用域中，以便编译具有派生类的脚本。这可以通过包装脚本来完成，该脚本对实际脚本进行点源，例如：

# example wrapper script to load script DOM assembly and execute the visitor script
Add-Type -Path "C:DacFxAssembliesMicrosoft.SqlServer.TransactSql.ScriptDom.dll"
. "C:ScriptsFind-NamedTableReferences.ps1"