mirror of
https://github.com/nyakang/nyaterm.git
synced 2026-11-06 16:07:55 +00:00
Kang
129e28f066
- Added functionality to rebind Zmodem recording during session transfers, ensuring accurate data capture. - Implemented new methods for starting and finishing Zmodem runtime, improving session event handling and state management. - Enhanced session event bridge to support protocol consumer binding and asynchronous event pushing, optimizing event processing. - Updated tests to validate new Zmodem runtime behaviors and ensure proper handling of session events during transfers. Validated on Windows with workspace checks, tests, formatting, and all-target clippy.
68 lines
3.9 KiB
Python
68 lines
3.9 KiB
Python
"""Build once and collect the ZMODEM matrix as JSONL (temporary loopback data only)."""
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
import platform
|
|
import subprocess
|
|
import time
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument('--bytes', type=int, default=8 * 1024 * 1024)
|
|
parser.add_argument('--repeats', type=int, default=3)
|
|
parser.add_argument('--rtt-ms', type=int, default=50)
|
|
parser.add_argument('--delay-ms', type=int, default=2)
|
|
parser.add_argument('--rate-kib', type=int, default=4096)
|
|
parser.add_argument('--output', type=Path, default=Path('zmodem-results.jsonl'))
|
|
parser.add_argument('--release', action='store_true')
|
|
args = parser.parse_args()
|
|
root = Path(__file__).resolve().parents[2]
|
|
build = ['cargo', 'build', '-j', '4', '-p', 'nyaterm-transport', '--example', 'zmodem_bench']
|
|
if args.release:
|
|
build.append('--release')
|
|
subprocess.run(build, cwd=root, check=True)
|
|
target = Path(os.environ.get('CARGO_TARGET_DIR', root / 'target'))
|
|
binary = target / ('release' if args.release else 'debug') / 'examples' / ('zmodem_bench.exe' if os.name == 'nt' else 'zmodem_bench')
|
|
base = ['--bytes', str(args.bytes), '--rtt-ms', str(args.rtt_ms)]
|
|
matrix = [
|
|
['--label', 'conservative', '--window', '10', '--receive', 'conservative', '--buffered', '0'],
|
|
['--label', 'bounded-path-buffered', '--window', '10', '--receive', 'conservative', '--buffered', '1'],
|
|
]
|
|
matrix += [['--label', 'nonstop-window', '--window', str(window), '--receive', 'nonstop', '--buffered', '0'] for window in (10, 64, 256, 1024)]
|
|
matrix += [['--label', 'nonstop-buffered', '--window', str(window), '--receive', 'nonstop', '--buffered', '1'] for window in (10, 64, 256, 1024)]
|
|
for direction in ('upload', 'download'):
|
|
for receive in ('conservative', 'nonstop'):
|
|
for window in (10, 64, 256, 1024):
|
|
matrix.append(['--label', 'runtime-loopback', '--layer', 'runtime', '--direction', direction, '--window', str(window), '--receive', receive])
|
|
matrix.append(['--label', 'runtime-slow-peer', '--layer', 'runtime', '--direction', direction, '--delay-ms', str(args.delay_ms), '--rate-kib', str(args.rate_kib)])
|
|
args.output.parent.mkdir(parents=True, exist_ok=True)
|
|
metadata = {'record': 'environment', 'platform': platform.platform(), 'profile': 'release' if args.release else 'debug', 'revision': subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=root, text=True).strip(), 'worktree_modified': bool(subprocess.check_output(['git', 'status', '--porcelain'], cwd=root)), 'unix_time': time.time()}
|
|
with args.output.open('w', encoding='utf-8') as output:
|
|
output.write(json.dumps(metadata) + '\n')
|
|
for repeat in range(args.repeats):
|
|
for config in matrix:
|
|
try:
|
|
completed = subprocess.run([str(binary), *base, *config], cwd=root, capture_output=True, text=True, timeout=180)
|
|
if completed.returncode:
|
|
result = {'success': False, 'config': config, 'error': completed.stderr.strip()}
|
|
else:
|
|
result = json.loads(completed.stdout)
|
|
except subprocess.TimeoutExpired:
|
|
result = {'success': False, 'config': config, 'error': 'benchmark exceeded 180 seconds'}
|
|
except (OSError, json.JSONDecodeError) as error:
|
|
result = {'success': False, 'config': config, 'error': str(error)}
|
|
result['repeat'] = repeat
|
|
output.write(json.dumps(result) + '\n')
|
|
output.flush()
|
|
print(json.dumps(result), flush=True)
|
|
if not result['success']:
|
|
raise SystemExit('benchmark failed; result preserved in JSONL')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|