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>
55 lines
2.2 KiB
Python
55 lines
2.2 KiB
Python
from typing import Any
|
|
|
|
from chia.plot_sync.util import ErrorCodes, State
|
|
from chia.protocols.harvester_protocol import PlotSyncIdentifier
|
|
from chia.server.ws_connection import NodeType
|
|
from chia.util.ints import uint64
|
|
|
|
|
|
class PlotSyncException(Exception):
|
|
def __init__(self, message: str, error_code: ErrorCodes) -> None:
|
|
super().__init__(message)
|
|
self.error_code = error_code
|
|
|
|
|
|
class AlreadyStartedError(Exception):
|
|
def __init__(self) -> None:
|
|
super().__init__("Already started!")
|
|
|
|
|
|
class InvalidValueError(PlotSyncException):
|
|
def __init__(self, message: str, actual: Any, expected: Any, error_code: ErrorCodes) -> None:
|
|
super().__init__(f"{message}: Actual {actual}, Expected {expected}", error_code)
|
|
|
|
|
|
class InvalidIdentifierError(InvalidValueError):
|
|
def __init__(self, actual_identifier: PlotSyncIdentifier, expected_identifier: PlotSyncIdentifier) -> None:
|
|
super().__init__("Invalid identifier", actual_identifier, expected_identifier, ErrorCodes.invalid_identifier)
|
|
self.actual_identifier: PlotSyncIdentifier = actual_identifier
|
|
self.expected_identifier: PlotSyncIdentifier = expected_identifier
|
|
|
|
|
|
class InvalidLastSyncIdError(InvalidValueError):
|
|
def __init__(self, actual: uint64, expected: uint64) -> None:
|
|
super().__init__("Invalid last-sync-id", actual, expected, ErrorCodes.invalid_last_sync_id)
|
|
|
|
|
|
class InvalidConnectionTypeError(InvalidValueError):
|
|
def __init__(self, actual: NodeType, expected: NodeType) -> None:
|
|
super().__init__("Unexpected connection type", actual, expected, ErrorCodes.invalid_connection_type)
|
|
|
|
|
|
class PlotAlreadyAvailableError(PlotSyncException):
|
|
def __init__(self, state: State, path: str) -> None:
|
|
super().__init__(f"{state.name}: Plot already available - {path}", ErrorCodes.plot_already_available)
|
|
|
|
|
|
class PlotNotAvailableError(PlotSyncException):
|
|
def __init__(self, state: State, path: str) -> None:
|
|
super().__init__(f"{state.name}: Plot not available - {path}", ErrorCodes.plot_not_available)
|
|
|
|
|
|
class SyncIdsMatchError(PlotSyncException):
|
|
def __init__(self, state: State, sync_id: uint64) -> None:
|
|
super().__init__(f"{state.name}: Sync ids are equal - {sync_id}", ErrorCodes.sync_ids_match)
|