Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Track weight proof tasks #18896

Open
wants to merge 3 commits into
base: main
Choose a base branch
from
+21 −10
Open

Track weight proof tasks #18896

Changes from 2 commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
c6540d6
keep track of wp tasks
almogdepaz Nov 19, 2024
d3aee2b
use cancel_task_safe
almogdepaz Nov 19, 2024
48bdbde
await at shutdown
almogdepaz Nov 26, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 12 additions & 9 deletions chia/full_node/full_node.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,7 @@ class FullNode:
log: logging.Logger
db_path: Path
wallet_sync_queue: asyncio.Queue[WalletUpdate]
_segment_task: Optional[asyncio.Task[None]] = None
_segment_task_list: list[asyncio.Task[None]] = dataclasses.field(default_factory=list)
initialized: bool = False
_server: Optional[ChiaServer] = None
_shut_down: bool = False
Expand Down Expand Up @@ -599,12 +599,11 @@ async def short_sync_batch(self, peer: WSChiaConnection, start_height: uint32, t
return False

batch_size = self.constants.MAX_BLOCK_COUNT_PER_REQUESTS
if self._segment_task is not None and (not self._segment_task.done()):
try:
self._segment_task.cancel()
except Exception as e:
self.log.warning(f"failed to cancel segment task {e}")
self._segment_task = None
for task in self._segment_task_list[:]:
if task.done():
self._segment_task_list.remove(task)
else:
cancel_task_safe(task=task, log=self.log)

try:
peer_info = peer.get_peer_logging()
Expand Down Expand Up @@ -2223,8 +2222,12 @@ async def add_block(

record = self.blockchain.block_record(block.header_hash)
if self.weight_proof_handler is not None and record.sub_epoch_summary_included is not None:
if self._segment_task is None or self._segment_task.done():
self._segment_task = asyncio.create_task(self.weight_proof_handler.create_prev_sub_epoch_segments())
self._segment_task_list.append(
asyncio.create_task(self.weight_proof_handler.create_prev_sub_epoch_segments())
)
for task in self._segment_task_list[:]:
if task.done():
self._segment_task_list.remove(task)
return None

async def add_unfinished_block(
Expand Down
Loading