31 lines
981 B
Python
31 lines
981 B
Python
from typing import TYPE_CHECKING
|
|
|
|
from ...constants import BLOCK_MAX_SIZE, BLOCK_MAX_SIZE_LEN
|
|
|
|
if TYPE_CHECKING:
|
|
from queue import Queue
|
|
import socket
|
|
from ...dandelion import DandelionPhase
|
|
|
|
from onionrblocks import Block
|
|
|
|
|
|
async def do_stem_stream(
|
|
peer_socket: 'socket.socket',
|
|
block_queue: "Queue[Block]",
|
|
d_phase: 'DandelionPhase'):
|
|
remaining_time = d_phase.remaining_time()
|
|
my_phase_id = d_phase.phase_id
|
|
|
|
with peer_socket:
|
|
while remaining_time > 5 and my_phase_id == d_phase.phase_id:
|
|
# Primary client component that communicate's with gossip.server.acceptstem
|
|
remaining_time = d_phase.remaining_time()
|
|
bl: 'Block' = block_queue.get(block=True, timeout=remaining_time)
|
|
|
|
block_size = str(len(bl.raw)).zfill(BLOCK_MAX_SIZE_LEN)
|
|
|
|
peer_socket.sendall(bl.id)
|
|
peer_socket.sendall(block_size.encode('utf-8'))
|
|
peer_socket.sendall(bl.raw)
|