mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2024-11-11 01:28:17 +03:00
ded9f68583
* protocols|server: Define new harvester plot refreshing protocol messages * protocols: Bump `protocol_version` to `0.0.34` * tests: Introduce `setup_farmer_multi_harvester` Allows to run a test setup with 1 farmer and mutiple harvesters. * plotting: Add an initial plot loading indication to `PlotManager` * plotting|tests: Don't add removed duplicates to `total_result.removed` `PlotRefreshResult.removed` should only contain plots that were loaded properly before they were removed. It shouldn't contain e.g. removed duplicates or invalid plots since those are synced in an extra sync step and not as diff but as whole list every time. * harvester: Reset `PlotManager` on shutdown * plot_sync: Implement plot sync protocol * farmer|harvester: Integrate and enable plot sync * tests: Implement tests for the plot sync protocol * farmer|tests: Drop obsolete harvester caching code * setup: Add `chia.plot_sync` to packages * plot_sync: Type hints in `DeltaType` * plot_sync: Drop parameters in `super()` calls * plot_sync: Introduce `send_response` helper in `Receiver._process` * plot_sync: Add some parentheses Co-authored-by: Kyle Altendorf <sda@fstab.net> * plot_sync: Additional hint for a `Receiver.process_path_list` parameter * plot_sync: Force named parameters in `Receiver.process_path_list` * test: Fix fixtures after rebase * tests: Fix sorting after rebase * tests: Return type hint for `plot_sync_setup` * tests: Rename `WSChiaConnection` and move it in the outer scope * tests|plot_sync: More type hints * tests: Rework some delta tests * tests: Drop a `range` and iterate over the list directly * tests: Use the proper flags to overwrite * test: More missing duplicates tests * tests: Drop `ExpectedResult.reset` * tests: Reduce some asserts * tests: Add messages to some `assert False` statements * tests: Introduce `ErrorSimulation` enum in `test_sync_simulated.py` * tests: Use `secrects` instead of `Crypto.Random` * Fixes after rebase * Import from `typing_extensions` to support python 3.7 * Drop task name to support python 3.7 * Introduce `Sender.syncing`, `Sender.connected` and a log about the task * Add `tests/plot_sync/config.py` * Align the multi harvester fixture with what we do in other places * Update the workflows Co-authored-by: Kyle Altendorf <sda@fstab.net>
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from dataclasses import dataclass, field
|
|
from typing import Dict, List, Union
|
|
|
|
from chia.protocols.harvester_protocol import Plot
|
|
|
|
|
|
@dataclass
|
|
class DeltaType:
|
|
additions: Union[Dict[str, Plot], List[str]]
|
|
removals: List[str]
|
|
|
|
def __str__(self) -> str:
|
|
return f"+{len(self.additions)}/-{len(self.removals)}"
|
|
|
|
def clear(self) -> None:
|
|
self.additions.clear()
|
|
self.removals.clear()
|
|
|
|
def empty(self) -> bool:
|
|
return len(self.additions) == 0 and len(self.removals) == 0
|
|
|
|
|
|
@dataclass
|
|
class PlotListDelta(DeltaType):
|
|
additions: Dict[str, Plot] = field(default_factory=dict)
|
|
removals: List[str] = field(default_factory=list)
|
|
|
|
|
|
@dataclass
|
|
class PathListDelta(DeltaType):
|
|
additions: List[str] = field(default_factory=list)
|
|
removals: List[str] = field(default_factory=list)
|
|
|
|
@staticmethod
|
|
def from_lists(old: List[str], new: List[str]) -> "PathListDelta":
|
|
return PathListDelta([x for x in new if x not in old], [x for x in old if x not in new])
|
|
|
|
|
|
@dataclass
|
|
class Delta:
|
|
valid: PlotListDelta = field(default_factory=PlotListDelta)
|
|
invalid: PathListDelta = field(default_factory=PathListDelta)
|
|
keys_missing: PathListDelta = field(default_factory=PathListDelta)
|
|
duplicates: PathListDelta = field(default_factory=PathListDelta)
|
|
|
|
def empty(self) -> bool:
|
|
return self.valid.empty() and self.invalid.empty() and self.keys_missing.empty() and self.duplicates.empty()
|
|
|
|
def __str__(self) -> str:
|
|
return (
|
|
f"valid {self.valid}, invalid {self.invalid}, keys missing: {self.keys_missing}, "
|
|
f"duplicates: {self.duplicates}"
|
|
)
|
|
|
|
def clear(self) -> None:
|
|
self.valid.clear()
|
|
self.invalid.clear()
|
|
self.keys_missing.clear()
|
|
self.duplicates.clear()
|