Powershell tr​​y 块不会释放 catch 块的文件句柄

Question

我正在编写一个脚本，它尝试使用两种文件下载方法下载文件。 我不打算详细介绍，但生成的函数看起来像这样：

function Download-FileRobust($url, $targetFile) {
    try {
        Download-File $url $targetFile
    }
    catch {
        Download-FileWget $url $targetFile
    }
}

当Download-File 函数失败时，PowerShell 不会松开在位置$targetFile 处创建的文件的句柄，并且Download-FileWget 无法写入该位置。

我习惯了 Python，所以我花了很长时间才弄清楚问题所在。

另外两个函数的源码如下：

function Download-FileWget($url, $targetFile){
    $wgetDir = (Get-ChildItem -Path "$env:userprofile\Downloads\wget*win32").FullName
    if($env:Path -notlike "*$wgetDir*"){
        $env:Path = "$wgetDir;$env:Path"
    }
    Invoke-Expression "wget '$url' -O '$targetFile'"
}

function Download-File($url, $targetFile){
   $uri = New-Object "System.Uri" "$url"
   $request = [System.Net.HttpWebRequest]::Create($uri)
   $request.set_Timeout(15000) #15 second timeout
   $response = $request.GetResponse()
   $totalLength = [System.Math]::Floor($response.get_ContentLength()/1024)
   $responseStream = $response.GetResponseStream()
   $targetStream = New-Object -TypeName System.IO.FileStream -ArgumentList $targetFile, Create
   $buffer = new-object byte[] 10KB
   $count = $responseStream.Read($buffer,0,$buffer.length)
   $downloadedBytes = $count

   while ($count -gt 0){
       $targetStream.Write($buffer, 0, $count)
       $count = $responseStream.Read($buffer,0,$buffer.length)
       $downloadedBytes = $downloadedBytes + $count
       Write-Progress -activity "Downloading file '$($url.split('/') | Select -Last 1)'" -status "Downloaded ($([System.Math]::Floor($downloadedBytes/1024))K of $($totalLength)K): " -PercentComplete ((([System.Math]::Floor($downloadedBytes/1024)) / $totalLength)  * 100)

   }
   Write-Progress -activity "Finished downloading file '$($url.split('/') | Select -Last 1)'"
   $targetStream.Flush()
   $targetStream.Close()
   $targetStream.Dispose()
   $responseStream.Dispose()
}

Answer 1

在Download-File 函数中，您实际上从不做任何事情来确保$targetStream 被正确冲洗和处理。您需要将流包装在 try/catch/finally 块中：

function Download-File {
    param([string]$url, [string]$targetFile)

    $uri = New-Object "System.Uri" "$url"
    $request = [System.Net.HttpWebRequest]::Create($uri)
    $request.set_Timeout(15000) #15 second timeout
    $response = $request.GetResponse()
    $totalLength = [System.Math]::Floor($response.get_ContentLength()/1024)
    $responseStream = $response.GetResponseStream()
    try {
        $targetStream = New-Object -TypeName System.IO.FileStream -ArgumentList $targetFile, Create
        try {
            $buffer = new-object byte[] 10KB
            $count = $responseStream.Read($buffer,0,$buffer.length)
            $downloadedBytes = $count

            while ($count -gt 0){
                $targetStream.Write($buffer, 0, $count)
                $count = $responseStream.Read($buffer,0,$buffer.length)
                $downloadedBytes = $downloadedBytes + $count
                Write-Progress -activity "Downloading file '$($url.split('/') | Select -Last 1)'" -status "Downloaded ($([System.Math]::Floor($downloadedBytes/1024))K of $($totalLength)K): " -PercentComplete ((([System.Math]::Floor($downloadedBytes/1024)) / $totalLength)  * 100)

            }
            Write-Progress -activity "Finished downloading file '$($url.split('/') | Select -Last 1)'"
        }
        catch {
            throw
        }
        finally {
            if($targetStream){
                $targetStream.Flush()
                $targetStream.Close()
                $targetStream.Dispose()
            }
        }
    }
    catch {
        throw
    }
    finally {
        $responseStream.Dispose()
    }
}