mirror of
https://github.com/logos-blockchain/logos-blockchain-specs.git
synced 2026-01-12 01:53:08 +00:00
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
import asyncio
|
|
import proto
|
|
from transport import Transport
|
|
|
|
class Executor:
|
|
def __init__(self, addr, port, col_num):
|
|
self.addr = addr
|
|
self.port = port
|
|
self.col_num = col_num
|
|
self.connections = []
|
|
self.interval = 10
|
|
|
|
async def execute(self):
|
|
data = "TEST DATA"
|
|
while True:
|
|
try:
|
|
for transport in self.connections:
|
|
message = proto.new_dispersal_put_msg(data)
|
|
await transport.write(message)
|
|
await asyncio.sleep(self.interval)
|
|
except asyncio.CancelledError:
|
|
break
|
|
except Exception as e:
|
|
print(f"Error during message sending: {e}")
|
|
|
|
async def connect(self):
|
|
try:
|
|
reader, writer = await asyncio.open_connection(self.addr, self.port)
|
|
conn_id = len(self.connections)
|
|
transport = Transport(conn_id, reader, writer, self._handle)
|
|
self.connections.append(transport)
|
|
print(f"Connected to {self.addr}:{self.port}, ID: {conn_id}")
|
|
asyncio.create_task(transport.read_and_process())
|
|
except Exception as e:
|
|
print(f"Failed to connect or lost connection: {e}")
|
|
|
|
async def _handle(self, conn_id, writer, message):
|
|
msg_type, msg_id, data = message
|
|
if msg_type == proto.DISPERSAL_OK:
|
|
print(f"Executor: Dispersal OK from connection {conn_id}, message ID {msg_id}.")
|
|
elif msg_type == proto.SAMPLE_OK:
|
|
print(f"Executor: Sample OK from connection {conn_id}, message ID {msg_id}.")
|
|
|
|
async def run(self):
|
|
await asyncio.gather(*(self.connect() for _ in range(self.col_num)))
|
|
await self.execute()
|
|
|