initial commit
This commit is contained in:
@@ -0,0 +1 @@
|
||||
.idea/
|
||||
@@ -0,0 +1,73 @@
|
||||
import socket
|
||||
import selectors
|
||||
from copy import deepcopy
|
||||
import time
|
||||
|
||||
|
||||
class MultiClient(object):
|
||||
def __init__(self, servers: list[dict]):
|
||||
self.servers = servers
|
||||
self.sock_list = deepcopy(self.servers)
|
||||
self.sel = None
|
||||
|
||||
for server in self.sock_list:
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
sock.setblocking(False)
|
||||
server['sock'] = sock
|
||||
|
||||
def connect(self):
|
||||
self.sel = selectors.DefaultSelector()
|
||||
events = selectors.EVENT_READ
|
||||
for sock in self.sock_list:
|
||||
print(f"Starting connection {sock['name']} to {sock['host']}:{sock['port']}")
|
||||
self.sel.register(sock['sock'], events, data=None)
|
||||
sock['sock'].connect_ex((sock['host'], sock['port']))
|
||||
|
||||
def disconnect(self):
|
||||
for sock in self.sock_list:
|
||||
sock['sock'].close()
|
||||
|
||||
def on_receive(self, key):
|
||||
data = key.fileobj.recv(1024)
|
||||
print(f"{self}: on_receive: {data}")
|
||||
|
||||
def on_send(self, key):
|
||||
return None
|
||||
|
||||
def send_request(self):
|
||||
events = selectors.EVENT_READ | selectors.EVENT_WRITE
|
||||
for sock in self.sock_list:
|
||||
self.sel.register(sock['sock'], events, data=kakac)
|
||||
print("receive")
|
||||
|
||||
def event_loop(self):
|
||||
try:
|
||||
while True:
|
||||
events = self.sel.select(timeout=1)
|
||||
for key, mask in events:
|
||||
if mask & selectors.EVENT_READ:
|
||||
self.on_receive(key)
|
||||
if mask & selectors.EVENT_WRITE:
|
||||
data = self.on_send(key)
|
||||
if data is None:
|
||||
events = selectors.EVENT_READ
|
||||
for sock in self.sock_list:
|
||||
self.sel.register(sock['sock'], events, data=None)
|
||||
|
||||
except KeyboardInterrupt:
|
||||
print("Caught keyboard interrupt, exiting")
|
||||
finally:
|
||||
self.sel.close()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
server_list = [
|
||||
{'name': "neo-f9p", 'host': "192.168.22.93", 'port': 8721},
|
||||
{'name': "zed-x20p", 'host': "192.168.22.93", 'port': 8731}
|
||||
]
|
||||
|
||||
mc = MultiClient(server_list)
|
||||
mc.connect()
|
||||
mc.event_loop()
|
||||
time.sleep(1)
|
||||
mc.disconnect()
|
||||
Reference in New Issue
Block a user