我可以判断 ado.net DbCommand 是否是查询（在执行之前）

Question

我正在尝试编写一个 Powershell 脚本来针对数据库运行通用 SQL 命令。这个想法是 Run-SQL“select ...”将针对当前打开的数据库运行 SQL 文本。如果SQL语句是一个查询，它应该返回一个DataTable。如果它是非查询（DDL 或 DML），则不应返回任何内容 ($null)。

为了做到这一点，我需要知道针对该命令执行哪个方法（ExecuteReader 或 ExecuteNonQuery）。有没有办法确定这一点？ （如果有帮助的话，我很乐意准备命令）。

作为替代方案，我可以添加一个由用户提供的 -query 参数，它区分这两种情况，但作为潜在用户，我会发现这很烦人（因为在我看来，我已经说过它是否是我使用的 SQL 查询，为什么再说一遍？）

我的主要用途是 Oracle 数据库，因此 Oracle 特定的答案对我来说没问题，尽管我更喜欢通用的答案。

Answer 1

我认为无论是否是查询，您都可以使用 ExecuteReader 。它可能有点矫枉过正，但在一些快速实验中进行 UPDATE （$reader 返回任何内容）和 COUNT （$reader[0] 输出标量结果） - 它似乎有效。

$connstr = "server=.\SQLEXPRESS;Database=AdventureWorks;" + 
           "Integrated Security=true;Persist Security Info=False"

$conn = new-object System.Data.SqlClient.SqlConnection $connstr    
#$query = "UPDATE Production.Product SET Name = 'ACME' WHERE Name = 'Blade'"
$query = "SELECT Count(*) FROM Production.Product"

$cmd = new-object System.Data.SqlClient.SqlCommand $query,$conn
$conn.Open()
try
{
    $reader = $cmd.ExecuteReader()
    while ($reader.Read())
    {
        $reader[0]
    }
}
finally
{
    $conn.Dispose()
}

Answer 2

作为 Keith 所说的替代方案，您可以尝试

$sql = 'Select count(1) From SomeTable;'
$sql = $sql.TrimStart(' ')
if ($sql -match "^select") { Write-Host 'ExecuteReader' }
else { Write-Host 'ExecuteNonQuery'}

修剪，以防 SQL 命令有前导空格，因为 $sql = ' select ' 不会匹配“^ select”