Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
90925fd321 | ||
|
|
9df2e6991e |
@@ -2,7 +2,7 @@ name: archive-control-archive
|
|||||||
|
|
||||||
services:
|
services:
|
||||||
archive-client:
|
archive-client:
|
||||||
image: sodium/archive-clients:v0.1.10
|
image: sodium/archive-clients:v0.1.12
|
||||||
user: "1000:1000"
|
user: "1000:1000"
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
command: ["--config", "/etc/archive-control/client.toml"]
|
command: ["--config", "/etc/archive-control/client.toml"]
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ name: archive-control-cache
|
|||||||
|
|
||||||
services:
|
services:
|
||||||
archive-client:
|
archive-client:
|
||||||
image: sodium/archive-clients:v0.1.10
|
image: sodium/archive-clients:v0.1.12
|
||||||
user: "1001:1001"
|
user: "1001:1001"
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
network_mode: host
|
network_mode: host
|
||||||
|
|||||||
@@ -2,7 +2,7 @@ name: archive-control-cache
|
|||||||
|
|
||||||
services:
|
services:
|
||||||
archive-client:
|
archive-client:
|
||||||
image: sodium/archive-clients:v0.1.10
|
image: sodium/archive-clients:v0.1.12
|
||||||
user: "1001:1001"
|
user: "1001:1001"
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
network_mode: host
|
network_mode: host
|
||||||
|
|||||||
@@ -218,7 +218,7 @@ cache/archive routes according to policy.
|
|||||||
```yaml
|
```yaml
|
||||||
services:
|
services:
|
||||||
archive-client:
|
archive-client:
|
||||||
image: sodium/archive-clients:v0.1.10
|
image: sodium/archive-clients:v0.1.12
|
||||||
user: "1001:1001"
|
user: "1001:1001"
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
command: ["archive-client", "--config", "/etc/archive-control/client.toml"]
|
command: ["archive-client", "--config", "/etc/archive-control/client.toml"]
|
||||||
|
|||||||
+1
-1
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
|
|||||||
|
|
||||||
[project]
|
[project]
|
||||||
name = "archive-clients"
|
name = "archive-clients"
|
||||||
version = "0.1.10"
|
version = "0.1.12"
|
||||||
requires-python = ">=3.11"
|
requires-python = ">=3.11"
|
||||||
dependencies = ["protobuf==7.35.1", "websockets==16.0"]
|
dependencies = ["protobuf==7.35.1", "websockets==16.0"]
|
||||||
|
|
||||||
|
|||||||
@@ -175,7 +175,12 @@ class ClientJobExecutor:
|
|||||||
if event_callback is not None:
|
if event_callback is not None:
|
||||||
for event in emitted:
|
for event in emitted:
|
||||||
event_callback(event)
|
event_callback(event)
|
||||||
speed_sample = [time.monotonic(), 0]
|
# A newly-started step must publish its first observation. In
|
||||||
|
# particular, a sparse Syncthing temporary file may retain the same
|
||||||
|
# allocated-block count for a long time while data is still needed;
|
||||||
|
# suppressing that first observation made a healthy transfer look
|
||||||
|
# permanently stalled to the control daemon.
|
||||||
|
speed_sample: list[float | int | None] = [None, 0]
|
||||||
|
|
||||||
def progress(
|
def progress(
|
||||||
fraction: float,
|
fraction: float,
|
||||||
@@ -185,8 +190,12 @@ class ClientJobExecutor:
|
|||||||
) -> None:
|
) -> None:
|
||||||
nonlocal cursor
|
nonlocal cursor
|
||||||
now = time.monotonic()
|
now = time.monotonic()
|
||||||
elapsed = now - float(speed_sample[0])
|
previous_time = speed_sample[0]
|
||||||
if fraction < 1 and elapsed < 1:
|
elapsed = (
|
||||||
|
now - float(previous_time)
|
||||||
|
if previous_time is not None else 0.0
|
||||||
|
)
|
||||||
|
if previous_time is not None and fraction < 1 and elapsed < 1:
|
||||||
return
|
return
|
||||||
speed = (
|
speed = (
|
||||||
max(0, bytes_complete - int(speed_sample[1])) / elapsed
|
max(0, bytes_complete - int(speed_sample[1])) / elapsed
|
||||||
|
|||||||
@@ -192,11 +192,19 @@ class SyncthingTransferObserver:
|
|||||||
for relative_path, expected_bytes in declared_files
|
for relative_path, expected_bytes in declared_files
|
||||||
)
|
)
|
||||||
observed_fraction = observed_bytes / total if total else 1.0
|
observed_fraction = observed_bytes / total if total else 1.0
|
||||||
# Folder completion remains useful when Syncthing has already
|
# ``/db/completion`` is folder-wide. It can be near zero when a
|
||||||
# atomically published a file but still reports it in its need
|
# route contains a freshly-created item even though the tracked
|
||||||
# queue; the allocated-byte estimate is needed for sparse temp
|
# job's temporary payload has already received many blocks. For
|
||||||
# files that are pre-sized before their blocks arrive.
|
# an incomplete temporary file, its allocated blocks are the only
|
||||||
fraction = min(observed_fraction, float(raw_fraction) / 100)
|
# job-specific signal, so do not cap them with that unrelated
|
||||||
|
# folder aggregate. Once every declared file is atomically
|
||||||
|
# present, retain the completion value as a conservative guard
|
||||||
|
# until the need queue has caught up.
|
||||||
|
fraction = (
|
||||||
|
observed_fraction
|
||||||
|
if observed_fraction < 1.0
|
||||||
|
else min(1.0, float(raw_fraction) / 100)
|
||||||
|
)
|
||||||
completed_bytes = int(total * fraction)
|
completed_bytes = int(total * fraction)
|
||||||
if complete:
|
if complete:
|
||||||
fraction = 1.0
|
fraction = 1.0
|
||||||
|
|||||||
@@ -177,6 +177,49 @@ class ClientJobHappyPathTests(unittest.TestCase):
|
|||||||
control_pb2.JOB_EVENT_TYPE_STEP_SUCCEEDED,
|
control_pb2.JOB_EVENT_TYPE_STEP_SUCCEEDED,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def test_first_partial_progress_is_durable(self):
|
||||||
|
"""A sparse transfer must not lose its only initial observation."""
|
||||||
|
with tempfile.TemporaryDirectory() as directory:
|
||||||
|
root = Path(directory)
|
||||||
|
store = ClientStore(root / "client.db")
|
||||||
|
store.initialize()
|
||||||
|
definition = job_pb2.JobDefinition(
|
||||||
|
job_id=str(uuid4()),
|
||||||
|
idempotency_key=str(uuid4()),
|
||||||
|
operation=job_pb2.JOB_OPERATION_ARCHIVE,
|
||||||
|
transfer={
|
||||||
|
"source_client_id": "cache-1",
|
||||||
|
"target_client_id": "archive-1",
|
||||||
|
"route_id": "route-1",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
definition.resource_id.info_hash_v1_hex = "a" * 40
|
||||||
|
definition.created_at.GetCurrentTime()
|
||||||
|
executor = ClientJobExecutor(
|
||||||
|
client_id="archive-1", qbittorrent=Mock(), store=store,
|
||||||
|
qb_root=root, qb_api_root=Path("/downloads"),
|
||||||
|
route_path=lambda _: root, syncthing_transport=Mock(),
|
||||||
|
sparse_supported=True, poll_interval=0,
|
||||||
|
)
|
||||||
|
executor.assign(control_pb2.AssignJobCommand(
|
||||||
|
job=definition, expected_job_revision=1,
|
||||||
|
expected_last_event_sequence=0,
|
||||||
|
))
|
||||||
|
|
||||||
|
def partial_progress(_definition, _step, progress):
|
||||||
|
progress(0.001, 10, 10_000, "still receiving")
|
||||||
|
|
||||||
|
with patch.object(executor, "_execute_step", partial_progress):
|
||||||
|
events = executor.execute(control_pb2.ExecuteStepCommand(
|
||||||
|
job_id=definition.job_id, expected_job_revision=1,
|
||||||
|
expected_last_event_sequence=1,
|
||||||
|
step=job_pb2.JOB_STEP_KIND_SYNCTHING_TRANSFER, attempt=1,
|
||||||
|
))
|
||||||
|
|
||||||
|
self.assertEqual(len(events), 3)
|
||||||
|
self.assertEqual(events[1].type, control_pb2.JOB_EVENT_TYPE_PROGRESS)
|
||||||
|
self.assertEqual(events[1].progress.bytes_complete, 10)
|
||||||
|
|
||||||
def _run_transfer(
|
def _run_transfer(
|
||||||
self,
|
self,
|
||||||
root: Path,
|
root: Path,
|
||||||
|
|||||||
Reference in New Issue
Block a user