ic_backup/workflow/ic_snapshot_metadata/
mod.rs1use crate::{
4 model::{
5 attempt_journal::{MutationOutcomeRecord, MutationReceiptRequest},
6 execution_workflow::ExecutionStagePredecessorRecord,
7 ic_snapshot_metadata::IcSnapshotMetadataRequest,
8 ic_snapshot_transfer_read::{
9 IcSnapshotTransferReadError, IcSnapshotTransferReadPayload,
10 IcSnapshotTransferReadRequest, IcSnapshotTransferReadResponse,
11 },
12 },
13 ops::persistence::{
14 AttemptJournalError, AttemptJournalGuard, ExecutionStageCheckpointError,
15 ExecutionStageGuard, ExecutionWorkflowPersistenceError,
16 },
17 policy::ic_snapshot_transfer_read::{
18 IcSnapshotTransferReadAssociationError, IcSnapshotTransferReadReply, validate_response,
19 },
20 ports::ic_snapshot_transfer_read::IcSnapshotTransferReadProvider,
21 workflow::ic_snapshot_transfer_read::{IcSnapshotTransferReadExecutionError, read_snapshot},
22};
23use thiserror::Error;
24
25pub async fn read_snapshot_metadata<E: std::error::Error + 'static>(
55 stage: &ExecutionStageGuard<'_>,
56 operation_sequence: u64,
57 payload: &IcSnapshotMetadataRequest,
58 provider: &mut impl IcSnapshotTransferReadProvider,
59 admit: impl AsyncFnOnce(&IcSnapshotTransferReadRequest<'_, '_>) -> Result<(), E>,
60 qualify: impl AsyncFnOnce(
61 &IcSnapshotTransferReadRequest<'_, '_>,
62 &IcSnapshotTransferReadResponse,
63 ) -> Result<MutationReceiptRequest, E>,
64) -> Result<
65 (
66 IcSnapshotTransferReadResponse,
67 ExecutionStagePredecessorRecord,
68 ),
69 IcSnapshotMetadataExecutionError<E>,
70> {
71 let operations = stage.plan().operations();
72 if operations.len() != 1
73 || operations[0].operation_sequence() != operation_sequence
74 || operations[0].request() != payload.digest().hash()
75 {
76 return Err(IcSnapshotMetadataExecutionError::OriginalMismatch);
77 }
78 let response = read_snapshot(
79 stage,
80 operation_sequence,
81 IcSnapshotTransferReadPayload::Metadata(payload),
82 provider,
83 admit,
84 )
85 .await?;
86 match settle_metadata(stage, operation_sequence, payload, &response, qualify).await {
87 Ok(predecessor) => Ok((response, predecessor)),
88 Err(source) => Err(IcSnapshotMetadataExecutionError::AfterReply {
89 source,
90 response: Box::new(response),
91 }),
92 }
93}
94
95async fn settle_metadata<E: std::error::Error + 'static>(
96 stage: &ExecutionStageGuard<'_>,
97 sequence: u64,
98 payload: &IcSnapshotMetadataRequest,
99 response: &IcSnapshotTransferReadResponse,
100 qualify: impl AsyncFnOnce(
101 &IcSnapshotTransferReadRequest<'_, '_>,
102 &IcSnapshotTransferReadResponse,
103 ) -> Result<MutationReceiptRequest, E>,
104) -> Result<ExecutionStagePredecessorRecord, IcSnapshotMetadataSettlementError<E>> {
105 let plan = stage.plan();
106 let authority = plan
107 .attempt_authority(sequence)
108 .map_err(IcSnapshotTransferReadError::from)?;
109 let mut journal = AttemptJournalGuard::open(stage.layout()?, &authority)?;
110 let request = IcSnapshotTransferReadRequest::new(
111 plan,
112 sequence,
113 journal.record()?,
114 IcSnapshotTransferReadPayload::Metadata(payload),
115 )?;
116 let admitted = validate_response(&request, journal.record()?, response)?;
117 let receipt = qualify(&request, response)
118 .await
119 .map_err(IcSnapshotMetadataSettlementError::Qualification)?;
120 if receipt.outcome != MutationOutcomeRecord::Applied
121 || receipt.attempt != request.mutation_attempt()
122 || receipt.request != payload.digest().hash()
123 {
124 return Err(IcSnapshotMetadataSettlementError::ReceiptRequired);
125 }
126 let IcSnapshotTransferReadReply::Metadata(metadata) = admitted.reply() else {
127 return Err(IcSnapshotMetadataSettlementError::ReceiptRequired);
128 };
129 stage.layout()?;
130 journal.record_mutation(receipt)?;
131 drop(journal);
132 Ok(stage.checkpoint(metadata.digest())?)
133}
134
135#[derive(Debug, Error)]
137pub enum IcSnapshotMetadataExecutionError<E: std::error::Error + 'static> {
138 #[error("snapshot metadata requires an exact singleton original stage")]
140 OriginalMismatch,
141 #[error(transparent)]
143 Read(#[from] IcSnapshotTransferReadExecutionError<E>),
144 #[error("snapshot metadata reply settlement failed: {source}")]
146 AfterReply {
147 source: IcSnapshotMetadataSettlementError<E>,
149 response: Box<IcSnapshotTransferReadResponse>,
151 },
152}
153
154#[derive(Debug, Error)]
156pub enum IcSnapshotMetadataSettlementError<E: std::error::Error + 'static> {
157 #[error("snapshot metadata qualification failed: {0}")]
159 Qualification(#[source] E),
160 #[error("snapshot metadata requires an explicit original Applied receipt")]
162 ReceiptRequired,
163 #[error(transparent)]
165 Stage(#[from] ExecutionWorkflowPersistenceError),
166 #[error(transparent)]
168 Journal(#[from] AttemptJournalError),
169 #[error(transparent)]
171 Request(#[from] IcSnapshotTransferReadError),
172 #[error(transparent)]
174 Association(#[from] IcSnapshotTransferReadAssociationError),
175 #[error(transparent)]
177 Checkpoint(#[from] ExecutionStageCheckpointError),
178}
179
180#[cfg(all(test, unix))]
181mod tests;