构建同时支持管道与非管道调用的高级多线程PowerShell函数

要让你的Merge-FlattenDirectory函数同时可靠支持管道和非管道输入，不需要刻意区分管道/非管道输入，而是统一处理所有输入项，确保每个路径都被正确收集到列表中。核心问题在于之前的实现错误地覆盖了泛型列表，且没有正确遍历数组输入。

修正后的函数代码 #

function Merge-FlattenDirectory {
    [CmdletBinding()]
    param (
        [Parameter(Mandatory, Position = 0, ValueFromPipeline)]
        [string[]]
        $InputPath
    )

    begin {
        # 初始化字符串类型的泛型列表，确保类型一致性
        $List = [System.Collections.Generic.List[string]]@()
    }

    process {
        # 遍历所有输入路径，不管是单个值还是数组，都逐个添加到列表
        foreach ($path in $InputPath) {
            $List.Add($path)
        }
    }

    end {
        # 并行处理所有收集到的路径
        $List | ForEach-Object -Parallel {
            # 这里写你的扁平化文件夹逻辑
            Write-Host "Processing path: $_"
        } -ThrottleLimit 16
    }
}

关键修正点说明 #

1. 统一输入处理逻辑：

   • 无论通过管道传入单个路径，还是通过参数传入路径数组，process块都会遍历$InputPath的每一项并添加到列表。
   • 管道输入时，PowerShell会为每个管道对象执行一次process块，此时$InputPath是包含单个元素的数组；非管道输入时，$InputPath是完整的路径数组，遍历操作同样能覆盖所有元素。

2. 保持列表类型一致性：

   • 之前的代码中，$List = $InputPath会将泛型列表替换为普通数组，导致后续Add操作失败。修正后始终使用泛型列表的Add方法，避免类型混乱。

3. 类型匹配优化：

   • 将泛型列表类型从PSObject改为string，与$InputPath的string[]类型匹配，提升类型安全性和性能。

验证两种调用方式 #

• 非管道调用：

  $Files = Get-Content $FileList
  Merge-FlattenDirectory -InputPath $Files
  

  此时$InputPath是路径数组，process块会遍历所有路径添加到列表，最终并行处理全部内容。

• 管道调用：

  $Files | Merge-FlattenDirectory
  

  管道会逐个传入每个路径，每次process块遍历单个元素的数组并添加到列表，最终所有路径都会被收集并并行处理。

内容的提问来源于stack exchange，提问作者fmotion1