blob: 469fdb0cd2c2cf345854f0cd604a1da851da07c4 (
plain)
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
63
64
65
|
#
# SPDX-FileCopyrightText: 2016 - 2026 Dimitri Staessens
# SPDX-License-Identifier: BSD-3-Clause
#
"""
A simple echo client/server example using Ouroboros flow event sets.
"""
# Standalone demo; duplicates oecho.py.
# pylint: disable=duplicate-code
from __future__ import annotations
import argparse
from ouroboros.dev import flow_accept, flow_alloc
from ouroboros.event import FEventQueue, FEventType, FlowSet
def client() -> None:
"""
Send a message and print the echo reply.
"""
with flow_alloc("oecho") as f:
f.writeline("Hello, PyOuroboros!")
print(f.readline())
def server() -> None:
"""
Accept flows and echo back received messages using flow events.
"""
print("Starting the server.")
while True:
with flow_accept() as f:
print("New flow.")
with FEventQueue() as fq, FlowSet([f]) as fs:
fs.wait(fq)
f2, t = fq.next()
if t != FEventType.FLOW_PKT:
continue
line = f2.readline()
print(f"Message from client is {line}")
f2.writeline(line)
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='A simple echo client/server')
parser.add_argument('-l', '--listen', help='run as a server',
action='store_true')
args = parser.parse_args()
if args.listen:
server()
else:
client()
|
