aws-ssm-bridge 0.2.0

Rust library implementing AWS Systems Manager Session Manager protocol
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62

"""
Example: Managing multiple concurrent sessions

This example demonstrates how to manage multiple SSM sessions concurrently.
"""

import asyncio
from aws_ssm_bridge import SessionManager


async def run_session(manager, instance_id, region, session_num):
    """Run a single session"""
    print(f"[Session {session_num}] Starting session to {instance_id}")

    session = await manager.start_session(
        target=instance_id,
        region=region,
        session_type="standard_stream",
        reason=f"Test session {session_num}",
    )

    print(f"[Session {session_num}] Started: {session.id}")

    # Send a command
    await session.send(f"echo 'Hello from session {session_num}!'\n".encode())

    # Wait a moment
    await asyncio.sleep(1)

    # Terminate
    await session.terminate()
    await session.wait_terminated()

    print(f"[Session {session_num}] Terminated")


async def main():
    # Configuration
    instances = [
        ("i-1234567890abcdef0", "us-east-1"),
        ("i-0987654321fedcba0", "us-east-1"),
        ("i-abcdef1234567890", "us-west-2"),
    ]

    print(f"Starting {len(instances)} concurrent sessions...\n")

    # Create session manager
    manager = await SessionManager.new()

    # Run all sessions concurrently
    tasks = [
        run_session(manager, instance_id, region, i + 1)
        for i, (instance_id, region) in enumerate(instances)
    ]

    await asyncio.gather(*tasks)

    print("\n✓ All sessions completed")


if __name__ == "__main__":
    asyncio.run(main())