PowerShell 多线程的使用

今天一个朋友问我在Powershell里面如何能够并发的ping上万台机器?默认的test-connection 尽管有-computer这个参数,他的方式是按顺序的挨个ping,全部跑下来可能有好几个小时。

比如我需要花18秒的时间才能ping完40台服务器,如果成千上万的话就很费时间了。

measure-commnd -expression {
Get-ADComputer -Filter {operatingsystem -like "*2012*"} | ForEach-Object {
Test-Connection -ComputerName $_.name -Count 1
}
}

这之前,豆子对多线程的使用仅仅限于了解invoke-command可以同时对30个对象操作,经过一番学习,终于发现还有其他 的高级方式。

PowerShell里面,对于多线程的使用大概是两大方式。

第一个是创建多个后台的job。这种方式通过start-job或者 -asjob创建后台job,然后通过get-job获取当前的任务,通过receive-job来获取完成任务的结果,最后还得remove-job来释放内存。缺点显而易见是比较麻烦。而且一旦多跑几个job 就灰常灰常地卡!

第二个方式是创建多个runspace,这个工作原理和invoke-command一样,每一个远程的session绑定一个runspace。我们可以创建一个runspace pool,指定在这个资源池里面最多可以同时执行多少个runspace。

豆子主要参考了这篇博客的方法,写了一个简单的脚本。

http://thesurlyadmin.com/2013/02/11/multithreading-powershell-scripts/

$Throttle = 20 #threads

#脚本块,对指定的计算机发送一个ICMP包测试,结果保存在一个对象里面
 
$ScriptBlock = {
   Param (
      [string]$Computer
   )
   $a=test-connection -ComputerName $Computer -Count 1 
   
   $RunResult = New-Object PSObject -Property @{
      IPv4Adress=$a.ipv4address.IPAddressToString
      ComputerName=$Computer
      
   }
   Return $RunResult
}

#创建一个资源池,指定多少个runspace可以同时执行

$RunspacePool = [RunspaceFactory]::CreateRunspacePool(1, $Throttle)
$RunspacePool.Open()
$Jobs = @()
 

#获取Windows 2012服务器的信息,对每一个服务器单独创建一个job,该job执行ICMP的测试,并把结果保存在一个PS对象中
 
(get-adcomputer -filter {operatingsystem -like "*2012*"}).name | % {
   
   #Start-Sleep -Seconds 1
   $Job = [powershell]::Create().AddScript($ScriptBlock).AddArgument($_)
   $Job.RunspacePool = $RunspacePool
   $Jobs += New-Object PSObject -Property @{
      Server = $_
      Pipe = $Job
      Result = $Job.BeginInvoke()
   }
}
 

 #循环输出等待的信息.... 直到所有的job都完成 
 
Write-Host "Waiting.." -NoNewline
Do {
   Write-Host "." -NoNewline
   Start-Sleep -Seconds 1
} While ( $Jobs.Result.IsCompleted -contains $false)
Write-Host "All jobs completed!"

#输出结果 
$Results = @()
ForEach ($Job in $Jobs)
{   $Results += $Job.Pipe.EndInvoke($Job.Result)
}
 
$Results

大概5秒之后 结果就出来了。 如果有兴趣的话可以使用measure-command命令来测试不同线程数的效果,根据我的测试,30个进程同时执行只需4秒出结果,而2个同时执行大概需要9秒才能出结果。

知道原理之后就可以进一步优化和抽象化脚本。这一点已经有人做好了。https://github.com/RamblingCookieMonster/Invoke-Parallel/blob/master/Invoke-Parallel/Invoke-Parallel.ps1

下载,Unlock和dot source之后就能直接调用了。这里提供了一些例子作为参考https://github.com/RamblingCookieMonster/Invoke-Parallel

依葫芦画瓢,我想通过他来调用test-connection也是成功的

get-adcomputer -Filter {operatingsystem -like "*2012*"} | select -ExpandProperty name | Invoke-Parallel -ScriptBlock {Test-Connection -computername $_ -count 1}

最后,网上也有现成的脚本用来并发的测试ping,原理也是调用上面的invoke-parallel函数,不过他还增加了其他的函数用来测试rdp,winrm,rpc等远程访问的端口是否打开,进一步扩充了功能。可以直接在这里下载

http://ramblingcookiemonster.github.io/Invoke-Ping/

invoke-ping -ComputerName (Get-ADComputer -Filter {operatingsystem -like "*2012*"}).name -Detail RDP,rpc | ft -Wrap

