def test_replication_progress_pre_calculate():
subprocess.call("zfs destroy -r data/src", shell=True)
subprocess.call("zfs destroy -r data/dst", shell=True)
subprocess.check_call("zfs create data/src", shell=True)
subprocess.check_call("zfs create data/src/alice", shell=True)
subprocess.check_call("zfs create data/src/bob", shell=True)
subprocess.check_call("zfs create data/src/charlie", shell=True)
subprocess.check_call("zfs snapshot -r data/src@2018-10-01_01-00",
shell=True)
subprocess.check_call("zfs create data/dst", shell=True)
subprocess.check_call(
"zfs send -R data/src@2018-10-01_01-00 | zfs recv -s -F data/dst",
shell=True)
subprocess.check_call("zfs create data/src/dave", shell=True)
subprocess.check_call("zfs snapshot -r data/src@2018-10-01_02-00",
shell=True)
definition = yaml.safe_load(
textwrap.dedent("""\
timezone: "UTC"
replication-tasks:
src:
direction: push
transport:
type: local
source-dataset: data/src
target-dataset: data/dst
recursive: true
also-include-naming-schema:
- "%Y-%m-%d_%H-%M"
auto: false
retention-policy: none
retries: 1
"""))
definition = Definition.from_data(definition)
zettarepl = create_zettarepl(definition)
zettarepl._spawn_replication_tasks(
select_by_class(ReplicationTask, definition.tasks))
wait_replication_tasks_to_complete(zettarepl)
calls = [
call for call in zettarepl.observer.call_args_list
if call[0][0].__class__ != ReplicationTaskDataProgress
]
result = [
ReplicationTaskStart("src"),
ReplicationTaskSnapshotStart("src", "data/src", "2018-10-01_02-00", 0,
5),
ReplicationTaskSnapshotSuccess("src", "data/src", "2018-10-01_02-00",
1, 5),
ReplicationTaskSnapshotStart("src", "data/src/alice",
"2018-10-01_02-00", 1, 5),
ReplicationTaskSnapshotSuccess("src", "data/src/alice",
"2018-10-01_02-00", 2, 5),
ReplicationTaskSnapshotStart("src", "data/src/bob", "2018-10-01_02-00",
2, 5),
ReplicationTaskSnapshotSuccess("src", "data/src/bob",
"2018-10-01_02-00", 3, 5),
ReplicationTaskSnapshotStart("src", "data/src/charlie",
"2018-10-01_02-00", 3, 5),
ReplicationTaskSnapshotSuccess("src", "data/src/charlie",
"2018-10-01_02-00", 4, 5),
ReplicationTaskSnapshotStart("src", "data/src/dave",
"2018-10-01_02-00", 4, 5),
ReplicationTaskSnapshotSuccess("src", "data/src/dave",
"2018-10-01_02-00", 5, 5),
ReplicationTaskSuccess("src"),
]
for i, message in enumerate(result):
call = calls[i]
assert call[0][0].__class__ == message.__class__
d1 = call[0][0].__dict__
d2 = message.__dict__
assert d1 == d2
def test_replication_progress(transport):
subprocess.call("zfs destroy -r data/src", shell=True)
subprocess.call("zfs destroy -r data/dst", shell=True)
subprocess.check_call("zfs create data/src", shell=True)
subprocess.check_call("zfs create data/src/src1", shell=True)
subprocess.check_call("zfs snapshot data/src/src1@2018-10-01_01-00",
shell=True)
subprocess.check_call(
"dd if=/dev/urandom of=/mnt/data/src/src1/blob bs=1M count=1",
shell=True)
subprocess.check_call("zfs snapshot data/src/src1@2018-10-01_02-00",
shell=True)
subprocess.check_call("rm /mnt/data/src/src1/blob", shell=True)
subprocess.check_call("zfs snapshot data/src/src1@2018-10-01_03-00",
shell=True)
subprocess.check_call("zfs create data/src/src2", shell=True)
subprocess.check_call("zfs snapshot data/src/src2@2018-10-01_01-00",
shell=True)
subprocess.check_call("zfs snapshot data/src/src2@2018-10-01_02-00",
shell=True)
subprocess.check_call("zfs snapshot data/src/src2@2018-10-01_03-00",
shell=True)
subprocess.check_call("zfs snapshot data/src/src2@2018-10-01_04-00",
shell=True)
definition = yaml.safe_load(
textwrap.dedent("""\
timezone: "UTC"
replication-tasks:
src:
direction: push
source-dataset:
- data/src/src1
- data/src/src2
target-dataset: data/dst
recursive: true
also-include-naming-schema:
- "%Y-%m-%d_%H-%M"
auto: false
retention-policy: none
retries: 1
"""))
definition["replication-tasks"]["src"]["transport"] = transport
if transport["type"] == "ssh":
definition["replication-tasks"]["src"]["speed-limit"] = 10240 * 9
definition = Definition.from_data(definition)
zettarepl = create_zettarepl(definition)
zettarepl._spawn_replication_tasks(
select_by_class(ReplicationTask, definition.tasks))
wait_replication_tasks_to_complete(zettarepl)
calls = [
call for call in zettarepl.observer.call_args_list
if call[0][0].__class__ != ReplicationTaskDataProgress
]
result = [
ReplicationTaskStart("src"),
ReplicationTaskSnapshotStart("src", "data/src/src1",
"2018-10-01_01-00", 0, 3),
ReplicationTaskSnapshotSuccess("src", "data/src/src1",
"2018-10-01_01-00", 1, 3),
ReplicationTaskSnapshotStart("src", "data/src/src1",
"2018-10-01_02-00", 1, 3),
ReplicationTaskSnapshotSuccess("src", "data/src/src1",
"2018-10-01_02-00", 2, 3),
ReplicationTaskSnapshotStart("src", "data/src/src1",
"2018-10-01_03-00", 2, 3),
ReplicationTaskSnapshotSuccess("src", "data/src/src1",
"2018-10-01_03-00", 3, 3),
ReplicationTaskSnapshotStart("src", "data/src/src2",
"2018-10-01_01-00", 3, 7),
ReplicationTaskSnapshotSuccess("src", "data/src/src2",
"2018-10-01_01-00", 4, 7),
ReplicationTaskSnapshotStart("src", "data/src/src2",
"2018-10-01_02-00", 4, 7),
ReplicationTaskSnapshotSuccess("src", "data/src/src2",
"2018-10-01_02-00", 5, 7),
ReplicationTaskSnapshotStart("src", "data/src/src2",
"2018-10-01_03-00", 5, 7),
ReplicationTaskSnapshotSuccess("src", "data/src/src2",
"2018-10-01_03-00", 6, 7),
ReplicationTaskSnapshotStart("src", "data/src/src2",
"2018-10-01_04-00", 6, 7),
ReplicationTaskSnapshotSuccess("src", "data/src/src2",
"2018-10-01_04-00", 7, 7),
ReplicationTaskSuccess("src"),
]
if transport["type"] == "ssh":
result.insert(
4,
ReplicationTaskSnapshotProgress(
"src",
"data/src/src1",
"2018-10-01_02-00",
1,
3,
10240 * 9 * 10, # We poll for progress every 10 seconds so
# we would have transferred 10x speed limit
2162784 # Empirical value
))
for i, message in enumerate(result):
call = calls[i]
assert call[0][0].__class__ == message.__class__, calls
d1 = call[0][0].__dict__
d2 = message.__dict__
if isinstance(message, ReplicationTaskSnapshotProgress):
bytes_sent_1 = d1.pop("bytes_sent")
bytes_total_1 = d1.pop("bytes_total")
bytes_sent_2 = d2.pop("bytes_sent")
bytes_total_2 = d2.pop("bytes_total")
assert 0.8 <= bytes_sent_1 / bytes_sent_2 <= 1.2
assert 0.8 <= bytes_total_1 / bytes_total_2 <= 1.2
assert d1 == d2
def run_replication_step(step: ReplicationStep, observer=None, observer_snapshot=None):
logger.info(
"For replication task %r: doing %s from %r to %r of snapshot=%r incremental_base=%r receive_resume_token=%r "
"encryption=%r",
step.replication_task.id, step.replication_task.direction.value, step.src_dataset, step.dst_dataset,
step.snapshot, step.incremental_base, step.receive_resume_token, step.encryption is not None,
)
observer_snapshot = observer_snapshot or step.snapshot
notify(observer, ReplicationTaskSnapshotStart(
step.replication_task.id, step.src_dataset, observer_snapshot,
step.src_context.context.snapshots_sent, step.src_context.context.snapshots_total,
))
# Umount target dataset because we will be overwriting its contents and children mountpoints
# will become dangling. ZFS will mount entire directory structure again after receiving.
try:
step.dst_context.shell.exec(["zfs", "umount", step.dst_dataset])
except ExecException:
pass
if step.replication_task.direction == ReplicationDirection.PUSH:
local_context = step.src_context
remote_context = step.dst_context
elif step.replication_task.direction == ReplicationDirection.PULL:
local_context = step.dst_context
remote_context = step.src_context
else:
raise ValueError(f"Invalid replication direction: {step.replication_task.direction!r}")
transport = remote_context.transport
process = transport.replication_process(
step.replication_task.id,
transport,
local_context.shell,
remote_context.shell,
step.replication_task.direction,
step.src_dataset,
step.dst_dataset,
step.snapshot,
step.replication_task.properties,
step.replication_task.properties_exclude,
step.replication_task.properties_override,
step.replication_task.replicate,
step.encryption,
step.incremental_base,
step.receive_resume_token,
step.replication_task.compression,
step.replication_task.speed_limit,
step.replication_task.dedup,
step.replication_task.large_block,
step.replication_task.embed,
step.replication_task.compressed,
step.replication_task.properties and step.src_context.datasets_encrypted[step.src_dataset],
)
process.add_progress_observer(
lambda bytes_sent, bytes_total:
notify(observer, ReplicationTaskSnapshotProgress(
step.replication_task.id, step.src_dataset, observer_snapshot,
step.src_context.context.snapshots_sent, step.src_context.context.snapshots_total,
bytes_sent, bytes_total,
))
)
monitor = ReplicationMonitor(step.dst_context.shell, step.dst_dataset)
ReplicationProcessRunner(process, monitor).run()
step.template.src_context.context.snapshots_sent_by_replication_step_template[step.template] += 1
notify(observer, ReplicationTaskSnapshotSuccess(
step.replication_task.id, step.src_dataset, observer_snapshot,
step.src_context.context.snapshots_sent, step.src_context.context.snapshots_total,
))
if step.incremental_base is None:
# Might have created dataset, need to set it to readonly
handle_readonly(step.template)
def test_replication_progress_resume():
subprocess.call("zfs destroy -r data/src", shell=True)
subprocess.call("zfs destroy -r data/dst", shell=True)
subprocess.check_call("zfs create data/src", shell=True)
subprocess.check_call("zfs snapshot data/src@2018-10-01_01-00", shell=True)
subprocess.check_call(
"dd if=/dev/urandom of=/mnt/data/src/blob bs=1M count=1", shell=True)
subprocess.check_call("zfs snapshot data/src@2018-10-01_02-00", shell=True)
subprocess.check_call(
"dd if=/dev/urandom of=/mnt/data/src/blob bs=1M count=1", shell=True)
subprocess.check_call("zfs snapshot data/src@2018-10-01_03-00", shell=True)
subprocess.check_call(
"dd if=/dev/urandom of=/mnt/data/src/blob bs=1M count=1", shell=True)
subprocess.check_call("zfs snapshot data/src@2018-10-01_04-00", shell=True)
subprocess.check_call("zfs create data/dst", shell=True)
subprocess.check_call(
"zfs send data/src@2018-10-01_01-00 | zfs recv -s -F data/dst",
shell=True)
subprocess.check_call(
"(zfs send -i data/src@2018-10-01_01-00 data/src@2018-10-01_02-00 | "
" throttle -b 102400 | zfs recv -s -F data/dst) & "
"sleep 1; killall zfs",
shell=True)
assert "receive_resume_token\t1-" in subprocess.check_output(
"zfs get -H receive_resume_token data/dst",
shell=True,
encoding="utf-8")
definition = yaml.safe_load(
textwrap.dedent("""\
timezone: "UTC"
replication-tasks:
src:
direction: push
transport:
type: local
source-dataset: data/src
target-dataset: data/dst
recursive: true
also-include-naming-schema:
- "%Y-%m-%d_%H-%M"
auto: false
retention-policy: none
retries: 1
"""))
definition = Definition.from_data(definition)
zettarepl = create_zettarepl(definition)
zettarepl._spawn_replication_tasks(
select_by_class(ReplicationTask, definition.tasks))
wait_replication_tasks_to_complete(zettarepl)
calls = [
call for call in zettarepl.observer.call_args_list
if call[0][0].__class__ != ReplicationTaskDataProgress
]
result = [
ReplicationTaskStart("src"),
ReplicationTaskSnapshotStart("src", "data/src", "2018-10-01_02-00", 0,
3),
ReplicationTaskSnapshotSuccess("src", "data/src", "2018-10-01_02-00",
1, 3),
ReplicationTaskSnapshotStart("src", "data/src", "2018-10-01_03-00", 1,
3),
ReplicationTaskSnapshotSuccess("src", "data/src", "2018-10-01_03-00",
2, 3),
ReplicationTaskSnapshotStart("src", "data/src", "2018-10-01_04-00", 2,
3),
ReplicationTaskSnapshotSuccess("src", "data/src", "2018-10-01_04-00",
3, 3),
ReplicationTaskSuccess("src"),
]
for i, message in enumerate(result):
call = calls[i]
assert call[0][0].__class__ == message.__class__
d1 = call[0][0].__dict__
d2 = message.__dict__
assert d1 == d2
def run_replication_step(step: ReplicationStep,
observer=None,
observer_snapshot=None):
logger.info(
"For replication task %r: doing %s from %r to %r of snapshot=%r incremental_base=%r "
"receive_resume_token=%r", step.replication_task.id,
step.replication_task.direction.value, step.src_dataset,
step.dst_dataset, step.snapshot, step.incremental_base,
step.receive_resume_token)
observer_snapshot = observer_snapshot or step.snapshot
notify(
observer,
ReplicationTaskSnapshotStart(
step.replication_task.id,
step.src_dataset,
observer_snapshot,
step.src_context.context.snapshots_sent,
step.src_context.context.snapshots_total,
))
if step.replication_task.direction == ReplicationDirection.PUSH:
local_context = step.src_context
remote_context = step.dst_context
elif step.replication_task.direction == ReplicationDirection.PULL:
local_context = step.dst_context
remote_context = step.src_context
else:
raise ValueError(
f"Invalid replication direction: {step.replication_task.direction!r}"
)
transport = remote_context.transport
process = transport.replication_process(
step.replication_task.id, transport, local_context.shell,
remote_context.shell, step.replication_task.direction,
step.src_dataset, step.dst_dataset, step.snapshot,
step.replication_task.properties, step.replication_task.replicate,
step.incremental_base, step.receive_resume_token,
step.replication_task.compression, step.replication_task.speed_limit,
step.replication_task.dedup, step.replication_task.large_block,
step.replication_task.embed, step.replication_task.compressed)
process.add_progress_observer(lambda bytes_sent, bytes_total: notify(
observer,
ReplicationTaskSnapshotProgress(
step.replication_task.id,
step.src_dataset,
observer_snapshot,
step.src_context.context.snapshots_sent,
step.src_context.context.snapshots_total,
bytes_sent,
bytes_total,
)))
monitor = ReplicationMonitor(step.dst_context.shell, step.dst_dataset)
ReplicationProcessRunner(process, monitor).run()
step.template.src_context.context.snapshots_sent_by_replication_step_template[
step.template] += 1
notify(
observer,
ReplicationTaskSnapshotSuccess(
step.replication_task.id,
step.src_dataset,
observer_snapshot,
step.src_context.context.snapshots_sent,
step.src_context.context.snapshots_total,
))
if step.incremental_base is None:
# Might have created dataset, need to set it to readonly
handle_readonly(step.template)
