182 lines
8.1 KiB
PowerShell
182 lines
8.1 KiB
PowerShell
# 性能基线（G1 闸门）：向指定终端 exe 注入同一份大流量混合负载，
|
|
# 在终端内部自计时（cmd /c type 大文件），量吞吐与内存，追加到 CSV。
|
|
#
|
|
# 用法:
|
|
# .\scripts\perf_baseline.ps1 -Exe <pebrel.exe> -Tag winit-old -ExeArgs '--working-directory D:\temp_build\nebula'
|
|
# .\scripts\perf_baseline.ps1 -Exe <nebula-gpui.exe> -Tag gpui-lab
|
|
#
|
|
# 方法论:
|
|
# - 负载混合 ANSI 256 色/真彩、CJK 宽字符、boxdraw 框线、Powerline 分隔符，
|
|
# 贴近 fastfetch/TUI/powerline 的真实着色与整形压力。
|
|
# - 计时在终端会话内完成（Stopwatch 包住 type），只量"终端消费 PTY 输出"
|
|
# 的速度，不含进程启动与提示符渲染。
|
|
# - 命令经 ConPTY 控制台输入缓冲注入（perf_inject.ps1），不碰窗口焦点；
|
|
# 窗口置顶但不激活（NOACTIVATE），保证两壳都真实渲染——旧渲染器被完全
|
|
# 遮挡时会跳帧，会虚高吞吐。
|
|
# - 第一轮为预热（文件缓存/JIT），丢弃；报告中位数。
|
|
# - 两壳窗口尺寸可能不同（网格越大每帧工作越多），rect 一并记录供对账。
|
|
param(
|
|
[Parameter(Mandatory = $true)][string]$Exe,
|
|
[Parameter(Mandatory = $true)][string]$Tag,
|
|
# 旧壳必须传 '--working-directory <dir>' 之类的显式意图参数，
|
|
# 否则普通启动会转交常驻实例后退出（mux hand-over），拿不到独立进程。
|
|
[string]$ExeArgs = '',
|
|
[int]$Runs = 3,
|
|
[int]$PayloadMB = 50,
|
|
[int]$TimeoutSec = 240
|
|
)
|
|
|
|
$ErrorActionPreference = 'Stop'
|
|
$root = Split-Path -Parent $PSScriptRoot
|
|
$payload = Join-Path $root '.tmp_perf_payload.txt'
|
|
$cmdFile = Join-Path $root '.tmp_perf_cmd.ps1'
|
|
$resultFile = Join-Path $root '.tmp_perf_result.txt'
|
|
$csv = Join-Path $root '.tmp_perf_baseline.csv'
|
|
$injector = Join-Path $PSScriptRoot 'perf_inject.ps1'
|
|
|
|
# ---- 1. 负载生成（按目标大小幂等）----------------------------------------
|
|
$targetBytes = $PayloadMB * 1MB
|
|
if (-not (Test-Path $payload) -or (Get-Item $payload).Length -lt $targetBytes) {
|
|
Write-Host "generating payload ~${PayloadMB}MB ..."
|
|
$e = [char]27
|
|
$block = [Text.StringBuilder]::new()
|
|
for ($i = 0; $i -lt 240; $i++) {
|
|
$line = switch ($i % 6) {
|
|
0 { "$e[38;5;$(($i * 7) % 256)m colored line $i lorem ipsum dolor sit amet consectetur adipiscing elit$e[0m" }
|
|
1 { "plain ascii line $i the quick brown fox jumps over the lazy dog 0123456789" }
|
|
2 { "中文宽字符混排 第${i}行 终端网格渲染必须逐格对齐 妙笔生花 龍飛鳳舞 テスト" }
|
|
3 { "┌─────┬─────┐ │ box $i │ ▀▄█░▒▓ ├─────┼─────┤ └─────┴─────┘" }
|
|
4 { "$e[48;5;$(($i * 3) % 256)m$e[38;5;15m$([char]0xE0B0) powerline seg $i $([char]0xE0B0)$e[0m tail" }
|
|
5 { "$e[1;31mbold red$e[0m $e[4;32munderline green$e[0m $e[38;2;$(($i*5)%256);128;200mtruecolor$e[0m line $i" }
|
|
}
|
|
[void]$block.AppendLine($line)
|
|
}
|
|
$blockText = $block.ToString()
|
|
$writer = [IO.StreamWriter]::new($payload, $false, [Text.UTF8Encoding]::new($false))
|
|
try {
|
|
while ($writer.BaseStream.Length -lt $targetBytes) { $writer.Write($blockText) }
|
|
} finally { $writer.Dispose() }
|
|
}
|
|
$payloadLen = (Get-Item $payload).Length
|
|
|
|
# ---- 2. 终端内自计时的 runner ---------------------------------------------
|
|
@"
|
|
`$sw = [Diagnostics.Stopwatch]::StartNew()
|
|
cmd /c type "$payload"
|
|
`$sw.Stop()
|
|
"`$(`$sw.Elapsed.TotalMilliseconds)" | Out-File -FilePath "$resultFile" -Encoding ascii
|
|
"@ | Set-Content -Path $cmdFile -Encoding ascii
|
|
|
|
# ---- 3. Win32：按 PID 找顶层窗口、置顶但不激活 -----------------------------
|
|
Add-Type @'
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Runtime.InteropServices;
|
|
public class PerfWin {
|
|
[DllImport("user32.dll")] static extern bool EnumWindows(EnumProc cb, IntPtr lp);
|
|
[DllImport("user32.dll")] static extern uint GetWindowThreadProcessId(IntPtr h, out uint pid);
|
|
[DllImport("user32.dll")] static extern bool IsWindowVisible(IntPtr h);
|
|
[DllImport("user32.dll")] public static extern bool GetWindowRect(IntPtr h, out RECT r);
|
|
[DllImport("user32.dll")] public static extern bool SetWindowPos(IntPtr h, IntPtr after, int x, int y, int w, int hh, uint flags);
|
|
public struct RECT { public int L, T, R, B; }
|
|
delegate bool EnumProc(IntPtr h, IntPtr lp);
|
|
public static List<IntPtr> WindowsOf(uint pid) {
|
|
var list = new List<IntPtr>();
|
|
EnumWindows((h, lp) => {
|
|
uint p; GetWindowThreadProcessId(h, out p);
|
|
if (p == pid && IsWindowVisible(h)) list.Add(h);
|
|
return true;
|
|
}, IntPtr.Zero);
|
|
return list;
|
|
}
|
|
}
|
|
'@
|
|
|
|
function Get-MainWindow([int]$ProcessId) {
|
|
for ($i = 0; $i -lt 40; $i++) {
|
|
$wins = [PerfWin]::WindowsOf($ProcessId)
|
|
if ($wins.Count -gt 0) {
|
|
# 取最宽者（宿主可能带辅助小窗）。
|
|
$best = $null; $bestW = -1
|
|
foreach ($h in $wins) {
|
|
$r = New-Object PerfWin+RECT
|
|
[void][PerfWin]::GetWindowRect($h, [ref]$r)
|
|
if (($r.R - $r.L) -gt $bestW) { $bestW = $r.R - $r.L; $best = $h }
|
|
}
|
|
return $best
|
|
}
|
|
Start-Sleep -Milliseconds 500
|
|
}
|
|
throw "no visible window for pid $ProcessId"
|
|
}
|
|
|
|
# ---- 4. 驱动 ----------------------------------------------------------------
|
|
Write-Host "launching $Exe $ExeArgs"
|
|
$proc = if ($ExeArgs) {
|
|
Start-Process -FilePath $Exe -ArgumentList $ExeArgs -PassThru
|
|
} else {
|
|
Start-Process -FilePath $Exe -PassThru
|
|
}
|
|
Start-Sleep -Seconds 7 # 等 shell 与提示符就绪
|
|
if ($proc.HasExited) { throw "terminal exited immediately (mux hand-over? pass -ExeArgs)" }
|
|
$hwnd = Get-MainWindow -ProcessId $proc.Id
|
|
$rect = New-Object PerfWin+RECT
|
|
[void][PerfWin]::GetWindowRect($hwnd, [ref]$rect)
|
|
$rectStr = "$($rect.R - $rect.L)x$($rect.B - $rect.T)"
|
|
# HWND_TOPMOST=-1; SWP_NOSIZE|NOMOVE|NOACTIVATE = 0x13：可见但不抢焦点。
|
|
[void][PerfWin]::SetWindowPos($hwnd, [IntPtr](-1), 0, 0, 0, 0, 0x13)
|
|
Write-Host "window $rectStr pid $($proc.Id)"
|
|
|
|
$command = "powershell -NoProfile -ExecutionPolicy Bypass -File $cmdFile"
|
|
|
|
function Invoke-Injected([int]$HostPid, [string]$Text) {
|
|
for ($attempt = 0; $attempt -lt 5; $attempt++) {
|
|
$out = & powershell -NoProfile -ExecutionPolicy Bypass -File $injector `
|
|
-HostPid $HostPid -Text $Text -Enter 2>&1 | Out-String
|
|
if ($out -match 'ok=True') { return $true }
|
|
Start-Sleep -Seconds 2
|
|
}
|
|
Write-Warning "inject failed: $out"
|
|
return $false
|
|
}
|
|
|
|
$times = @()
|
|
for ($run = 0; $run -le $Runs; $run++) {
|
|
Remove-Item $resultFile -ErrorAction SilentlyContinue
|
|
if (-not (Invoke-Injected -HostPid $proc.Id -Text $command)) { continue }
|
|
|
|
$deadline = (Get-Date).AddSeconds($TimeoutSec)
|
|
while (-not (Test-Path $resultFile) -and (Get-Date) -lt $deadline) {
|
|
Start-Sleep -Milliseconds 250
|
|
}
|
|
if (-not (Test-Path $resultFile)) {
|
|
Write-Warning "run ${run}: timeout waiting for result"
|
|
continue
|
|
}
|
|
Start-Sleep -Milliseconds 300
|
|
$ms = [double](Get-Content $resultFile -Raw).Trim()
|
|
$mbps = [Math]::Round($payloadLen / 1MB / ($ms / 1000.0), 1)
|
|
if ($run -eq 0) {
|
|
Write-Host ("warmup: {0,8:N0} ms {1,6} MB/s (discarded)" -f $ms, $mbps)
|
|
} else {
|
|
Write-Host ("run ${run}: {0,8:N0} ms {1,6} MB/s" -f $ms, $mbps)
|
|
$times += $ms
|
|
}
|
|
Start-Sleep -Seconds 1
|
|
}
|
|
|
|
$ws = [Math]::Round((Get-Process -Id $proc.Id).WorkingSet64 / 1MB, 1)
|
|
Stop-Process -Id $proc.Id -Force -ErrorAction SilentlyContinue
|
|
|
|
if ($times.Count -eq 0) { throw "no successful runs" }
|
|
$sorted = $times | Sort-Object
|
|
$median = $sorted[[int](($sorted.Count - 1) / 2)]
|
|
$medianMbps = [Math]::Round($payloadLen / 1MB / ($median / 1000.0), 1)
|
|
Write-Host ("== $Tag median {0:N0} ms {1} MB/s ws {2} MB window $rectStr" -f $median, $medianMbps, $ws)
|
|
|
|
if (-not (Test-Path $csv)) {
|
|
'timestamp,tag,window,payload_mb,runs,median_ms,median_mbps,workingset_mb' | Set-Content $csv -Encoding ascii
|
|
}
|
|
"$(Get-Date -Format s),$Tag,$rectStr,$([Math]::Round($payloadLen/1MB)),$($times.Count),$([Math]::Round($median)),$medianMbps,$ws" |
|
|
Add-Content $csv -Encoding ascii
|