时间: 2024-08-11 01:25:05

PowerShell 多线程的使用的相关文章

31. PowerShell -- 多线程执行前后台作业

使用后台作业执行多个任务从先前的技巧中看不是非常高效,它在处理每个后台作业返回结果时将会浪费很多性能.一个更有效的方法是使用进程内的任务.他能分别单独的执行任务与Powershell类似,所以它不是按顺序返回值的. 下面例子使用Powershell线程运行了两个后台任务和一个前台任务,创建几个运行时间长点的任务,并且每个任务命令中添加使用Start-Sleep. 代码如下: $start = Get-Date $task1 = { Start-Sleep -Seconds 4; Get-Serv

Python 多进程使用之监控

上一篇文章中写过使用powershell多线程监控网站状态,但powershell只能运行在Windows环境,于是将powershell代码改成了python代码,下面简单写下实现方式,具体代码不会全部贴出来. 需要使用的py库有下面几个 import urllib2 import socket from multiprocessing import Pool from threading import Thread urllib2用于请求网页 socket用于限制网络请求超时 pool用于设

发布支持多线程的PowerShell模块 —— MultiThreadTaskRunner

应用场景 多线程在需要批量处理一些任务的时候相当有用,也更加有利于充分利用现有计算机的能力.所有主流的开发语言都支持多线程. 默认情况下,PowerShell作为一个脚本语言,是不支持多线程操作的,虽然有所谓的后台任务,但实际上控制起来相当麻烦. 解决方案 我很喜欢PowerShell,所以为此专门写了一个自定义的模块,来实现多线程的功能.该模块使用C#编写,源代码可以在下面的地址访问到 https://github.com/chenxizhang/MultiThreadTaskRunner 与

Powershell 之多线程监控网页状态

#设置线程数量 $throttleLimit = 10 $SessionState = [system.management.automation.runspaces.initialsessionstate]::CreateDefault() $Pool = [runspacefactory]::CreateRunspacePool(1, $throttleLimit, $SessionState, $Host) $Pool.Open() #url configure $urlFile="D:\

Powershell使用 OpenFileDialog类 假死问题解决。

我创建了一个打开文件的函数.代码内容: function Openfile {   Add-Type -AssemblyName System.Windows.Forms   $openfile = New-Object System.Windows.Forms.OpenFileDialog   $openfile.Filter = '图片|*.jpg|All Files|*.*'   $openfile.InitialDirectory = "C:\ManageOperate\Staff Ph

多线程异步执行脚本

获取远程计算机信息: $d=get-date $servers="10.4.34.41","10.4.34.40","10.4.34.36","10.4.34.37","10.4.34.39" $serverpass="Dell1950" $UserName="Administrator" #定义线程数量$throttleLimit = 5$SessionState

PowerShell runspace 的创建,使用和查错

今天拜读了The Scripting Guy关于runspace的几篇大作,温故而知新,一些忽略的地方更为清楚. https://blogs.technet.microsoft.com/heyscriptingguy/2015/11/26/beginning-use-of-powershell-runspaces-part-1/ runspace这两年在PowerShell使用的频率越来越高,由于他的高效率,基本上很多时候已经取代了传统的Job后台操作.不管是多线程实现,或者是后台操作,亦或是入

从Powershell 入侵脚本学到的如何执行后台runspace~

豆子今天无聊在github上看看有什么有意思的PowerShell脚本,无意中发现了PowerSploit这个项目,仔细看了看,这个模块是针对入侵测试写的,里面有大量相关的黑客脚本,随便找了一个试试看. 比如说这个,可以用来记录键盘的输入内容,完整的脚本我就不贴出来了. https://github.com/PowerShellMafia/PowerSploit/blob/dev/Exfiltration/Get-Keystrokes.ps1 具体实现的功能先不去考虑,我很好奇他是怎么在后台执行

37. PowerShell- 性能优化,多线程,提速

PowerShell优化和性能测试 概述 PowerShell已经存在很长一段时间了,许多基本的问题都得到了很好的解决.对于脚本编写者来说,借助于PowerShell的Get-Help帮助文档,互联网,和社区资源,绝大多数普通的任务应当用什么,怎么样来完成,都应该能快速找到答案.一旦脚本写好了,没有bug,准备投入到生产环境中了,可能另外一些问题会浮出水面,正好也是我们平时讨论的比较少的.怎么尽可能的写出更好的脚本?将已经正在使用的脚本改进的更加高效,强化工具.而这些正好需要一些不同的问题和思想